arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.04867cs.CV
收藏

无需类先验的多级对比学习细粒度表示学习

Fine-Grained Representation Learning via Multi-Level Contrastive Learning without Class Priors

School of computer science and information engineering, Fujian Agriculture and Forestry University(福建农林大学计算机与信息学院)

Houwang Jiang, Zhuxian Liu, Guodong Liu, Xiaolong Liu, Shihua Zhan

AI总结:

本文提出无需类先验的 Contrastive Disentangling (CD) 框架,通过结合实例级、特征级对比损失与归一化熵损失的多级对比学习策略,学习细粒度表示,在类别信息不可用或模糊时优于现有方法。

摘要:

无监督表示学习的最新进展通常依赖于已知类别数量来改进特征提取和聚类。然而,这一假设引出了一个重要问题:类别数量总是必要的吗,类标签能否完全捕捉数据中的细粒度特征?在本文中,我们提出了 Contrastive Disentangling (CD),这是一个旨在不依赖类先验学习表示的框架。CD 利用多级对比学习策略,将实例级和特征级对比损失与归一化熵损失相结合,以捕捉语义丰富且细粒度的表示。具体而言,(1) 实例级对比损失分离样本间的特征表示;(2) 特征级对比损失促进特征头之间的独立性;(3) 归一化熵损失确保特征多样性并防止特征坍塌。在 CIFAR-10、CIFAR-100、STL-10 和 ImageNet-10 上的大量实验表明,CD 在类别信息不可用或模糊的场景中优于现有方法。代码可在 https://github.com/Hoper-J/Contrastive-Disentangling 获取。

英文摘要:

Recent advances in unsupervised representation learning often rely on knowing the number of classes to improve feature extraction and clustering. However, this assumption raises an important question: is the number of classes always necessary, and do class labels fully capture the fine-grained features within the data? In this paper, we propose Contrastive Disentangling (CD), a framework designed to learn representations without relying on class priors. CD leverages a multi-level contrastive learning strategy, integrating instance-level and feature-level contrastive losses with a normalized entropy loss to capture semantically rich and fine-grained representations. Specifically, (1) the instance-level contrastive loss separates feature representations across samples; (2) the feature-level contrastive loss promotes independence among feature heads; and (3) the normalized entropy loss ensures feature diversity and prevents feature collapse. Extensive experiments on CIFAR-10, CIFAR-100, STL-10, and ImageNet-10 demonstrate that CD outperforms existing methods in scenarios where class information is unavailable or ambiguous. The code is available at https://github.com/Hoper-J/Contrastive-Disentangling.

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

↑