Geometry of the Loss Landscape in Overparameterized Neural Networks: Symmetries and Invariances
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
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,250
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
- 2019Linear Mode Connectivity and the Lottery Ticket Hypothesiscited by 627
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 585
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
- 2019Fine-Grained Analysis of Optimization and Generalization for Overparameterized Two-Layer Neural Networkscited by 562
- 2018On the Global Convergence of Gradient Descent for Over-parameterized Models using Optimal Transportcited by 422
- 2018Essentially No Barriers in Neural Network Energy Landscapecited by 328
- 2014Identifying and attacking the saddle point problem in high-dimensional non-convex optimizationcited by 286
17 more
- 2017How to Escape Saddle Points Efficientlycited by 284
- 2017Empirical Analysis of the Hessian of Over-Parametrized Neural Networkscited by 283
- 2020Deep learning versus kernel learning: an empirical study of loss landscape geometry and the time evolution of the Neural Tangent Kernelcited by 185
- 2020Finite Versus Infinite Neural Networks: an Empirical Studycited by 156
- 2016Topology and Geometry of Half-Rectified Network Optimizationcited by 102
- 2020Neural Mechanics: Symmetry and Broken Conservation Laws in Deep Learning Dynamicscited by 93
- 2017Gradient Descent Can Take Exponential Time to Escape Saddle Pointscited by 74
- 2019Large Scale Structure of Neural Network Loss Landscapescited by 68
- 2019Explaining Landscape Connectivity of Low-cost Solutions for Multilayer Netscited by 63
- 2018The jamming transition as a paradigm to understand the loss landscape of deep neural networkscited by 61
- 2019Weight-space symmetry in deep networks gives rise to permutation saddles, connected by equal-loss valleys across the loss landscapecited by 61
- 2019On Connected Sublevel Sets in Deep Learningcited by 51
- 2019Semi-flat minima and saddle points by embedding neural networks to overparameterizationcited by 29
- 2021Noether: The More Things Change, the More Stay the Samecited by 19
- 2014Explorations on high dimensional landscapescited by 8
- 2020The critical locus of overparameterized neural networkscited by 6
- 2020GENNI: Visualising the Geometry of Equivalences for Neural Network Identifiabilitycited by 6
Built on by
Citing papers per year: 2021: 4; 2022: 12; 2023: 21; 2024: 30; 2025: 27; 2026: 31.
125 papers.
Sort:
By area:
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 178
- 2022REPAIR: REnormalizing Permuted Activations for Interpolation Repaircited by 119
- 2024Model merging with SVD to tie the Knotscited by 118
- 2023Equivariant Architectures for Learning in Deep Weight Spacescited by 106
- 2022Improving Chest X-Ray Report Generation by Leveraging Warm Startingcited by 93
- 2023Deep Model Fusion: A Surveycited by 89
- 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 83
- 2022Mechanistic Mode Connectivitycited by 50
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 38
38 more
- 2021Embedding Principle of Loss Landscape of Deep Neural Networkscited by 36
- 2022Symmetries, flat minima, and the conserved quantities of gradient flowcited by 35
- 2024Training-Free Pretrained Model Mergingcited by 32
- 2025Symmetry in Neural Network Parameter Spacescited by 31
- 2024Approaching Deep Learning through the Spectral Dynamics of Weightscited by 29
- 2022Symmetry Teleportation for Accelerated Optimizationcited by 25
- 2023Symmetry Induces Structure and Constraint of Learningcited by 25
- 2022Random initialisations performing above chance and how to find themcited by 23
- 2023Improving Convergence and Generalization Using Parameter Symmetriescited by 23
- 2024The Real Tropical Geometry of Neural Networkscited by 23
- 2023Layer-wise Linear Mode Connectivitycited by 22
- 2021Embedding Principle: a hierarchical structure of loss landscape of deep neural networkscited by 21
- 2024Training Neural Networks from Scratch with Parallel Low-Rank Adapterscited by 17
- 2023Mildly Overparameterized ReLU Networks Have a Favorable Loss Landscapecited by 16
- 2024Merging Text Transformer Models from Different Initializationscited by 16
- 2025Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusioncited by 16
- 2023Loss Spike in Training Neural Networkscited by 15
- 2025Saddle-to-Saddle Dynamics Explains A Simplicity Bias Across Neural Network Architecturescited by 14
- 2025Deep Weight Factorization: Sparse Learning Through the Lens of Artificial Symmetriescited by 13
- 2025An overview of condensation phenomenon in deep learningcited by 13
- 2024Task Arithmetic Through The Lens Of One-Shot Federated Learningcited by 12
- 2024Coding schemes in neural networks learning classification taskscited by 11
- 2025Parameter Symmetry Potentially Unifies Deep Learning Theorycited by 11
- 2024On the Ability of Deep Networks to Learn Symmetries from Data: A Neural Kernel Theorycited by 10
- 2025Understanding Mode Connectivity via Parameter Space Symmetrycited by 10
- 2024Exploring Neural Network Landscapes: Star-Shaped and Geodesic Connectivitycited by 9
- 2024Simultaneous linear connectivity of neural networks modulo permutationcited by 9
- 2025The Butterfly Effect: Neural Network Training Trajectories Are Highly Sensitive to Initial Conditionscited by 9
- 2022Deep Double Descent via Smooth Interpolationcited by 8
- 2023Type-II Saddles and Probabilistic Stability of Stochastic Gradient Descentcited by 8
- 2023Should Under-parameterized Student Networks Copy or Average Teacher Weights?cited by 8
- 2023Disentangling Linear Mode-Connectivitycited by 8
- 2024Remove Symmetries to Control Model Expressivity and Improve Optimizationcited by 8
- 2024Ensembles provably learn equivariance through data augmentationcited by 8
- 2024Learning Gaussian Multi-Index Models with Gradient Flow: Time Complexity and Directional Convergencecited by 8
- 2025A Tale of Two Symmetries: Exploring the Loss Landscape of Equivariant Modelscited by 8
- 2022Embedding Principle in Depth for the Loss Landscape Analysis of Deep Neural Networkscited by 7
- 2022Linear Stability Hypothesis and Rank Stratification for Nonlinear Modelscited by 7
50 most cited of 125 All 125 (JSON)
- 2026Removing spurious minima for planar features by skip connections1 Oct 2026
- 2026A Polyphonic Conception of AI Understanding28 Sep 2026
- 2026On Parameter Symmetries and Conservation Laws in Gradient Flow28 Sep 2026
- 2026Teacher Geometry Shapes Learnability in Teacher-Student Networks9 Sep 2026
- 2026Percolation Dynamics in Optimization: Variance Cascades and Nested Symmetry2 Sep 2026
- 2026TAGTorch: A PyTorch Library for Geometry, Topology, and Symmetry-Aware Machine Learning30 Jul 2026
- 2026PAC--Bayes Bounds on Quotient Parameter Spaces: Geometry-induced Implicit-Bias Priors20 Jul 2026
- 2026Effects of width-dependent model hyperparameters and ℓ2-regularization on the loss landscape of two-layer ReLU networks18 Jul 2026
- 2026Gradient Flow Dynamics and Implicit Bias of Diagonal Linear Networks under Infinitesimal Initialization14 Jul 2026
- 2026A Bifurcation Theory Framework for Gradient Descent on the Edge of Stability14 Jun 2026
- 2026FlexPooling with Simple Auxiliary Classifiers in Deep Networks12 Jun 2026
- 2026Beyond Structural Symmetries: Linear Mode Connectivity via Neuron Identifiability3 Jun 2026
38 more
- 2026A Geometric Characterization of the Stationary Plateau for Two-Layer Neural Networks3 Jun 2026
- 2026Exploiting weight-space symmetries for approximating curvature30 May 2026
- 2026On the Epistemic Uncertainty of Overparametrized Neural Networks24 May 2026
- 2026Stimulus symmetries can confound representational similarity analyses20 May 2026
- 2026Most ReLU Networks Admit Identifiable Parameters5 May 2026
- 2026A Theory of Saddle Escape in Deep Nonlinear Networks2 May 2026
- 2026The Role of Symmetry in Optimizing Overparameterized Networks28 Apr 2026
- 2026Curvature-Aligned Probing for Local Loss-Landscape Stabilization16 Apr 2026
- 2026Sharp description of local minima in the loss landscape of high-dimensional two-layer ReLU neural networks10 Apr 2026
- 2026Escape dynamics and implicit bias of one-pass SGD in overparameterized quadratic networks3 Apr 2026
- 2026Plateaus, Optima, and Overfitting in Multi-Layer Perceptrons: A Saddle-Saddle-Attractor Scenario2 Apr 2026
- 2026The Geometric Cost of Normalization: Affine Bounds on the Bayesian Complexity of Neural Networks28 Mar 2026
- 2026Evidence of an Emergent "Self" in Continual Robot Learning25 Mar 2026
- 2026On the Interplay of Priors and Overparametrization in Bayesian Neural Network Posteriors23 Mar 2026
- 2026Resolving Interference (RI): Disentangling Models for Improved Model Merging13 Mar 2026
- 2026Towards Understanding Adam Convergence on Highly Degenerate Polynomials10 Mar 2026
- 2026The Mean is the Mirage: Entropy-Adaptive Model Merging under Heterogeneous Domain Shifts in Medical Imaging24 Feb 2026
- 2026From Approximation Rates to Loss-Landscape Barrier Decay in Shallow ReLU Networks19 Feb 2026
- 2026Visualizing the loss landscapes of physics-informed neural networks5 Feb 2026
- 2025Saddle-to-Saddle Dynamics Explains A Simplicity Bias Across Neural Network Architectures23 Dec 2025
- 2025WARP: Weight Teleportation for Attack-Resilient Unlearning Protocols29 Nov 2025
- 2025Fundamentals of Regression27 Nov 2025
- 2025Data Augmentation Techniques to Reverse-Engineer Neural Network Weights from Input-Output Queries25 Nov 2025
- 2025A mathematical theory for understanding when abstract representations emerge in neural networks10 Oct 2025
- 2025Do We Really Need Permutations? Impact of Model Width on Linear Mode Connectivity9 Oct 2025
- 2025Fractal basin geometry and the limits to predictability and reproducibility in deep learning7 Oct 2025
- 2025Saddle Hierarchy in Dense Associative Memory26 Aug 2025
- 2025Beyond Static Models: Hypernetworks for Adaptive and Generalizable Forecasting in Complex Parametric Dynamical Systems24 Jun 2025
- 2025Flat Channels to Infinity in Neural Loss Landscapes17 Jun 2025
- 2025Understanding Learning Invariance in Deep Linear Networks16 Jun 2025
- 2025The Butterfly Effect: Neural Network Training Trajectories Are Highly Sensitive to Initial Conditions16 Jun 2025
- 2025Symmetry in Neural Network Parameter Spaces16 Jun 2025
- 2025Circumventing Backdoor Space via Weight Symmetry9 Jun 2025
- 2025A Tale of Two Symmetries: Exploring the Loss Landscape of Equivariant Models2 Jun 2025
- 2025Understanding Mode Connectivity via Parameter Space Symmetry29 May 2025
- 2025Favorability of Loss Landscape with Weight Decay Requires Both Large Overparametrization and Initialization28 May 2025
- 2025Saddle-To-Saddle Dynamics in Deep ReLU Networks: Low-Rank Bias in the First Saddle Escape27 May 2025
- 2025Uncovering Critical Sets of Deep Neural Networks via Sample-Independent Critical Lifting19 May 2025
50 newest of 125 All 125 (JSON)
- 2022Git Re-Basin: Merging Models modulo Permutation Symmetriescited by 460
- 2021The Role of Permutation Invariance in Linear Mode Connectivity of Neural Networkscited by 285
- 2024Model Merging in LLMs, MLLMs, and Beyond: Methods, Theories, Applications and Opportunitiescited by 247
- 2023ZipIt! Merging Models from Different Tasks without Trainingcited by 178
- 2022REPAIR: REnormalizing Permuted Activations for Interpolation Repaircited by 119
- 2024Model merging with SVD to tie the Knotscited by 118
- 2023Equivariant Architectures for Learning in Deep Weight Spacescited by 106
- 2022Improving Chest X-Ray Report Generation by Leveraging Warm Startingcited by 93
- 2023Deep Model Fusion: A Surveycited by 89
- 2021Saddle-to-Saddle Dynamics in Deep Linear Networks: Small Initialization Training, Symmetry, and Sparsitycited by 83
- 2022Mechanistic Mode Connectivitycited by 50
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 38
12 most cited of 124 All 124 (JSON)
- 2026Removing spurious minima for planar features by skip connections1 Oct 2026
- 2026A Polyphonic Conception of AI Understanding28 Sep 2026
- 2026On Parameter Symmetries and Conservation Laws in Gradient Flow28 Sep 2026
- 2026Teacher Geometry Shapes Learnability in Teacher-Student Networks9 Sep 2026
- 2026Percolation Dynamics in Optimization: Variance Cascades and Nested Symmetry2 Sep 2026
- 2026TAGTorch: A PyTorch Library for Geometry, Topology, and Symmetry-Aware Machine Learning30 Jul 2026
- 2026PAC--Bayes Bounds on Quotient Parameter Spaces: Geometry-induced Implicit-Bias Priors20 Jul 2026
- 2026Effects of width-dependent model hyperparameters and ℓ2-regularization on the loss landscape of two-layer ReLU networks18 Jul 2026
- 2026Gradient Flow Dynamics and Implicit Bias of Diagonal Linear Networks under Infinitesimal Initialization14 Jul 2026
- 2026A Bifurcation Theory Framework for Gradient Descent on the Edge of Stability14 Jun 2026
- 2026FlexPooling with Simple Auxiliary Classifiers in Deep Networks12 Jun 2026
- 2026Beyond Structural Symmetries: Linear Mode Connectivity via Neuron Identifiability3 Jun 2026
12 newest of 124 All 124 (JSON)