Self-supervised Pairing Image Clustering and Its Application in Cyber Manufacturing

被引:4
作者
Dai, Wenting [1 ]
Jiao, Yutao [1 ]
Erdt, Marius [2 ]
Sourin, Alexei [1 ]
机构
[1] Nanyang Technol Univ, Sch Comp Sci & Engn, Singapore, Singapore
[2] Nanyang Technol Univ, Fraunhofer Singapore, Singapore, Singapore
来源
2020 INTERNATIONAL CONFERENCE ON CYBERWORLDS (CW 2020) | 2020年
关键词
Image clustering; Self-supervised pairing; Cyber-manufacturing applications;
D O I
10.1109/CW49994.2020.00012
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Artificial intelligence is being increasingly applied in manufacturing to maximize industrial productivity. Image clustering, as a fundamental research direction in unsupervised learning, has been used in various fields. Since no label information is required in clustering, it can perform a preliminary analysis of the data while saving lots of manpower. In this paper, we propose a novel end-to-end clustering network called Self-supervised Pairing Image Clustering (SPIC) for industrial application, which produces clustering prediction for input images in an advanced pair classification network. For training this network, a self-supervised pairing module is built to form balanced pairs accurately and efficiently without label information. Since the existence of trivial solutions cannot be avoided in most of unsupervised learning methods, two additional information theoretic-constraints regularize the training that ensures the clustering prediction to be unambiguous and close to the real data distribution during training. Experimental results indicate that the proposed SPIC outperforms the state-of-art approaches on manufacturing datasets-NEU and DAGM. It also shows the execellent generalization capability on other genral public datasets, such as MNIST, Omniglot, CIFAR10, and CIFAR100.
引用
收藏
页码:25 / 32
页数:8
相关论文
共 31 条
[31]   Joint Unsupervised Learning of Deep Representations and Image Clusters [J].
Yang, Jianwei ;
Parikh, Devi ;
Batra, Dhruv .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :5147-5156