What shapes the loss landscape of self-supervised learning?
Not on the map. Why?
Abstract (arXiv)
Prevention of complete and dimensional collapse of representations has recently become a design principle for self-supervised learning (SSL). However, questions remain in our theoretical understanding: When do those collapses occur? What are the mechanisms and causes? We answer these questions by deriving and thoroughly analyzing an analytically tractable theory of SSL loss landscapes. In this theory, we identify the causes of the dimensional collapse and study the effect of normalization and bias. Finally, we leverage the interpretability afforded by the analytical theory to understand how dimensional collapse can be beneficial and what affects the robustness of SSL against data imbalance.
Key facts. Cited by 19 papers in the map (data through 7 Oct 2026). Builds on 44 of its 51 references that are in the map. First posted on arXiv 2 Oct 2022.
Builds on
44 of 51 references.
References · 44
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 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
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,393
- 2020Big Self-Supervised Models are Strong Semi-Supervised Learnerscited by 1,375
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
32 more
- 2020What Makes for Good Views for Contrastive Learning?cited by 843
- 2019Measuring the Effects of Non-Identical Data Distribution for Federated Visual Classificationcited by 836
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 748
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2019A Theoretical Analysis of Contrastive Unsupervised Representation Learningcited by 530
- 2020Understanding the Behaviour of Contrastive Losscited by 397
- 2021Understanding Dimensional Collapse in Contrastive Self-supervised Learningcited by 374
- 2021Provable Guarantees for Self-Supervised Deep Learning with Spectral Contrastive Losscited by 311
- 2021Self-Supervised Learning with Data Augmentations Provably Isolates Content from Stylecited by 308
- 2020Whitening for Self-Supervised Representation Learningcited by 278
- 2021Understanding self-supervised Learning Dynamics without Contrastive Pairscited by 254
- 2016Deep Learning without Poor Local Minimacited by 233
- 2021Contrastive Learning Inverts the Data Generating Processcited by 220
- 2020Representation Learning via Invariant Causal Mechanismscited by 197
- 2020Theoretical Analysis of Self-Training with Deep Networks on Unlabeled Datacited by 179
- 2021On Feature Decorrelation in Self-Supervised Learningcited by 160
- 2020Contrastive learning, multi-view redundancy, and linear modelscited by 146
- 2021Self-supervised Learning is More Robust to Dataset Imbalancecited by 122
- 2019Don't Blame the ELBO! A Linear VAE Perspective on Posterior Collapsecited by 120
- 2021Can contrastive learning avoid shortcut solutions?cited by 114
- 2022Chaos is a Ladder: A New Theoretical Understanding of Contrastive Learning via Augmentation Overlapcited by 105
- 2022Understanding Contrastive Learning Requires Incorporating Inductive Biasescited by 96
- 2020Contrastive estimation reveals topic posterior information to linear modelscited by 54
- 2021Investigating the Role of Negatives in Contrastive Representation Learningcited by 54
- 2021Understanding Negative Samples in Instance Discriminative Self-supervised Representation Learningcited by 47
- 2021Self-supervised Representation Learning with Relative Predictive Codingcited by 27
- 2022Exact Solutions of a Deep Linear Networkcited by 24
- 2022Contrasting the landscape of contrastive and non-contrastive learningcited by 24
- 2022Posterior Collapse of a Linear Latent Variable Modelcited by 18
- 2022Toward a Geometrical Understanding of Self-supervised Contrastive Learningcited by 17
- 2021SGD with a Constant Large Learning Rate Can Converge to Local Maximacited by 8
- 2022Exact Phase Transitions in Deep Learningcited by 5
Built on by
Citing papers per year: 2023: 8; 2024: 8; 2025: 2; 2026: 1.
19 papers.
Sort:
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 38
- 2023On the Stepwise Nature of Self-Supervised Learningcited by 36
- 2025Symmetry in Neural Network Parameter Spacescited by 31
- 2023LiDAR: Sensing Linear Probing Performance in Joint Embedding SSL Architecturescited by 27
- 2024A Percolation Model of Emergence: Analyzing Transformers Trained on a Formal Languagecited by 26
- 2023Symmetry Induces Structure and Constraint of Learningcited by 25
- 2024Parameter Symmetry and Noise Equilibrium of Stochastic Gradient Descentcited by 19
- 2023Type-II Saddles and Probabilistic Stability of Stochastic Gradient Descentcited by 8
- 2024Remove Symmetries to Control Model Expressivity and Improve Optimizationcited by 8
- 2023Hard-Negative Sampling for Contrastive Learning: Optimal Representation Geometry and Neural- vs Dimensional-Collapsecited by 6
- 2024Infinite Width Limits of Self Supervised Neural Networkscited by 4
- 2023Representation Learning Dynamics of Self-Supervised Modelscited by 3
7 more
- 2025Dual Perspectives on Non-Contrastive Self-Supervised Learningcited by 3
- 2023Feature Normalization Prevents Collapse of Non-contrastive Learning Dynamicscited by 2
- 2024PhiNets: Brain-inspired Non-contrastive Learning Based on Temporal Prediction Hypothesiscited by 2
- 2024On Learning Discriminative Features from Synthesized Data for Self-Supervised Fine-Grained Visual Recognitioncited by 2
- 2024When can we Approximate Wide Contrastive Models with Neural Tangent Kernels and Principal Component Analysis?cited by 1
- 2024Training Dynamics of Nonlinear Contrastive Learning Model in the High Dimensional Limitcited by 1
- 2026Asymptotic and Finite-Time Guarantees for Langevin-Based Temperature Annealing in InfoNCEcited by 1
- 2026Asymptotic and Finite-Time Guarantees for Langevin-Based Temperature Annealing in InfoNCE13 Mar 2026
- 2025Dual Perspectives on Non-Contrastive Self-Supervised Learning18 Jun 2025
- 2025Symmetry in Neural Network Parameter Spaces16 Jun 2025
- 2024Infinite Width Limits of Self Supervised Neural Networks17 Nov 2024
- 2024Remove Symmetries to Control Model Expressivity and Improve Optimization28 Aug 2024
- 2024A Percolation Model of Emergence: Analyzing Transformers Trained on a Formal Language22 Aug 2024
- 2024On Learning Discriminative Features from Synthesized Data for Self-Supervised Fine-Grained Visual Recognition19 Jul 2024
- 2024Training Dynamics of Nonlinear Contrastive Learning Model in the High Dimensional Limit11 Jun 2024
- 2024PhiNets: Brain-inspired Non-contrastive Learning Based on Temporal Prediction Hypothesis23 May 2024
- 2024When can we Approximate Wide Contrastive Models with Neural Tangent Kernels and Principal Component Analysis?13 Mar 2024
- 2024Parameter Symmetry and Noise Equilibrium of Stochastic Gradient Descent11 Feb 2024
- 2023LiDAR: Sensing Linear Probing Performance in Joint Embedding SSL Architectures7 Dec 2023
7 more
- 2023Hard-Negative Sampling for Contrastive Learning: Optimal Representation Geometry and Neural- vs Dimensional-Collapse9 Nov 2023
- 2023Symmetry Induces Structure and Constraint of Learning29 Sep 2023
- 2023Feature Normalization Prevents Collapse of Non-contrastive Learning Dynamics28 Sep 2023
- 2023Representation Learning Dynamics of Self-Supervised Models5 Sep 2023
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworks7 Jun 2023
- 2023On the Stepwise Nature of Self-Supervised Learning27 Mar 2023
- 2023Type-II Saddles and Probabilistic Stability of Stochastic Gradient Descent23 Mar 2023
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 38
- 2023On the Stepwise Nature of Self-Supervised Learningcited by 36
- 2025Symmetry in Neural Network Parameter Spacescited by 31
- 2023LiDAR: Sensing Linear Probing Performance in Joint Embedding SSL Architecturescited by 27
- 2024A Percolation Model of Emergence: Analyzing Transformers Trained on a Formal Languagecited by 26
- 2023Symmetry Induces Structure and Constraint of Learningcited by 25
- 2024Parameter Symmetry and Noise Equilibrium of Stochastic Gradient Descentcited by 19
- 2023Type-II Saddles and Probabilistic Stability of Stochastic Gradient Descentcited by 8
- 2024Remove Symmetries to Control Model Expressivity and Improve Optimizationcited by 8
- 2023Hard-Negative Sampling for Contrastive Learning: Optimal Representation Geometry and Neural- vs Dimensional-Collapsecited by 6
- 2024Infinite Width Limits of Self Supervised Neural Networkscited by 4
- 2023Representation Learning Dynamics of Self-Supervised Modelscited by 3
12 most cited of 19 All 19 (JSON)
- 2026Asymptotic and Finite-Time Guarantees for Langevin-Based Temperature Annealing in InfoNCE13 Mar 2026
- 2025Dual Perspectives on Non-Contrastive Self-Supervised Learning18 Jun 2025
- 2025Symmetry in Neural Network Parameter Spaces16 Jun 2025
- 2024Infinite Width Limits of Self Supervised Neural Networks17 Nov 2024
- 2024Remove Symmetries to Control Model Expressivity and Improve Optimization28 Aug 2024
- 2024A Percolation Model of Emergence: Analyzing Transformers Trained on a Formal Language22 Aug 2024
- 2024On Learning Discriminative Features from Synthesized Data for Self-Supervised Fine-Grained Visual Recognition19 Jul 2024
- 2024Training Dynamics of Nonlinear Contrastive Learning Model in the High Dimensional Limit11 Jun 2024
- 2024PhiNets: Brain-inspired Non-contrastive Learning Based on Temporal Prediction Hypothesis23 May 2024
- 2024When can we Approximate Wide Contrastive Models with Neural Tangent Kernels and Principal Component Analysis?13 Mar 2024
- 2024Parameter Symmetry and Noise Equilibrium of Stochastic Gradient Descent11 Feb 2024
- 2023LiDAR: Sensing Linear Probing Performance in Joint Embedding SSL Architectures7 Dec 2023
12 newest of 19 All 19 (JSON)