A Hybrid Autoencoder Network for Unsupervised Image Clustering

被引:11
作者
Chen, Pei-Yin [1 ]
Huang, Jih-Jeng [1 ]
机构
[1] SooChow Univ, Dept Comp Sci & Informat Management, 56 Kueiyang St,Sect 1, Taipei 100, Taiwan
关键词
image clustering; convolutional autoencoder (CAE); adversarial autoencoder (AAE); stacked autoencoder (SAE);
D O I
10.3390/a12060122
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Image clustering involves the process of mapping an archive image into a cluster such that the set of clusters has the same information. It is an important field of machine learning and computer vision. While traditional clustering methods, such as k-means or the agglomerative clustering method, have been widely used for the task of clustering, it is difficult for them to handle image data due to having no predefined distance metrics and high dimensionality. Recently, deep unsupervised feature learning methods, such as the autoencoder (AE), have been employed for image clustering with great success. However, each model has its specialty and advantages for image clustering. Hence, we combine three AE-based models-the convolutional autoencoder (CAE), adversarial autoencoder (AAE), and stacked autoencoder (SAE)-to form a hybrid autoencoder (BAE) model for image clustering. The MNIST and CIFAR-10 datasets are used to test the result of the proposed models and compare the results with others. The results of the clustering criteria indicate that the proposed models outperform others in the numerical experiment.
引用
收藏
页数:11
相关论文
共 34 条
  • [31] Auto-encoder based dimensionality reduction
    Wang, Yasi
    Yao, Hongxun
    Zhao, Sicheng
    [J]. NEUROCOMPUTING, 2016, 184 : 232 - 242
  • [32] Xie JY, 2016, PR MACH LEARN RES, V48
  • [33] Oracle Based Active Set Algorithm for Scalable Elastic Net Subspace Clustering
    You, Chong
    Li, Chun-Guang
    Robinson, Daniel P.
    Vidal, Rene
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 3928 - 3937
  • [34] Zhao Deli, 2009, ADV NEURAL INFORM PR, P1953