Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

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

About

Self-Supervised learning (SSL) with Joint-Embedding Architectures (JEA) has led to outstanding performances. All instantiations of this paradigm were trained using strong and well-established hand-crafted data augmentations, leading to the general belief that they are required for the proper training and performance of such models. On the other hand, generative reconstruction-based models such as BEIT and MAE or Joint-Embedding Predictive Architectures such as I-JEPA have shown strong performance without using data augmentations except masking. In this work, we challenge the importance of invariance and data-augmentation in JEAs at scale. By running a case-study on a recent SSL foundation model - DINOv2 - we show that strong image representations can be obtained with JEAs and only cropping without resizing provided the training data is large enough, reaching state-of-the-art results and using the least amount of augmentation in the literature. Through this study, we also discuss the impact of compute constraints on the outcomes of experimental deep learning research, showing that they can lead to very different conclusions.

Th\'eo Moutakanni, Maxime Oquab, Marc Szafraniec, Maria Vakalopoulou, Piotr Bojanowski• 2024

Related benchmarks

TaskDatasetResultRank
Semantic segmentationADE20K
mIoU43.5
936
Image ClassificationiNaturalist 2018
Top-1 Accuracy75.1
287
Depth EstimationNYU Depth V2
RMSE0.411
177
Image ClassificationPlaces205
Top-1 Accuracy65.1
55
Image ClassificationImageNet 1k (v2)
Top-1 Accuracy74.8
21
Image ClassificationImageNet 1k (real)
Top-1 Accuracy88.6
9
Showing 6 of 6 rows

Other info

Code

Follow for update