On Pretraining Data Diversity for Self-Supervised Learning
Not on the map. Why?
Abstract (arXiv)
We explore the impact of training with more diverse datasets, characterized by the number of unique samples, on the performance of self-supervised learning (SSL) under a fixed computational budget. Our findings consistently demonstrate that increasing pretraining data diversity enhances SSL performance, albeit only when the distribution distance to the downstream data is minimal. Notably, even with an exceptionally large pretraining data diversity achieved through methods like web crawling or diffusion-generated data, among other ways, the distribution shift remains a challenge. Our experiments are comprehensive with seven SSL methods using large-scale datasets such as ImageNet and YFCC100M amounting to over 200 GPU days. Code and trained models are available at https://github.com/hammoudhasan/DiversitySSL
Key facts. Cited by 7 papers in the map (data through 7 Oct 2026). Builds on 60 of its 73 references that are in the map. First posted on arXiv 20 Mar 2024.
Builds on
60 of 73 references.
References · 60
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,694
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
38 more
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2022Emergent Abilities of Large Language Modelscited by 2,483
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,360
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,282
- 2018Unsupervised Feature Learning via Non-Parametric Instance-level Discriminationcited by 1,838
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,610
- 2018Unsupervised Representation Learning by Predicting Image Rotationscited by 1,586
- 2016Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzlescited by 1,379
- 2020Big Self-Supervised Models are Strong Semi-Supervised Learnerscited by 1,375
- 2019Contrastive Multiview Codingcited by 1,297
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,265
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,147
- 2016Colorful Image Colorizationcited by 1,113
- 2015Unsupervised Visual Representation Learning by Context Predictioncited by 1,062
- 2018Deep Clustering for Unsupervised Learning of Visual Featurescited by 1,020
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
- 2021Scaling Vision Transformerscited by 993
- 2020Designing Network Design Spacescited by 954
- 2017Revisiting Unreasonable Effectiveness of Data in Deep Learning Eracited by 916
- 2019Self-Supervised Learning of Pretext-Invariant Representationscited by 806
- 2019Learning Representations by Maximizing Mutual Information Across Viewscited by 733
- 2023Scaling Vision Transformers to 22 Billion Parameterscited by 690
- 2019Data-Efficient Image Recognition with Contrastive Predictive Codingcited by 672
- 2020Prototypical Contrastive Learning of Unsupervised Representationscited by 636
- 2023A Cookbook of Self-Supervised Learningcited by 248
- 2020How Well Do Self-Supervised Models Transfer?cited by 213
- 2019Local Aggregation for Unsupervised Learning of Visual Embeddingscited by 204
- 2019Scaling and Benchmarking Self-Supervised Visual Representation Learningcited by 201
- 2017Multi-task Self-Supervised Visual Learningcited by 200
- 2021Self-supervised Pretraining of Visual Features in the Wildcited by 197
- 2021Benchmarking Representation Learning for Natural World Image Collectionscited by 167
- 2021Are Large-scale Datasets Necessary for Self-Supervised Pre-training?cited by 117
- 2021Solo-learn: A Library of Self-supervised Methods for Visual Representation Learningcited by 116
- 2020What makes instance discrimination good for transfer learning?cited by 111
- 2021When Does Contrastive Visual Representation Learning Work?cited by 92
- 2021Divide and Contrast: Self-supervised Learning from Uncurated Datacited by 85
- 2022Vision Models Are More Robust And Fair When Pretrained On Uncurated Images Without Supervisioncited by 80
50 most cited of 60 All 60 (JSON)
Built on by
Citing papers per year: 2024: 2; 2025: 3; 2026: 2.
7 papers.
- 2026THOR: A Versatile Foundation Model for Earth Observation Climate and Society Applicationscited by 6
- 2024S3PT: Scene Semantics and Structure Guided Clustering to Boost Self-Supervised Pre-Training for Autonomous Drivingcited by 4
- 2025Towards Reliable Identification of Diffusion-based Image Manipulationscited by 4
- 2024Precision at Scale: Domain-Specific Datasets On-Demandcited by 1
- 2025On the Importance of Pretraining Data Alignment for Atomic Property Predictioncited by 0
- 2025Understanding Ice Crystal Habit Diversity with Self-Supervised Learningcited by 0
- 2026Parameter-Efficient Self-Supervised Adaptation for EEG-FM under Fixed Computational Budgetscited by 0
- 2026Parameter-Efficient Self-Supervised Adaptation for EEG-FM under Fixed Computational Budgets25 Aug 2026
- 2026THOR: A Versatile Foundation Model for Earth Observation Climate and Society Applications22 Jan 2026
- 2025Understanding Ice Crystal Habit Diversity with Self-Supervised Learning9 Sep 2025
- 2025Towards Reliable Identification of Diffusion-based Image Manipulations5 Jun 2025
- 2025On the Importance of Pretraining Data Alignment for Atomic Property Prediction16 Feb 2025
- 2024S3PT: Scene Semantics and Structure Guided Clustering to Boost Self-Supervised Pre-Training for Autonomous Driving30 Oct 2024
- 2024Precision at Scale: Domain-Specific Datasets On-Demand3 Jul 2024