arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.17142cs.CV
收藏

软近邻是对比视觉表示学习的积极支持者

Soft Neighbors are Positive Supporters in Contrastive Visual Representation Learning

The University of Hong Kong(香港大学);The Chinese University of Hong Kong(香港中文大学);Tencent AI Lab(腾讯人工智能实验室);AI Institute, Fudan University(复旦大学人工智能研究院)

Chongjian Ge, Jiangliu Wang, Zhan Tong, Shoufa Chen, Yibing Song, Ping Luo

摘要:

对比学习方法通过将来自一个实例的视图与来自其他实例的视图进行比较来训练视觉编码器。通常,从一个实例创建的视图设置为正,而从其他实例创建的视图设置为负。这种二进制实例判别被广泛研究,以改善自监督学习中的特征表示。在本文中,我们重新思考的实例判别框架,发现二进制实例标记不足以衡量不同样本之间的相关性。对于直观的示例,给定随机图像实例,在小批量中可能存在其内容含义相同的其他图像(即,属于同一类别)或部分相关(即,属于同类)。如何处理与当前图像实例类似地相关的图像留下了一个未探索的问题。因此,我们提出通过探索其他相关实例来支持当前图像(即,软邻居)。我们首先仔细培养一个候选邻居集,这将进一步用于探索高度相关的实例。然后引入交叉注意模块来预测其他相关实例相对于当前实例的相关性得分(表示为正性)。积极性得分定量地测量来自每个相关实例的积极支持,并被编码到用于借口训练的目标中。为此,我们提出的方法的好处,在区分不相关的实例,同时吸收相关的SSL实例。我们评估我们的软邻居的对比学习方法(SNPLESS)的标准视觉识别基准,包括图像分类,目标检测,和实例分割。最先进的识别性能表明,SNARY在改善ViT和CNN编码器的特征表示方面是有效的。

英文摘要:

Contrastive learning methods train visual encoders by comparing views from one instance to others. Typically, the views created from one instance are set as positive, while views from other instances are negative. This binary instance discrimination is studied extensively to improve feature representations in self-supervised learning. In this paper, we rethink the instance discrimination framework and find the binary instance labeling insufficient to measure correlations between different samples. For an intuitive example, given a random image instance, there may exist other images in a mini-batch whose content meanings are the same (i.e., belonging to the same category) or partially related (i.e., belonging to a similar category). How to treat the images that correlate similarly to the current image instance leaves an unexplored problem. We thus propose to support the current image by exploring other correlated instances (i.e., soft neighbors). We first carefully cultivate a candidate neighbor set, which will be further utilized to explore the highly-correlated instances. A cross-attention module is then introduced to predict the correlation score (denoted as positiveness) of other correlated instances with respect to the current one. The positiveness score quantitatively measures the positive support from each correlated instance, and is encoded into the objective for pretext training. To this end, our proposed method benefits in discriminating uncorrelated instances while absorbing correlated instances for SSL. We evaluate our soft neighbor contrastive learning method (SNCLR) on standard visual recognition benchmarks, including image classification, object detection, and instance segmentation. The state-of-the-art recognition performance shows that SNCLR is effective in improving feature representations from both ViT and CNN encoders.

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

↑