3 个月前

动量对比在无监督视觉表征学习中的应用

动量对比在无监督视觉表征学习中的应用

摘要

我们提出了动量对比(Momentum Contrast, MoCo)方法,用于无监督视觉表征学习。从对比学习作为字典查询的视角出发,我们构建了一个动态字典,该字典由一个队列和一个动量更新的编码器组成。这一设计使得系统能够实时构建一个规模庞大且一致的字典,从而有效支持对比式的无监督学习。在ImageNet分类任务中,MoCo采用常见的线性评估协议,取得了具有竞争力的性能表现。更重要的是,MoCo所学习到的表征在下游任务中具有良好的迁移能力。在PASCAL VOC、COCO及其他数据集上的7项检测与分割任务中,MoCo的表现优于其监督预训练的对应方法,有时甚至显著超越。这一结果表明,在众多视觉任务中,无监督与监督表征学习之间的差距已基本弥合。

代码仓库

ernie1/pi-nas
pytorch
GitHub 中提及
1170500804/MoCo
pytorch
GitHub 中提及
AidenDurrant/MoCo-Pytorch
pytorch
GitHub 中提及
lightly-ai/lightly
pytorch
GitHub 中提及
HobbitLong/CMC
pytorch
GitHub 中提及
szq0214/CMC_with_Image_Mixture
pytorch
GitHub 中提及
linwenye/light-moco
pytorch
GitHub 中提及
ehsanik/muscleTorch
pytorch
GitHub 中提及
giakoumoglou/synco
pytorch
GitHub 中提及
Westlake-AI/openmixup
pytorch
GitHub 中提及
YyzHarry/imbalanced-semi-self
pytorch
GitHub 中提及
smartdanny/MoCoV2_CIFAR10
pytorch
GitHub 中提及
facebookresearch/vissl
pytorch
GitHub 中提及
PaperCodeReview/MoCo-TF
tf
GitHub 中提及
trannhiem/ma_ssrl_pytorch
pytorch
GitHub 中提及
zhangyifei01/EIR
pytorch
GitHub 中提及
xiangli13/MoCo
tf
GitHub 中提及
kikacaty/adv_guide
pytorch
GitHub 中提及
CoinCheung/denseCL
pytorch
GitHub 中提及
facebookresearch/moco
官方
pytorch
GitHub 中提及
leftthomas/moco
pytorch
GitHub 中提及
tranc012/SMILE-Rep
pytorch
GitHub 中提及
alihassanijr/DISCERN
pytorch
GitHub 中提及
davidsvy/hard-negative-mixing
pytorch
GitHub 中提及
joshr17/IFM
pytorch
GitHub 中提及
bl0/moco
pytorch
GitHub 中提及
srinidhiPY/SSL_CR_Histo
pytorch
GitHub 中提及
Evgeneus/MoCo
pytorch
GitHub 中提及
AndrewTal/MoCo-Pytorch
pytorch
GitHub 中提及
ali-nsua/DISCERN
pytorch
GitHub 中提及
open-mmlab/mmselfsup
pytorch
GitHub 中提及
ppwwyyxx/moco.tensorflow
官方
tf
GitHub 中提及

基准测试

基准方法指标
contrastive-learning-on-imagenet-1kResNet50
ImageNet Top-1 Accuracy: 60.6
image-classification-on-omnibenchmarkMoCoV2
Average Top-1 Accuracy: 34.8
self-supervised-image-classification-onMoCo (ResNet-50 4x)
Number of Params: 375M
Top 1 Accuracy: 68.6%
self-supervised-image-classification-onMoCo (ResNet-50)
Number of Params: 24M
Top 1 Accuracy: 60.6%
self-supervised-image-classification-onMoCo (ResNet-50 2x)
Number of Params: 94M
Top 1 Accuracy: 65.4%
self-supervised-image-classification-on-1MoCo (Resnet-50)
Top 1 Accuracy: 77.0%
self-supervised-image-classification-on-1MoCo (Resnet-50)
Top 1 Accuracy: 77.3%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
动量对比在无监督视觉表征学习中的应用 | 论文 | HyperAI超神经