
摘要
对比无监督学习近期取得了令人鼓舞的进展,例如在动量对比(Momentum Contrast, MoCo)和SimCLR中。在这篇笔记中,我们通过在MoCo框架中实现SimCLR的两项设计改进来验证其有效性。通过对MoCo进行简单的修改——即使用多层感知机(MLP)投影头和更多的数据增强技术——我们建立了更强的基线模型,这些模型不仅超越了SimCLR,而且不需要大规模的训练批次。我们希望这能使最先进水平的无监督学习研究更加易于获取。代码将公开发布。
代码仓库
1170500804/MoCo
pytorch
GitHub 中提及
facebookresearch/simsiam
pytorch
GitHub 中提及
Pangoraw/GeoSSL
pytorch
GitHub 中提及
cjrd/self-supervised-pretraining
pytorch
GitHub 中提及
AidenDurrant/MoCo-Pytorch
pytorch
GitHub 中提及
lightly-ai/lightly
pytorch
linwenye/light-moco
pytorch
GitHub 中提及
derrickwang005/mosrep
pytorch
GitHub 中提及
lavoiems/SADt
pytorch
GitHub 中提及
Westlake-AI/openmixup
pytorch
GitHub 中提及
technioncomputationalmrilab/colorectal_cancer_ai
pytorch
GitHub 中提及
HobbitLong/PyContrast
pytorch
vturrisi/solo-learn
pytorch
GitHub 中提及
lavoiems/SPUDT
pytorch
GitHub 中提及
smartdanny/MoCoV2_CIFAR10
pytorch
GitHub 中提及
open-mmlab/mmdetection
pytorch
danielgordon10/vince
pytorch
GitHub 中提及
PaperCodeReview/MoCo-TF
tf
GitHub 中提及
alibaba/EasyCV
pytorch
kikacaty/adv_guide
pytorch
GitHub 中提及
aheuillet/nasiam
pytorch
GitHub 中提及
lavoiems/Cats-UDT
pytorch
GitHub 中提及
CoinCheung/denseCL
pytorch
GitHub 中提及
robml/Deep-Learning-SSL-MoCo-SP21
pytorch
GitHub 中提及
facebookresearch/moco
官方
pytorch
GitHub 中提及
szq0214/S2-BNN
pytorch
GitHub 中提及
a07458666/simsiamfacenet
pytorch
GitHub 中提及
Ghiara/diva
pytorch
GitHub 中提及
ClaudiaShu/SSL-FER
pytorch
GitHub 中提及
TRAILab/ST-SLidR
pytorch
GitHub 中提及
Evgeneus/MoCo
pytorch
GitHub 中提及
Seonghoon-Yu/MoCov2_Pytorch_tutorial
pytorch
GitHub 中提及
AndrewTal/MoCo-Pytorch
pytorch
GitHub 中提及
ppwwyyxx/moco.tensorflow
tf
GitHub 中提及
open-mmlab/mmselfsup
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| contrastive-learning-on-imagenet-1k | ResNet50 | ImageNet Top-1 Accuracy: 71.1 |
| image-classification-on-places205 | MoCo v2 | Top 1 Accuracy: 52.9 |
| person-re-identification-on-sysu-30k | MoCo v2 (self-supervised) | Rank-1: 11.6 |
| self-supervised-image-classification-on | MoCo v2 (ResNet-50) | Number of Params: 24M Top 1 Accuracy: 71.1% Top 5 Accuracy: 90.1% |
| self-supervised-person-re-identification-on | MoCo v2 | Rank-1: 11.6 |