vix.ing · top · new · best · stats

Improved Baselines with Momentum Contrastive Learning

2020/03/09 by Xinlei Chen, Haoqi Fan, Chen, Xinlei +5 · 413 citations
Computer Science · #Domain Adaptation and Few-Shot Learning #Multimodal Machine Learning Applications #Speech Recognition and Synthesis #cs.CV

paper · pdf · doi:10.48550/arxiv.2003.04297

Tech report, 2 pages + references

arxiv created 2020/03/09 · arxiv updated 2020/03/10

Abstract

Contrastive unsupervised learning has recently shown encouraging progress, e.g., in Momentum Contrast (MoCo) and SimCLR. In this note, we verify the effectiveness of two of SimCLR's design improvements by implementing them in the MoCo framework. With simple modifications to MoCo---namely, using an MLP projection head and more data augmentation---we establish stronger baselines that outperform SimCLR and do not require large training batches. We hope this will make state-of-the-art unsupervised learning research more accessible. Code will be made public.

Citations

Cited by

Related