基于对比学习的深度嵌入图像聚类算法OA
Deep Embedding Image Clustering Algorithm Based on Contrastive Learning
对比聚类是一种结合对比学习与聚类的无监督学习方法.该方法利用样本之间的相似性与差异性提取有用的特征表示,提升无监督学习任务中的聚类效果和模型的判别能力.然而,现有的对比聚类算法未能充分保留原始数据信息,导致嵌入空间中信息丢失,从而限制了特征表示的表达能力和效果.此外,现有算法更多关注于正负样本对的构建,缺乏针对聚类任务的优化设计,因此难以学习到真正适合于聚类任务的特征表示.针对这些问题,本文提出一种基于对比学习的深度嵌入图像聚类算法.该算法通过实例级和聚类级的对比学习,深入挖掘样本间的多层次关系,并有效提高了模型在聚类任务中的稳定性.同时,通过联合优化自编码器的重构损失和深度嵌入表示的约束,使模型能够保留更多的原始样本特征并增强嵌入表示的判别能力,确保嵌入空间中关键信息有效保留,从而提升聚类性能.实验结果表明,本文方法在5个公开数据集上的测试中展现出了优异性能,充分验证了该算法的有效性.
Contrastive clustering is an unsupervised learning method that combines contrastive learning with clustering.The method utilizes the similarities and differences between samples to extract useful feature representations,which improves the clustering effect and the discriminative ability of the model in unsupervised learning tasks.However,existing comparative clus-tering algorithms fail to fully retain the original data information,resulting in the loss of information in the embedding space,which limits the abillity of expression and effectiveness of the feature representation.In addition,the existing algorithms focus more on the construction of positive and negative sample pairs,and lack the optimized design for the clustering task,thus mak-ing it difficult to learn feature representations that are truly suitable for the clustering task.To address these issues,this paper proposes a deep embedded image clustering algorithm based on contrast learning.The algorithm digs deeply into the multilevel re-lationship between samples through instance-level and cluster-level contrast learning,and effectively improves the stability of the model in the clustering task.Meanwhile,by jointly optimizing the reconstruction loss of the self-encoder and the constraints of the deep embedding representation,the model is able to retain more original sample features and enhance the discriminative ability of the embedding representation to ensure that the key information in the embedding space is effectively retained,thus im-proving the clustering performance.Experimental results show that the method in this paper demonstrates excellent performance in tests on five public datasets,which fully validates the effectiveness of the algorithm.
周亮亮;王松;韩少伟;孙梦茹;李猛
西安工程大学计算机科学学院,陕西 西安 710048西安工程大学计算机科学学院,陕西 西安 710048西安工程大学计算机科学学院,陕西 西安 710048西安工程大学计算机科学学院,陕西 西安 710048西安工程大学计算机科学学院,陕西 西安 710048
信息技术与安全科学
对比学习深度嵌入重构损失图像聚类
contrastive learningdeep embeddingreconstruction lossimage clustering
《计算机与现代化》 2026 (3)
49-55,7
陕西省自然科学基础研究计划项目(2024JC-YBMS-473)陕西省教育厅重点科学研究计划项目(22JS019)
评论