基于对比学习的超多类深度图像聚类模型

作     者：胡深 钱宇华 王婕婷 李飞江 吕维 HU Shen;QIAN Yuhua;WANG Jieting;LI Feijiang;LYU Wei

作者机构：山西大学计算机与信息技术学院太原030006 山西大学计算智能与中文信息处理教育部重点实验室太原030006 山西大学大数据科学与产业研究院太原030006 

基　　金：国家自然科学基金重点项目(62136005) 国家重点研发计划(2021ZD0112400) 国家自然科学基金青年科学基金(62106132) 山西省三晋学者项目资助 山西省基础研究计划(20210302124271,202103021223026) 

出 版 物：《计算机科学》 (Computer Science)

年 卷 期：2023年第50卷第9期

页      码：192-201页

摘      要：图像聚类通过表征学习对图像数据降维并提取有效特征而后进行聚类分析。当图像数据存在超多类别时,数据分布的复杂性和类簇的密集性严重影响了现有方法的实用性。为此,提出了基于对比学习的超多类深度图像聚类模型,主要分为3个阶段:首先,改进对比学习方法训练特征模型以使类簇分布均匀;其次,基于语义相似性原则多视角挖掘实例语义最近邻信息;最后,将实例及其最近邻作为自监督信息训练聚类模型。根据实验类型的不同,设计了消融实验和对比实验。在消融实验中,证明了所提方法使类簇均匀分布在映射空间,并可靠挖掘语义最近邻信息。在对比实验中,将其与先进算法在7个基准数据集上进行了比较,在ImageNet-200类数据集上,其准确率比目前先进方法提升了10.6%;在ImageNet-1000类数据集上,其准确率比目前先进算法提升了9.2%。

主 题 词：超多类聚类 对比学习 特征模型 语义相似性 图像聚类 

学科分类：081203[081203] 08[工学] 0835[0835] 0812[工学-测绘类] 

D　O　I：10.11896/jsjkx.220900133

馆 藏 号：203123328...