Super Multi-class Deep Image Clustering Model Based on Contrastive Learning
Abstract
Image clustering reduces the dimensionality of image data,extracts effective features through representation learning,and performs cluster analysis.When there are many categories of image data,the complexity of data distribution and the density of clusters seriously affect the practicability of existing methods.To this end,this paper proposes a super-multi-class deep image clustering model based on contrastive learning,which is mainly divided into three stages:firstly,improving the contrastive lear-ning method to train the feature model to make the cluster distribution uniform;secondly,based on the principle of semantic similarity,the perspective mines instance semantic nearest neighbor information;and finally,the instance and its nearest neighbors are used as self-supervised information to train a clustering model.According to the different types of experiments,ablation experiments and contrast experiments are designed in this paper.The ablation experiments prove that the proposed method could make the clusters evenly distributed in the mapping space and mine the semantic nearest neighbor information reliably.In the comparative experiments,it's compared with the advanced algorithms on 7 benchmark datasets.On the ImageNet-200 class dataset,it's accuracy is 10.6% higher than the advanced method.It's accuracy rate on the ImageNet-1000 class dataset is higher than that of the advanced algorithm,which improves by 9.2%.
Date
01-09-2023Author
HU Shen, QIAN Yuhua, WANG Jieting, LI Feijiang, LYU Wei
