Improved deep convolutional embedded clustering with re-selectable sample training

被引:31
作者
Lu, Hu [1 ,3 ]
Chen, Chao [1 ]
Wei, Hui [2 ]
Ma, Zhongchen [1 ]
Jiang, Ke [1 ]
Wang, Yingquan [1 ]
机构
[1] JiangSu Univ, Comp Sci & Commun Engn, Zhenjiang 212013, Jiangsu, Peoples R China
[2] Fudan Univ, Sch Comp Sci, Shanghai, Peoples R China
[3] Jiangsu Prov Big Data Ubiquitous Percept & Intell, Zhenjiang, Peoples R China
关键词
Unsupervised clustering; Deep embedded clustering; Autoencoder; Reliable samples;
D O I
10.1016/j.patcog.2022.108611
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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.
引用
收藏
页数:10
相关论文
共 39 条
  • [1] Spectral clustering via ensemble deep autoencoder learning (SC-EDAE)
    Affeldt, Severine
    Labiod, Lazhar
    Nadif, Mohamed
    [J]. PATTERN RECOGNITION, 2020, 108
  • [2] Aljalbout E., 2018, ABS180107648 CORR
  • [3] Structural Deep Clustering Network
    Bo, Deyu
    Wang, Xiao
    Shi, Chuan
    Zhu, Meiqi
    Lu, Emiao
    Cui, Peng
    [J]. WEB CONFERENCE 2020: PROCEEDINGS OF THE WORLD WIDE WEB CONFERENCE (WWW 2020), 2020, : 1400 - 1410
  • [4] Unsupervised embedded feature learning for deep clustering with stacked sparse auto-encoder
    Cai, Jinyu
    Wang, Shiping
    Guo, Wenzhong
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2021, 186
  • [5] Chen X., 2011, PROC AAAI C ARTIF IN, P313
  • [6] Ester M., 1996, P 2 INT C KNOWL DISC, DOI DOI 10.5555/3001460.3001507
  • [7] Deep Fuzzy Clustering-A Representation Learning Approach
    Feng, Qiying
    Chen, Long
    Chen, C. L. Philip
    Guo, Li
    [J]. IEEE TRANSACTIONS ON FUZZY SYSTEMS, 2020, 28 (07) : 1420 - 1433
  • [8] A survey of kernel and spectral methods for clustering
    Filippone, Maurizio
    Camastra, Francesco
    Masulli, Francesco
    Rovetta, Stefano
    [J]. PATTERN RECOGNITION, 2008, 41 (01) : 176 - 190
  • [9] Network Structural Transformation-Based Community Detection with Autoencoder
    Geng, Xia
    Lu, Hu
    Sun, Jun
    [J]. SYMMETRY-BASEL, 2020, 12 (06):
  • [10] Guo XF, 2017, PROCEEDINGS OF THE TWENTY-SIXTH INTERNATIONAL JOINT CONFERENCE ON ARTIFICIAL INTELLIGENCE, P1753