mlsanityData to 7 Oct 2026

Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariances

arXiv 2105.12221 ↗First posted 25 May 2021cs.LGcited by 125

Not on the map. Why?

Abstract (arXiv)

We study how permutation symmetries in overparameterized multi-layer neural networks generate `symmetry-induced' critical points. Assuming a network with L layers of minimal widths r1^*, ldots, rL-1^* reaches a zero-loss minimum at r1^*! cdots rL-1^*! isolated points that are permutations of one another, we show that adding one extra neuron to each layer is sufficient to connect all these previously discrete minima into a single manifold. For a two-layer overparameterized network of width r^*+ h =: m we explicitly describe the manifold of global minima: it consists of T(r^*, m) affine subspaces of dimension at least h that are connected to one another. For a network of width m, we identify the number G(r,m) of affine subspaces containing only symmetry-induced critical points that are related to the critical points of a smaller network of width r<r^*. Via a combinatorial analysis, we derive closed-form formulas for T and G and show that the number of symmetry-induced critical subspaces dominates the number of affine subspaces forming the global minima manifold in the mildly overparameterized regime (small h ) and vice versa in the vastly overparameterized regime (h gg r^*). Our results provide new insights into the minimization of the non-convex loss function of overparameterized neural networks.

Key facts. Cited by 125 papers in the map (data through 7 Oct 2026). Builds on 29 of its 38 references that are in the map. First posted on arXiv 25 May 2021.

Builds on

29 of 38 references.

References · 29

  1. 2014Adam: A Method for Stochastic Optimizationcited by 36,960
  2. 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
  3. 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
  4. 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
  5. 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
  6. 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
  7. 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
  8. 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
  9. 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
  10. 2018On the Global Convergence of Gradient Descent for Over-parameterized Models using Optimal Transportcited by 422
  11. 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
  12. 2014Identifying and attacking the saddle point problem in high-dimensional non-convex optimizationcited by 286
17 more
  1. 2017How to Escape Saddle Points Efficientlycited by 284
  2. 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 283
  3. 2020Deep learning versus kernel learning: an empirical study of loss landscape geometry and the time evolution of the Neural Tangent Kernelcited by 185
  4. 2020Finite Versus Infinite Neural Networks: an Empirical Studycited by 156
  5. 2016Topology and Geometry of Half-Rectified Network Optimizationcited by 102
  6. 2020Neural Mechanics: Symmetry and Broken Conservation Laws in Deep Learning Dynamicscited by 93
  7. 2017Gradient Descent Can Take Exponential Time to Escape Saddle Pointscited by 74
  8. 2019Large Scale Structure of Neural Network Loss Landscapescited by 68
  9. 2019Explaining Landscape Connectivity of Low-cost Solutions for Multilayer Netscited by 63
  10. 2018The jamming transition as a paradigm to understand the loss landscape of deep neural networkscited by 61
  11. 2019Weight-space symmetry in deep networks gives rise to permutation saddles, connected by equal-loss valleys across the loss landscapecited by 61
  12. 2019On Connected Sublevel Sets in Deep Learningcited by 51
  13. 2019Semi-flat minima and saddle points by embedding neural networks to overparameterizationcited by 29
  14. 2021Noether: The More Things Change, the More Stay the Samecited by 19
  15. 2014Explorations on high dimensional landscapescited by 8
  16. 2020The critical locus of overparameterized neural networkscited by 6
  17. 2020GENNI: Visualising the Geometry of Equivalences for Neural Network Identifiabilitycited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2021: 4; 2022: 12; 2023: 21; 2024: 30; 2025: 27; 2026: 31.

125 papers.

Sort:

By area:

  1. 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
  2. 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
  3. 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
  4. 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 178
  5. 2022REPAIR: REnormalizing Permuted Activations for Interpolation Repaircited by 119
  6. 2024Model merging with SVD to tie the Knotscited by 118
  7. 2023Equivariant Architectures for Learning in Deep Weight Spacescited by 106
  8. 2022Improving Chest X-Ray Report Generation by Leveraging Warm Startingcited by 93
  9. 2023Deep Model Fusion: A Surveycited by 89
  10. 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 83
  11. 2022Mechanistic Mode Connectivitycited by 50
  12. 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 38
38 more
  1. 2021Embedding Principle of Loss Landscape of Deep Neural Networkscited by 36
  2. 2022Symmetries, flat minima, and the conserved quantities of gradient flowcited by 35
  3. 2024Training-Free Pretrained Model Mergingcited by 32
  4. 2025Symmetry in Neural Network Parameter Spacescited by 31
  5. 2024Approaching Deep Learning through the Spectral Dynamics of Weightscited by 29
  6. 2022Symmetry Teleportation for Accelerated Optimizationcited by 25
  7. 2023Symmetry Induces Structure and Constraint of Learningcited by 25
  8. 2022Random initialisations performing above chance and how to find themcited by 23
  9. 2023Improving Convergence and Generalization Using Parameter Symmetriescited by 23
  10. 2024The Real Tropical Geometry of Neural Networkscited by 23
  11. 2023Layer-wise Linear Mode Connectivitycited by 22
  12. 2021Embedding Principle: a hierarchical structure of loss landscape of deep neural networkscited by 21
  13. 2024Training Neural Networks from Scratch with Parallel Low-Rank Adapterscited by 17
  14. 2023Mildly Overparameterized ReLU Networks Have a Favorable Loss Landscapecited by 16
  15. 2024Merging Text Transformer Models from Different Initializationscited by 16
  16. 2025Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusioncited by 16
  17. 2023Loss Spike in Training Neural Networkscited by 15
  18. 2025Saddle-to-Saddle Dynamics Explains A Simplicity Bias Across Neural Network Architecturescited by 14
  19. 2025Deep Weight Factorization: Sparse Learning Through the Lens of Artificial Symmetriescited by 13
  20. 2025An overview of condensation phenomenon in deep learningcited by 13
  21. 2024Task Arithmetic Through The Lens Of One-Shot Federated Learningcited by 12
  22. 2024Coding schemes in neural networks learning classification taskscited by 11
  23. 2025Parameter Symmetry Potentially Unifies Deep Learning Theorycited by 11
  24. 2024On the Ability of Deep Networks to Learn Symmetries from Data: A Neural Kernel Theorycited by 10
  25. 2025Understanding Mode Connectivity via Parameter Space Symmetrycited by 10
  26. 2024Exploring Neural Network Landscapes: Star-Shaped and Geodesic Connectivitycited by 9
  27. 2024Simultaneous linear connectivity of neural networks modulo permutationcited by 9
  28. 2025The Butterfly Effect: Neural Network Training Trajectories Are Highly Sensitive to Initial Conditionscited by 9
  29. 2022Deep Double Descent via Smooth Interpolationcited by 8
  30. 2023Type-II Saddles and Probabilistic Stability of Stochastic Gradient Descentcited by 8
  31. 2023Should Under-parameterized Student Networks Copy or Average Teacher Weights?cited by 8
  32. 2023Disentangling Linear Mode-Connectivitycited by 8
  33. 2024Remove Symmetries to Control Model Expressivity and Improve Optimizationcited by 8
  34. 2024Ensembles provably learn equivariance through data augmentationcited by 8
  35. 2024Learning Gaussian Multi-Index Models with Gradient Flow: Time Complexity and Directional Convergencecited by 8
  36. 2025A Tale of Two Symmetries: Exploring the Loss Landscape of Equivariant Modelscited by 8
  37. 2022Embedding Principle in Depth for the Loss Landscape Analysis of Deep Neural Networkscited by 7
  38. 2022Linear Stability Hypothesis and Rank Stratification for Nonlinear Modelscited by 7

50 most cited of 125 All 125 (JSON)