Preprint
Machine Learning

SimCLRv2

June 1, 2020

0

Citations

0

Influential Citations

Venue

2020

Year

Abstract

A Semi-supervised learning framework which uses unsupervised pre training followed by supervised fine-tuning and distillation with unlabeled examples.

Analysis

Why This Paper Matters

SimCLRv2 extends the original SimCLR framework into a powerful semi-supervised learning pipeline. In an era where labeled data is scarce and expensive, the ability to leverage vast amounts of unlabeled data is crucial. This paper demonstrates a practical recipe: start with a large self-supervised pretrained model, fine-tune on a tiny labeled set, then distill into a smaller model using unlabeled data. The result is a model that approaches fully supervised performance with orders of magnitude fewer labels. This matters because it reduces the dependency on manual annotation, making deep learning more accessible for domains with limited labeled data.

Technical Contributions

The key innovations of SimCLRv2 include:

  • Unsupervised pretraining + supervised fine-tuning + distillation: A three-stage pipeline that maximizes the use of unlabeled data.
  • Task-specific pretraining: Unlike generic pretraining (e.g., on ImageNet), SimCLRv2 pretrains on the same unlabeled dataset as the target task, which yields better representations.
  • Bigger model for pretraining, smaller for deployment: Using a larger teacher model during pretraining and then distilling into a smaller student model improves accuracy without increasing inference cost.
  • Distillation with unlabeled data: The student learns from the teacher's soft labels on unlabeled examples, effectively transferring knowledge beyond the limited labeled set.

Results

SimCLRv2 achieves impressive results on ImageNet with only 1% of labels: 73.9% top-1 accuracy, compared to 48.4% for a supervised baseline trained on the same 1% labels. With 10% labels, it reaches 80.5% top-1 accuracy, close to the fully supervised ResNet-50 (76.5% top-1). The paper also shows that increasing the teacher model size (e.g., from ResNet-50 to ResNet-152) and then distilling into a ResNet-50 yields consistent gains. These results highlight the effectiveness of the distillation step in semi-supervised learning.

Significance

SimCLRv2 has had a lasting impact on semi-supervised and self-supervised learning. It popularized the idea of using self-supervised pretraining as a foundation for semi-supervised learning, a paradigm that has been adopted in many subsequent works (e.g., BYOL, SwAV, DINO). The paper also reinforced the importance of distillation in leveraging unlabeled data, influencing later methods like Noisy Student and self-training approaches. For practitioners, SimCLRv2 provides a clear, reproducible recipe for achieving high accuracy with minimal labels, which is valuable in real-world applications where annotation budgets are limited.