动量对比:用于无监督视觉表示学习

Momentum Contrast for Unsupervised Visual Representation Learning

罗斯·吉尔希克 Ross Girshick · Facebook AI Research (FAIR) · 2019-11-13 · arXiv:1911.05722 ↗ · 被引 15197

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

我们提出了用于无监督视觉表示学习的动量对比(MoCo)。从对比学习作为字典查找的角度出发,我们利用队列和移动平均编码器构建了一个动态字典。这使得能够即时构建一个大型且一致的字典,从而促进对比无监督学习。MoCo 在 ImageNet 分类的常见线性协议下提供了有竞争力的结果。更重要的是,MoCo 学习到的表示能够很好地迁移到下游任务中。在 PASCAL VOC、COCO 及其他数据集上的 7 个检测/分割任务中,MoCo 可以超越其监督预训练对应方法,有时甚至大幅领先。这表明在许多视觉任务中,无监督和监督表示学习之间的差距已基本被弥合。

We present Momentum Contrast (MoCo) for unsupervised visual representation learning. From a perspective on contrastive learning as dictionary look-up, we build a dynamic dictionary with a queue and a moving-averaged encoder. This enables building a large and consistent dictionary on-the-fly that facilitates contrastive unsupervised learning. MoCo provides competitive results under the common linear protocol on ImageNet classification. More importantly, the representations learned by MoCo transfer well to downstream tasks. MoCo can outperform its supervised pre-training counterpart in 7 detection/segmentation tasks on PASCAL VOC, COCO, and other datasets, sometimes surpassing it by large margins. This suggests that the gap between unsupervised and supervised representation learning has been largely closed in many vision tasks.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 19)

阅读逐段中英对照全文 →