arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.04891cs.CVcs.LG

用于增强显著特征的聚类友好表示学习

Clustering-friendly Representation Learning for Enhancing Salient Features

Corporate R&D Center, Toshiba(东芝公司研发中心)

Toshiyuki Oshima, Kentaro Takagi, Kouta Nakata

AI总结:

本文面向无监督图像聚类,扩展聚类友好对比学习并在损失中引入基于参考数据集的对比分析,以增强聚类关键特征,在三个数据集上均优于现有方法。

摘要:

近年来,采用对比学习算法的表示学习已成功应用于具有挑战性的无标注数据集。然而,在简单的无监督设置下,这些方法无法区分重要特征与不重要特征,而且重要性的定义会随下游任务或分析目标的类型而变化,例如识别物体或背景。本文聚焦于将无监督图像聚类作为下游任务,提出一种增强对聚类任务至关重要特征的表示学习方法。我们扩展了一种聚类友好的对比学习方法,并将对比分析方法纳入损失函数设计;该方法利用参考数据集,将重要特征与不重要特征分离开来。我们在三个具有特征性背景的数据集上开展图像聚类实验评估,结果表明,与传统对比分析方法和深度聚类方法相比,我们的方法在所有数据集上均取得了更高的聚类分数。

英文摘要:

Recently, representation learning with contrastive learning algorithms has been successfully applied to challenging unlabeled datasets. However, these methods are unable to distinguish important features from unimportant ones under simply unsupervised settings, and definitions of importance vary according to the type of downstream task or analysis goal, such as the identification of objects or backgrounds. In this paper, we focus on unsupervised image clustering as the downstream task and propose a representation learning method that enhances features critical to the clustering task. We extend a clustering-friendly contrastive learning method and incorporate a contrastive analysis approach, which utilizes a reference dataset to separate important features from unimportant ones, into the design of loss functions. Conducting an experimental evaluation of image clustering for three datasets with characteristic backgrounds, we show that for all datasets, our method achieves higher clustering scores compared with conventional contrastive analysis and deep clustering methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑