Solo-learn: A Library of Self-supervised Methods for Visual Representation Learning
Not on the map. Why?
Abstract (arXiv)
This paper presents solo-learn, a library of self-supervised methods for visual representation learning. Implemented in Python, using Pytorch and Pytorch lightning, the library fits both research and industry needs by featuring distributed training pipelines with mixed-precision, faster data loading via Nvidia DALI, online linear evaluation for better prototyping, and many additional training tricks. Our goal is to provide an easy-to-use library comprising a large amount of Self-supervised Learning (SSL) methods, that can be easily extended and fine-tuned by the community. solo-learn opens up avenues for exploiting large-budget SSL solutions on inexpensive smaller infrastructures and seeks to democratize SSL by making it accessible to all. The source code is available at https://github.com/vturrisi/solo-learn.
Key facts. Cited by 116 papers in the map (data through 7 Oct 2026). Builds on 16 of its 20 references that are in the map. First posted on arXiv 3 Aug 2021.
Builds on
16 of 20 references.
References · 16
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,082
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,893
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2020Exploring Simple Siamese Representation Learningcited by 2,853
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2018UMAP: Uniform Manifold Approximation and Projection for Dimension Reductioncited by 2,841
- 2020Supervised Contrastive Learningcited by 2,803
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,282
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
4 more
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
- 2021With a Little Help from My Friends: Nearest-Neighbor Contrastive Learning of Visual Representationscited by 324
- 2020Whitening for Self-Supervised Representation Learningcited by 278
- 2021ReSSL: Relational Self-Supervised Learning with Weak Augmentationcited by 83
Built on by
Citing papers per year: 2021: 1; 2022: 28; 2023: 29; 2024: 27; 2025: 22; 2026: 9.
116 papers.
Sort:
By area:
- 2021Self-Supervised Models are Continual Learnerscited by 149
- 2022Adversarial Masking for Self-Supervised Learningcited by 78
- 2022How Does SimSiam Avoid Collapse Without Negative Samples? A Unified Understanding with Self-supervised Contrastive Learningcited by 65
- 2022Dual Temperature Helps Contrastive Learning Without Many Negative Samples: Towards Understanding and Simplifying MoCocited by 52
- 2022Spectral Feature Augmentation for Graph Contrastive Learning and Beyondcited by 49
- 2022Neural Manifold Clustering and Embeddingcited by 44
- 2023Is ImageNet worth 1 video? Learning strong image encoders from 1 long unlabelled videocited by 40
- 2023Know Your Self-supervised Learning: A Survey on Image-based Generative and Discriminative Trainingcited by 39
- 2022Self-Supervised Learning with an Information Maximization Criterioncited by 38
- 2025Joint Embedding vs Reconstruction: Provable Benefits of Latent Space Prediction for Self Supervised Learningcited by 33
- 2023Towards a Unified Theoretical Understanding of Non-contrastive Learning via Rank Differential Mechanismcited by 32
- 2023Identifying Interpretable Subspaces in Image Representationscited by 32
38 more
- 2023Downstream-agnostic Adversarial Examplescited by 32
- 2023Learning Efficient Coding of Natural Images with Maximum Manifold Capacity Representationscited by 31
- 2023Group Robust Classification Without Any Group Informationcited by 25
- 2024Do Generated Data Always Help Contrastive Learning?cited by 25
- 2022How Robust is Unsupervised Representation Learning to Distribution Shift?cited by 24
- 2022Self-Supervised Pretraining for 2D Medical Image Segmentationcited by 24
- 2022Simplicial Embeddings in Self-Supervised Learning and Downstream Classificationcited by 23
- 2022Decoupled Adversarial Contrastive Learning for Self-supervised Adversarial Robustnesscited by 20
- 2023Rethinking the Effect of Data Augmentation in Adversarial Contrastive Learningcited by 20
- 2023Semi-supervised learning made simple with self-supervised clusteringcited by 20
- 2023EMP-SSL: Towards Self-Supervised Learning in One Training Epochcited by 18
- 2023Class Incremental Learning with Self-Supervised Pre-Training and Prototype Learningcited by 18
- 2022Is Self-Supervised Learning More Robust Than Supervised Learning?cited by 17
- 2022Unsupervised visualization of image datasets using contrastive learningcited by 16
- 2022Implicit variance regularization in non-contrastive SSLcited by 15
- 2023All4One: Symbiotic Neighbour Contrastive Learning via Self-Attention and Redundancy Reductioncited by 11
- 2022On the Pros and Cons of Momentum Encoder in Self-Supervised Visual Representation Learningcited by 10
- 2023PILLAR: How to make semi-private learning more effectivecited by 10
- 2024Making Better Use of Unlabelled Data in Bayesian Active Learningcited by 10
- 2022Bag of Image Patch Embedding Behind the Success of Self-Supervised Learningcited by 9
- 2023Modulate Your Spectrum in Self-Supervised Learningcited by 9
- 2024Preventing Dimensional Collapse in Self-Supervised Learning via Orthogonality Regularizationcited by 9
- 2024Learning predictable and robust neural representations by straightening image sequencescited by 9
- 2025stable-pretraining-v1: Foundation Model Research Made Simplecited by 9
- 2023Learning Symbolic Representations Through Joint GEnerative and DIscriminative Trainingcited by 8
- 2023Systematic comparison of semi-supervised and self-supervised learning for medical image classificationcited by 8
- 2024Occam's Razor for Self Supervised Learning: What is Sufficient to Learn Good Representations?cited by 8
- 2025Clustering Properties of Self-Supervised Learningcited by 8
- 2026Rectified LpJEPA: Joint-Embedding Predictive Architectures with Sparse and Maximum-Entropy Representationscited by 8
- 2022Unsupervised Visual Representation Learning via Mutual Information Regularized Assignmentcited by 7
- 2022Self-Supervised Visual Representation Learning via Residual Momentumcited by 7
- 2024On Pretraining Data Diversity for Self-Supervised Learningcited by 7
- 2024Collapse-Proof Non-Contrastive Self-Supervised Learningcited by 7
- 2024HEX: Hierarchical Emergence Exploitation in Self-Supervised Algorithmscited by 7
- 2022Learning Task-Independent Game State Representations from Unlabeled Imagescited by 6
- 2022Towards Efficient and Effective Self-Supervised Learning of Visual Representationscited by 6
- 2022GEDI: GEnerative and DIscriminative Training for Self-Supervised Learningcited by 6
- 2023To Stay or Not to Stay in the Pre-train Basin: Insights on Ensembling in Transfer Learningcited by 6
50 most cited of 116 All 116 (JSON)
- 2026I Have a Stream: Making Self-Supervised Learning Work on Continuous Video30 Sep 2026
- 2026Geometry-Aware Graph Construction via Adaptive Spectral Bandwidth Control3 Sep 2026
- 2026FML-bench: A Controlled Study of AI Research Agent Strategies from the Perspective of Search Dynamics17 May 2026
- 2026Learning to Perceive "Where": Spatial Pretext Tasks for Robust Self-Supervised Learning11 May 2026
- 2026When Losses Align: Gradient-Based Composite Loss Weighting for Efficient Pretraining8 May 2026
- 2026Understanding Self-Supervised Learning via Latent Distribution Matching5 May 2026
- 2026Leveraging Complementary Embeddings for Replay Selection in Continual Learning with Small Buffers9 Apr 2026
- 2026Temporal Slowness in Central Vision Drives Semantic Object Learning4 Feb 2026
- 2026Rectified LpJEPA: Joint-Embedding Predictive Architectures with Sparse and Maximum-Entropy Representations1 Feb 2026
- 2025stable-pretraining-v1: Foundation Model Research Made Simple23 Nov 2025
- 2025Contrastive Predictive Coding Done Right for Mutual Information Estimation29 Oct 2025
- 2025T-REGS: Minimum Spanning Tree Regularization for Self-Supervised Learning27 Oct 2025
38 more
- 2025Revisiting Orbital Minimization Method for Neural Operator Decomposition24 Oct 2025
- 2025DReS: Dual Reconstruction Smoothing for Functional Regularization30 Sep 2025
- 2025Interpretable Self-Supervised Learning via Representer Landmarks and Nystr"om Approximation29 Sep 2025
- 2025Boost Self-Supervised Dataset Distillation via Parameterization, Predefined Augmentation, and Approximation29 Jul 2025
- 2025Cluster Contrast for Unsupervised Visual Representation Learning16 Jul 2025
- 2025Reducing Variability of Multiple Instance Learning Methods for Digital Pathology30 Jun 2025
- 2025Contrastive Self-Supervised Learning As Neural Manifold Packing16 Jun 2025
- 2025On the Similarities of Embeddings in Contrastive Learning11 Jun 2025
- 2025Collaborative Unlabeled Data Optimization20 May 2025
- 2025FlowPure: Continuous Normalizing Flows for Adversarial Purification19 May 2025
- 2025AdaDim: Dimensionality Adaptation for SSL Representational Dynamics18 May 2025
- 2025Joint Embedding vs Reconstruction: Provable Benefits of Latent Space Prediction for Self Supervised Learning18 May 2025
- 2025Conditioning Residuals for Diffusion Models via Representation Feedback16 May 2025
- 2025Self-Supervision Enhances Instance-based Multiple Instance Learning Methods in Digital Pathology: A Benchmark Study2 May 2025
- 2025Variational Self-Supervised Learning6 Apr 2025
- 2025Open-source framework for detecting bias and overfitting for large pathology images3 Mar 2025
- 2025Deep Clustering via Probabilistic Ratio-Cut Optimization5 Feb 2025
- 2025Clustering Properties of Self-Supervised Learning30 Jan 2025
- 2025Human Gaze Boosts Object-Centered Representation Learning6 Jan 2025
- 2024GUESS: Generative Uncertainty Ensemble for Self Supervision3 Dec 2024
- 2024MAGMA: Manifold Regularization for MAEs3 Dec 2024
- 2024Rethinking Self-Supervised Learning Within the Framework of Partial Information Decomposition3 Dec 2024
- 2024Direct Coloring for Self-Supervised Enhanced Feature Decoupling3 Dec 2024
- 2024Learning predictable and robust neural representations by straightening image sequences4 Nov 2024
- 2024Preventing Dimensional Collapse in Self-Supervised Learning via Orthogonality Regularization1 Nov 2024
- 2024HEX: Hierarchical Emergence Exploitation in Self-Supervised Algorithms30 Oct 2024
- 2024SigCLR: Sigmoid Contrastive Learning of Visual Representations22 Oct 2024
- 2024SynFER: Towards Boosting Facial Expression Recognition with Synthetic Data13 Oct 2024
- 2024Collapse-Proof Non-Contrastive Self-Supervised Learning7 Oct 2024
- 2024Exploiting Supervised Poison Vulnerability to Strengthen Self-Supervised Defense13 Sep 2024
- 2024Hierarchical Multi-Label Classification with Missing Information for Benthic Habitat Imagery10 Sep 2024
- 2024Explicit Mutual Information Maximization for Self-Supervised Learning7 Sep 2024
- 2024Self-Supervised Learning for Identifying Defects in Sewer Footage2 Sep 2024
- 2024Contrastive Learning with Synthetic Positives30 Aug 2024
- 2024An accurate detection is not all you need to combat label noise in web-noisy datasets8 Jul 2024
- 2024DCoM: Active Learning for All Learners1 Jul 2024
- 2024Occam's Razor for Self Supervised Learning: What is Sufficient to Learn Good Representations?15 Jun 2024
- 2024Efficiency for Free: Ideal Data Are Transportable Representations23 May 2024
50 newest of 116 All 116 (JSON)
- 2025FlowPure: Continuous Normalizing Flows for Adversarial Purification19 May 2025
- 2025Conditioning Residuals for Diffusion Models via Representation Feedback16 May 2025
- 2024Contrastive Learning with Synthetic Positives30 Aug 2024
- 2024Do Generated Data Always Help Contrastive Learning?19 Mar 2024
- 2021Self-Supervised Models are Continual Learnerscited by 149
- 2022Adversarial Masking for Self-Supervised Learningcited by 78
- 2022How Does SimSiam Avoid Collapse Without Negative Samples? A Unified Understanding with Self-supervised Contrastive Learningcited by 65
- 2022Dual Temperature Helps Contrastive Learning Without Many Negative Samples: Towards Understanding and Simplifying MoCocited by 52
- 2022Spectral Feature Augmentation for Graph Contrastive Learning and Beyondcited by 49
- 2022Neural Manifold Clustering and Embeddingcited by 44
- 2023Is ImageNet worth 1 video? Learning strong image encoders from 1 long unlabelled videocited by 40
- 2023Know Your Self-supervised Learning: A Survey on Image-based Generative and Discriminative Trainingcited by 39
- 2022Self-Supervised Learning with an Information Maximization Criterioncited by 38
- 2025Joint Embedding vs Reconstruction: Provable Benefits of Latent Space Prediction for Self Supervised Learningcited by 33
- 2023Towards a Unified Theoretical Understanding of Non-contrastive Learning via Rank Differential Mechanismcited by 32
- 2023Identifying Interpretable Subspaces in Image Representationscited by 32
12 most cited of 110 All 110 (JSON)
- 2026I Have a Stream: Making Self-Supervised Learning Work on Continuous Video30 Sep 2026
- 2026Geometry-Aware Graph Construction via Adaptive Spectral Bandwidth Control3 Sep 2026
- 2026Learning to Perceive "Where": Spatial Pretext Tasks for Robust Self-Supervised Learning11 May 2026
- 2026When Losses Align: Gradient-Based Composite Loss Weighting for Efficient Pretraining8 May 2026
- 2026Understanding Self-Supervised Learning via Latent Distribution Matching5 May 2026
- 2026Leveraging Complementary Embeddings for Replay Selection in Continual Learning with Small Buffers9 Apr 2026
- 2026Temporal Slowness in Central Vision Drives Semantic Object Learning4 Feb 2026
- 2026Rectified LpJEPA: Joint-Embedding Predictive Architectures with Sparse and Maximum-Entropy Representations1 Feb 2026
- 2025stable-pretraining-v1: Foundation Model Research Made Simple23 Nov 2025
- 2025Contrastive Predictive Coding Done Right for Mutual Information Estimation29 Oct 2025
- 2025T-REGS: Minimum Spanning Tree Regularization for Self-Supervised Learning27 Oct 2025
- 2025Revisiting Orbital Minimization Method for Neural Operator Decomposition24 Oct 2025
12 newest of 110 All 110 (JSON)