Improved Baselines with Momentum Contrastive Learning
Shows that adding an MLP head and stronger augmentation to MoCo yields improved contrastive learning baselines that surpass SimCLR.
Contrastive unsupervised learning has progressed with methods like MoCo and SimCLR. This note verifies two of SimCLR's design improvements by implementing them in the MoCo framework: an MLP projection head and stronger data augmentation. With these simple modifications, the authors establish stronger baselines that outperform SimCLR while not requiring large training batches, aiming to make state-of-the-art unsupervised learning research more accessible. Code will be released.
Based on: Improved Baselines with Momentum Contrastive Learning · arXiv.org
Curated by Aramai Editorial
Read summary →