Deep Comprehensive Correlation Mining for Image Clustering

Recent developed deep unsupervised methods allow us to jointly learn representation and cluster unlabelled data. These deep clustering methods mainly focus on the correlation among samples, e.g., selecting high precision pairs to gradually tune the feature representation, which neglects other useful correlations. In this paper, we propose a novel clustering framework, named deep comprehensive correlation mining(DCCM), for exploring and taking full advantage of various kinds of correlations behind the unlabeled data from three aspects: 1) Instead of only using pair-wise information, pseudo-label supervision is proposed to investigate category information and learn discriminative features. 2) The features' robustness to image transformation of input space is fully explored, which benefits the network learning and significantly improves the performance. 3) The triplet mutual information among features is presented for clustering problem to lift the recently discovered instance-level deep mutual information to a triplet-level formation, which further helps to learn more discriminative features. Extensive experiments on several challenging datasets show that our method achieves good performance, e.g., attaining $62.3\%$ clustering accuracy on CIFAR-10, which is $10.1\%$ higher than the state-of-the-art results.

PDF Abstract ICCV 2019 PDF ICCV 2019 Abstract

Results from the Paper


Task Dataset Model Metric Name Metric Value Global Rank Uses Extra
Training Data
Result Benchmark
Image Clustering CIFAR-10 DCCM Accuracy 0.623 # 14
NMI 0.496 # 13
Train set Train+Test # 1
ARI 0.408 # 12
Backbone AlexNet # 1
Image Clustering CIFAR-100 DCCM Accuracy 0.327 # 12
NMI 0.285 # 10
Train Set Train+Test # 1
Image Clustering ImageNet-10 DCCM Accuracy 0.71 # 8
NMI 0.608 # 8
Image Clustering Imagenet-dog-15 DCCM Accuracy 0.383 # 8
NMI 0.321 # 8
Image Clustering STL-10 DCCM Accuracy 0.482 # 14
NMI 0.376 # 12
Train Split Train+Test # 1
Backbone AlexNet # 1
Image Clustering Tiny-ImageNet DCCM Accuracy 0.108 # 6
NMI 0.224 # 6

Methods


No methods listed for this paper. Add relevant methods here