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

学习自监督学习中相似性损失与聚类损失之间的关系

Learning the Relation between Similarity Loss and Clustering Loss in Self-Supervised Learning

Nanjing University(南京大学);Southeast University(东南大学);Purple Mountain Laboratories(紫金山实验室);Amazon.com L.L.C.(亚马逊有限责任公司);Alibaba Group(阿里巴巴集团);The Hong Kong University of Science and Technology(香港科技大学);Southern Methodist University(南卫理公会大学)

Jidong Ge, Yuxiang Liu, Jie Gui, Lanting Fang, Ming Lin, James Tin-Yau Kwok, LiGuo Huang, Bin Luo

AI总结:

本文研究自监督学习中相似性损失与特征级交叉熵损失的关系,提出结合两幅不同图像的相似性信息,理论分析与实验证明两者的合适组合能取得最先进的结果。

摘要:

自监督学习使网络能够从海量数据本身学习判别性特征。大多数最先进的方法基于对比学习,最大化一张图像的两个增强视图之间的相似性。通过利用两个增强视图的一致性,可以免除人工标注的负担。对比学习利用实例级信息来学习鲁棒特征。然而,学习到的信息可能局限于同一实例的不同视图。在本文中,我们尝试利用两幅不同图像之间的相似性来提升自监督学习中的表示能力。与实例级信息相比,两幅不同图像之间的相似性可能提供更有用的信息。此外,我们分析了相似性损失与特征级交叉熵损失之间的关系。这两种损失对大多数深度学习方法至关重要。然而,这两种损失之间的关系尚不明确。相似性损失有助于获取实例级表示,而特征级交叉熵损失有助于挖掘两幅不同图像之间的相似性。我们提供了理论分析和实验,表明这两种损失的合适组合能够获得最先进的结果。代码可在 https://github.com/guijiejie/ICCL 获取。

英文摘要:

Self-supervised learning enables networks to learn discriminative features from massive data itself. Most state-of-the-art methods maximize the similarity between two augmentations of one image based on contrastive learning. By utilizing the consistency of two augmentations, the burden of manual annotations can be freed. Contrastive learning exploits instance-level information to learn robust features. However, the learned information is probably confined to different views of the same instance. In this paper, we attempt to leverage the similarity between two distinct images to boost representation in self-supervised learning. In contrast to instance-level information, the similarity between two distinct images may provide more useful information. Besides, we analyze the relation between similarity loss and feature-level cross-entropy loss. These two losses are essential for most deep learning methods. However, the relation between these two losses is not clear. Similarity loss helps obtain instance-level representation, while feature-level cross-entropy loss helps mine the similarity between two distinct images. We provide theoretical analyses and experiments to show that a suitable combination of these two losses can get state-of-the-art results. Code is available at https://github.com/guijiejie/ICCL.

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

↑