Improved deep convolutional embedded clustering with re-selectable sample training

The deep clustering algorithm can learn the latent features of the embedded subspace, and further realize the clustering of samples in the feature space. The existing deep clustering algorithms mostly integrate neural networks and traditional clustering algorithms. However, for sample sets with many noise points, the effect of the clustering remains unsatisfactory. To address this issue, we propose an improved deep convolutional embedded clustering algorithm using reliable samples (IDCEC) in this paper. The algorithm first uses the convolutional autoencoder to extract features and cluster the samples. Then we select reliable samples with pseudo-labels and pass them to the convolutional neural network for training to get a better clustering model. We construct a new loss function for backpropagation training and implement an unsupervised deep clustering method. To verify the performance of the method proposed in this paper, we conducted experimental tests on standard data sets such as MNIST and USPS. Experimental results show that our method has better performance compared to traditional clustering algorithms and the state-of-the-art deep clustering algorithm under four clustering metrics.(c) 2022 Elsevier Ltd. All rights reserved.

Automated summary

This paper proposes an improved deep convolutional embedded clustering algorithm (IDCEC) that improves clustering performance by using reliable samples. Experimental results show that the proposed method outperforms traditional clustering algorithms and state-of-the-art deep clustering algorithms under multiple clustering evaluation metrics.