
摘要
我们提出了动量对比(Momentum Contrast, MoCo)方法,用于无监督视觉表征学习。从对比学习作为字典查询的视角出发,我们构建了一个动态字典,该字典由一个队列和一个动量更新的编码器组成。这一设计使得系统能够实时构建一个规模庞大且一致的字典,从而有效支持对比式的无监督学习。在ImageNet分类任务中,MoCo采用常见的线性评估协议,取得了具有竞争力的性能表现。更重要的是,MoCo所学习到的表征在下游任务中具有良好的迁移能力。在PASCAL VOC、COCO及其他数据集上的7项检测与分割任务中,MoCo的表现优于其监督预训练的对应方法,有时甚至显著超越。这一结果表明,在众多视觉任务中,无监督与监督表征学习之间的差距已基本弥合。
代码仓库
ernie1/pi-nas
pytorch
GitHub 中提及
1170500804/MoCo
pytorch
GitHub 中提及
Pangoraw/GeoSSL
pytorch
giakou4/pyssl
pytorch
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 中提及
beresandras/contrastive-classification-keras
tf
GitHub 中提及
ehsanik/muscleTorch
pytorch
GitHub 中提及
giakoumoglou/synco
pytorch
GitHub 中提及
Westlake-AI/openmixup
pytorch
GitHub 中提及
yangyucheng000/MoCo
mindspore
HobbitLong/PyContrast
pytorch
YyzHarry/imbalanced-semi-self
pytorch
GitHub 中提及
smartdanny/MoCoV2_CIFAR10
pytorch
GitHub 中提及
open-mmlab/mmdetection
pytorch
facebookresearch/vissl
pytorch
GitHub 中提及
PaperCodeReview/MoCo-TF
tf
GitHub 中提及
trannhiem/ma_ssrl_pytorch
pytorch
GitHub 中提及
zhangyifei01/EIR
pytorch
GitHub 中提及
alibaba/EasyCV
pytorch
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 中提及
amazon-research/exponential-moving-average-normalization
pytorch
GitHub 中提及
alihassanijr/DISCERN
pytorch
GitHub 中提及
davidsvy/hard-negative-mixing
pytorch
GitHub 中提及
KevinMusgrave/pytorch-metric-learning
pytorch
GitHub 中提及
joshr17/IFM
pytorch
GitHub 中提及
bl0/moco
pytorch
GitHub 中提及
KevinMusgrave/pytorch_metric_learning
pytorch
GitHub 中提及
srinidhiPY/SSL_CR_Histo
pytorch
GitHub 中提及
Evgeneus/MoCo
pytorch
GitHub 中提及
AndrewTal/MoCo-Pytorch
pytorch
GitHub 中提及
ali-nsua/DISCERN
pytorch
GitHub 中提及
LeiLiu-byte/the-Essence-of-LT-Rocognition
pytorch
GitHub 中提及
open-mmlab/mmselfsup
pytorch
GitHub 中提及
ppwwyyxx/moco.tensorflow
官方
tf
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| contrastive-learning-on-imagenet-1k | ResNet50 | ImageNet Top-1 Accuracy: 60.6 |
| image-classification-on-omnibenchmark | MoCoV2 | Average Top-1 Accuracy: 34.8 |
| self-supervised-image-classification-on | MoCo (ResNet-50 4x) | Number of Params: 375M Top 1 Accuracy: 68.6% |
| self-supervised-image-classification-on | MoCo (ResNet-50) | Number of Params: 24M Top 1 Accuracy: 60.6% |
| self-supervised-image-classification-on | MoCo (ResNet-50 2x) | Number of Params: 94M Top 1 Accuracy: 65.4% |
| self-supervised-image-classification-on-1 | MoCo (Resnet-50) | Top 1 Accuracy: 77.0% |
| self-supervised-image-classification-on-1 | MoCo (Resnet-50) | Top 1 Accuracy: 77.3% |