嵌入混合自编码器的无监督图像聚类算法

UNSUPERVISED IMAGE CLUSTERING ALGORITHM EMBEDDED IN HYBRID AUTOENCODER

  • 摘要: 自编码器的网络结构简单,难以捕捉复杂图像的局部特征,显著降低图像的聚类效果。针对这种情况,提出嵌入混合自编码器的无监督图像聚类算法(Unsupervised image Clustering algorithm embedded in Hybrid-Autoencoder,UCHAE)。算法首先分别用卷积自编码器和自编码器提取图像特征进行自适应融合,然后通过正交变换矩阵将融合特征转换为新的特征空间,最后根据目标损失函数,不断优化聚类结果。将UCHAE算法应用于MNIST、FashionMNIST、COIL-20和COIL-100图像数据集上,聚类准确率分别达到96.5%、65.8%、74.5%和72.0%。实验结果表明,该算法能有效地提高图像的聚类效果。

     

    Abstract: Due to the simple network structure of the autoencoder, it is difficult to capture the local features of complex images, which significantly reduces the clustering effect of images. To address this situation, unsupervised image clustering algorithm embedded in hybrid-autoencoder (UCHAE) is proposed. The algorithm extracted image features by convolutional autoencoder and autoencoder respectively for adaptive fusion, converted the fused features into a new feature space by orthogonal transformation matrix, and continuously optimized the clustering results according to the target loss function. The UCHAE algorithm was applied to image datasets, namely MNIST, FashionMNIST, COIL-20 and COIL-100, and the clustering accuracy reached 96.5%, 65.8%, 74.5% and 72.0%, respectively. The experimental results show that the algorithm can effectively improve the clustering effect of images.

     

/

返回文章
返回