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

面向数据高效无监督表示学习的多粒度对比方法

Multi-Grained Contrast for Data-Efficient Unsupervised Representation Learning

Central South University(中南大学)

Chengchao Shen, Jianzhong Chen, Jianxin Wang

AI总结:

针对单粒度对比学习迁移性不足的问题,本文提出MGC,通过正视图间的多粒度对应关系进行对比,在无大规模预训练条件下提升多项下游视觉任务性能。

摘要:

现有对比学习方法主要聚焦于单粒度表示学习,例如部件级、对象级或场景级表示,因此不可避免地忽视了表示在其他粒度层级上的可迁移性。本文旨在学习多粒度表示,使其能够在不同粒度层级上有效描述图像,从而提升在广泛下游任务上的泛化能力。为此,我们提出了一种新颖的用于无监督表示学习的多粒度对比方法(Multi-Grained Contrast,MGC)。具体而言,我们在正样本视图之间构建精细的多粒度对应关系,随后利用这些对应关系执行多粒度对比,以学习更通用的无监督表示。在未使用大规模数据集进行预训练的情况下,我们的方法在目标检测、实例分割、场景解析、语义分割和关键点检测等广泛下游任务上显著优于现有最先进方法。此外,实验结果支持了本文方法的数据高效特性和出色的表示可迁移性。源代码和训练权重可在 https://github.com/visresearch/mgc 获取。

英文摘要:

The existing contrastive learning methods mainly focus on single-grained representation learning, e.g., part-level, object-level or scene-level ones, thus inevitably neglecting the transferability of representations on other granularity levels. In this paper, we aim to learn multi-grained representations, which can effectively describe the image on various granularity levels, thus improving generalization on extensive downstream tasks. To this end, we propose a novel Multi-Grained Contrast method (MGC) for unsupervised representation learning. Specifically, we construct delicate multi-grained correspondences between positive views and then conduct multi-grained contrast by the correspondences to learn more general unsupervised representations. Without pretrained on large-scale dataset, our method significantly outperforms the existing state-of-the-art methods on extensive downstream tasks, including object detection, instance segmentation, scene parsing, semantic segmentation and keypoint detection. Moreover, experimental results support the data-efficient property and excellent representation transferability of our method. The source code and trained weights are available at \url{https://github.com/visresearch/mgc}.

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

↑