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 条
  • [21] Multiple-kernel combination fuzzy clustering for community detection
    Lu, Hu
    Song, Yuqing
    Wei, Hui
    [J]. SOFT COMPUTING, 2020, 24 (18) : 14157 - 14165
  • [22] Multi-kernel fuzzy clustering based on auto-encoder for fMRI functional network
    Lu, Hu
    Liu, Saixiong
    Wei, Hui
    Tu, Juanjuan
    [J]. EXPERT SYSTEMS WITH APPLICATIONS, 2020, 159
  • [23] MacQueen J. B., 1967, P 5 BERKELEY S MATH, V1, P281
  • [24] Stacked Convolutional Auto-Encoders for Hierarchical Feature Extraction
    Masci, Jonathan
    Meier, Ueli
    Ciresan, Dan
    Schmidhuber, Juergen
    [J]. ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING - ICANN 2011, PT I, 2011, 6791 : 52 - 59
  • [25] Ng AY, 2002, ADV NEUR IN, V14, P849
  • [26] Opochinsky Y, 2020, INT CONF ACOUST SPEE, P4037, DOI [10.1109/icassp40776.2020.9053109, 10.1109/ICASSP40776.2020.9053109]
  • [27] Convolutional Autoencoder Based Feature Extraction and Clustering for Customer Load Analysis
    Ryu, Seunghyoung
    Choi, Hyungeun
    Lee, Hyoseop
    Kim, Hongseok
    [J]. IEEE TRANSACTIONS ON POWER SYSTEMS, 2020, 35 (02) : 1048 - 1060
  • [28] A generalized least-squares approach regularized with graph embedding for dimensionality reduction
    Shen, Xiang-Jun
    Liu, Si-Xing
    Bao, Bing-Kun
    Pan, Chun-Hong
    Zha, Zheng-Jun
    Fan, Jianping
    [J]. PATTERN RECOGNITION, 2020, 98 (98)
  • [29] Normalized cuts and image segmentation
    Shi, JB
    Malik, J
    [J]. IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2000, 22 (08) : 888 - 905
  • [30] Detection of tomato organs based on convolutional neural network under the overlap and occlusion backgrounds
    Sun, Jun
    He, Xiaofei
    Wu, Minmin
    Wu, Xiaohong
    Shen, Jifeng
    Lu, Bing
    [J]. MACHINE VISION AND APPLICATIONS, 2020, 31 (05)