atlas

Research map / Label-efficient and robust learning / Self-supervised and contrastive

You Don’t Need Domain-Specific Data Augmentations When Scaling Self-Supervised Learning

NeurIPS 2024 · Label-efficient and robust learning › Self-supervised and contrastive · approach: Data-efficient and robust self-supervision

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Shows that DINOv2 joint-embedding SSL can reach state-of-the-art results with only random cropping (no resizing, no photometric augmentations) when pretrained on sufficiently large data, and that augmentation mainly acts by increasing effective dataset size.

Problem

Joint-embedding architecture SSL methods are believed to require hand-crafted data augmentations for invariance learning and strong performance; the paper asks whether these augmentations are necessary at scale.

Approach

Case study on DINOv2 varying augmentation configurations (Original, Shared, Crop+Resize, Crop) across ImageNet-1k, ImageNet-22k and LVD-142M and model sizes; finds that with enough data, cropping without resizing plus masking matches augmented training, and that augmentation's main role is artificially increasing dataset size rather than enforcing invariance.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use RandomCrop without random resizing or photometric augmentations, with maskingDINOv2 ViT-L pretrained on ImageNet-22k · ImageNet-1k linear evaluationaccuracy: 84.0Mixed
Replace Original augmentations with Crop (ablation)DINOv2 ViT-L pretrained on ImageNet-22k · FireRisk, RESISC, MIMIC, CheXpert, VinDr, NIH-14accuracy / AUROC: 59.0 -> 60.9; 91.9 -> 92.3; 75.3 -> 75.5; 83.3 -> 83.0; 80.0 -> 82.5; 72.5 -> 74.2Mixed
Replace Original augmentations with Crop (ablation)DINOv2 ViT-L pretrained on ImageNet-22k · FlickrLogos32, GTSRB, RP2K, Products-10kaccuracy: 80.1 -> 76.9; 80.3 -> 85.1; 92.8 -> 93.4; 66.5 -> 67.3Mixed
Share photometric augmentations across views instead of applying them independently (ablation)DINOv2 ViT-L · ImageNet-1k linear evaluationaccuracy: -1.2%Hurt
Remove iBOT loss and masking (ablation)DINOv2 ViT-L, Crop setup · ImageNet-1k linear probingaccuracy: 1.0% with iBOT to 1.9% withoutHurt
Remove iBOT loss and masking (ablation)DINOv2 ViT-L, Crop setup · NYU-DepthRMSE: 0.04 to 0.015Hurt
Train longer on small ImageNet-1k without dataset-size-increasing augmentations (ablation)DINOv2 · ImageNet-1klinear-probe accuracyHurt
Use high-compute hyperparameters rather than low-compute hyperparameters with Crop or Crop+Resize (ablation)DINOv2 ViT-L trained on ImageNet-1k · ImageNet-1k linear probeaccuracy: 46% -> 73.5%Hurt
Replace Original with Crop+Resize using high-compute hyperparameters (ablation)DINOv2 ViT-L trained on ImageNet-1k · ImageNet-1k linear probeaccuracy: 29.1%Hurt
Scale model size while using Crop or Crop+Resize instead of Original (ablation)DINOv2 · ImageNet-1k linear evaluationaccuracy: -0.6% for ViT-S and ViT-B; -1.1% for ViT-L; -1.5% for ViT-HHurt
Increase dataset size when using Crop or Crop+Resize (ablation)DINOv2 · ImageNet-1k linear evaluationaccuracy: below 1%Helped

Surprising findings

Limitations

Experimental setup

Datasets: ImageNet-1k ImageNet-22k LVD-142M Places205 iNaturalist18 ImageNet-v2 ImageNet-Real ADE20k NYU-Depth v2 FlickrLogos32 GTSRB RP2K

Benchmarks: ImageNet-1k ImageNet-v2 ImageNet-Real Places205 iNaturalist18 ADE20k NYU-Depth v2 FlickrLogos32 GTSRB RP2K Products-10k FireRisk

Metrics: linear probe accuracy mIoU RMSE AUROC cosine similarity

Baselines: AIM BEiT MAE I-JEPA DINOv2 Original augmentation setup

Builds on: DINOv2 DINO iBOT I-JEPA SimCLR

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy