4 个月前

动量对比学习改进基线

动量对比学习改进基线

摘要

对比无监督学习近期取得了令人鼓舞的进展,例如在动量对比(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 中提及
linwenye/light-moco
pytorch
GitHub 中提及
derrickwang005/mosrep
pytorch
GitHub 中提及
lavoiems/SADt
pytorch
GitHub 中提及
Westlake-AI/openmixup
pytorch
GitHub 中提及
vturrisi/solo-learn
pytorch
GitHub 中提及
lavoiems/SPUDT
pytorch
GitHub 中提及
smartdanny/MoCoV2_CIFAR10
pytorch
GitHub 中提及
danielgordon10/vince
pytorch
GitHub 中提及
PaperCodeReview/MoCo-TF
tf
GitHub 中提及
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 中提及
AndrewTal/MoCo-Pytorch
pytorch
GitHub 中提及
ppwwyyxx/moco.tensorflow
tf
GitHub 中提及
open-mmlab/mmselfsup
pytorch
GitHub 中提及

基准测试

基准方法指标
contrastive-learning-on-imagenet-1kResNet50
ImageNet Top-1 Accuracy: 71.1
image-classification-on-places205MoCo v2
Top 1 Accuracy: 52.9
person-re-identification-on-sysu-30kMoCo v2 (self-supervised)
Rank-1: 11.6
self-supervised-image-classification-onMoCo v2 (ResNet-50)
Number of Params: 24M
Top 1 Accuracy: 71.1%
Top 5 Accuracy: 90.1%
self-supervised-person-re-identification-onMoCo v2
Rank-1: 11.6

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
动量对比学习改进基线 | 论文 | HyperAI超神经