The Asymmetric Maximum Margin Bias of Quasi-Homogeneous Neural Networks
Not on the map. Why?
Abstract (arXiv)
In this work, we explore the maximum-margin bias of quasi-homogeneous neural networks trained with gradient flow on an exponential loss and past a point of separability. We introduce the class of quasi-homogeneous models, which is expressive enough to describe nearly all neural networks with homogeneous activations, even those with biases, residual connections, and normalization layers, while structured enough to enable geometric analysis of its gradient dynamics. Using this analysis, we generalize the existing results of maximum-margin bias for homogeneous networks to this richer class of models. We find that gradient flow implicitly favors a subset of the parameters, unlike in the case of a homogeneous model where all parameters are treated equally. We demonstrate through simple examples how this strong favoritism toward minimizing an asymmetric norm can degrade the robustness of quasi-homogeneous models. On the other hand, we conjecture that this norm-minimization discards, when possible, unnecessary higher-order parameters, reducing the model to a sparser parameterization. Lastly, by applying our theorem to sufficiently expressive neural networks with normalization layers, we reveal a universal mechanism behind the empirical phenomenon of Neural Collapse.
Key facts. Cited by 25 papers in the map (data through 14 Oct 2026). Builds on 28 of its 45 references that are in the map. First posted on arXiv 7 Oct 2022.
Builds on
28 of 45 references.
References · 28
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
- 2020Kernel and Rich Regimes in Overparametrized Modelscited by 306
- 2020Implicit Bias of Gradient Descent for Wide Two-layer Neural Networks Trained with the Logistic Losscited by 268
- 2018Implicit Bias of Gradient Descent on Linear Convolutional Networkscited by 251
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
- 2018Gradient descent aligns the layers of deep linear networkscited by 182
- 2020Directional convergence and alignment in deep learningcited by 178
- 2021Exploring Deep Neural Networks via Layer-Peeled Model: Minority Collapse in Imbalanced Trainingcited by 166
- 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 158
16 more
- 2020Neural collapse with unconstrained featurescited by 142
- 2018Regularization Matters: Generalization and Optimization of Neural Nets v.s. their Induced Kernelcited by 131
- 2018Stochastic subgradient method converges on tame functionscited by 124
- 2018Convergence of Gradient Descent on Separable Datacited by 110
- 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
- 2020Neural Mechanics: Symmetry and Broken Conservation Laws in Deep Learning Dynamicscited by 96
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2021Gradient Descent on Two-layer Nets: Margin Maximization and Simplicity Biascited by 83
- 2020A Unifying View on Implicit Bias in Training Linear Neural Networkscited by 75
- 2020Implicit Bias in Deep Linear Classification: Initialization Scale vs Training Accuracycited by 71
- 2018Stochastic Gradient Descent on Separable Data: Exact Convergence with a Fixed Learning Ratecited by 64
- 2019Lexicographic and Depth-Sensitive Margins in Homogeneous and Non-Homogeneous Deep Modelscited by 53
- 2020On the emergence of simplex symmetry in the final and penultimate layers of neural network classifierscited by 52
- 2021On Margin Maximization in Linear and ReLU Networkscited by 34
- 2022Training invariances and the low-rank phenomenon: beyond linear networkscited by 33
- 2018When Will Gradient Methods Converge to Max-margin Classifier under ReLU Models?cited by 2
Built on by
Citing papers per year: 2023: 9; 2024: 9; 2025: 4; 2026: 3.
25 papers.
Sort:
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 49
- 2025Grokking at the Edge of Numerical Stabilitycited by 46
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 39
- 2023Benign Overfitting in Linear Classifiers and Leaky ReLU Networks from KKT Conditions for Margin Maximizationcited by 29
- 2023The Double-Edged Sword of Implicit Bias: Generalization vs. Robustness in ReLU Networkscited by 23
- 2024Average gradient outer product as a mechanism for deep neural collapsecited by 22
- 2023The Law of Parsimony in Gradient Descent for Learning Deep Linear Networkscited by 20
- 2024Flavors of Margin: Implicit Bias of Steepest Descent in Homogeneous Neural Networkscited by 18
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 17
13 more
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimizationcited by 17
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 16
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularizationcited by 14
- 2024Provable Emergence of Deep Neural Collapse and Low-Rank Bias in L2-Regularized Nonlinear Networkscited by 11
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networkscited by 11
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 10
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescalingcited by 4
- 2023Generator Born from Classifiercited by 4
- 2026The Impact of Anisotropic Covariance Structure on the Training Dynamics and Generalization Error of Linear Networkscited by 2
- 2025The late-stage training dynamics of (stochastic) subgradient descent on homogeneous neural networkscited by 1
- 2023Hierarchical Simplicity Bias of Neural Networkscited by 0
- 2025Embedding principle of homogeneous neural network for classification problemcited by 0
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networkscited by 0
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networks18 Feb 2026
- 2026The Impact of Anisotropic Covariance Structure on the Training Dynamics and Generalization Error of Linear Networks11 Jan 2026
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025Embedding principle of homogeneous neural network for classification problem18 May 2025
- 2025The late-stage training dynamics of (stochastic) subgradient descent on homogeneous neural networks8 Feb 2025
- 2025Grokking at the Edge of Numerical Stability8 Jan 2025
- 2024Flavors of Margin: Implicit Bias of Steepest Descent in Homogeneous Neural Networks29 Oct 2024
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networks22 Sep 2024
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization12 Jun 2024
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learning10 Jun 2024
13 more
- 2024Improving Generalization and Convergence by Enhancing Implicit Regularization31 May 2024
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?23 May 2024
- 2024Average gradient outer product as a mechanism for deep neural collapse21 Feb 2024
- 2024Provable Emergence of Deep Neural Collapse and Low-Rank Bias in L2-Regularized Nonlinear Networks6 Feb 2024
- 2023Generator Born from Classifier5 Dec 2023
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokking30 Nov 2023
- 2023Achieving Margin Maximization Exponentially Fast via Progressive Norm Rescaling24 Nov 2023
- 2023Hierarchical Simplicity Bias of Neural Networks5 Nov 2023
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworks7 Jun 2023
- 2023The Law of Parsimony in Gradient Descent for Learning Deep Linear Networks1 Jun 2023
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Model22 May 2023
- 2023Benign Overfitting in Linear Classifiers and Leaky ReLU Networks from KKT Conditions for Margin Maximization2 Mar 2023
- 2023The Double-Edged Sword of Implicit Bias: Generalization vs. Robustness in ReLU Networks2 Mar 2023
- 2023Dichotomy of Early and Late Phase Implicit Biases Can Provably Induce Grokkingcited by 80
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 49
- 2025Grokking at the Edge of Numerical Stabilitycited by 46
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
- 2023Stochastic Collapse: How Gradient Noise Attracts SGD Dynamics Towards Simpler Subnetworkscited by 39
- 2023Benign Overfitting in Linear Classifiers and Leaky ReLU Networks from KKT Conditions for Margin Maximizationcited by 29
- 2023The Double-Edged Sword of Implicit Bias: Generalization vs. Robustness in ReLU Networkscited by 23
- 2024Average gradient outer product as a mechanism for deep neural collapsecited by 22
- 2023The Law of Parsimony in Gradient Descent for Learning Deep Linear Networkscited by 20
- 2024Flavors of Margin: Implicit Bias of Steepest Descent in Homogeneous Neural Networkscited by 18
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 17
12 most cited of 25 All 25 (JSON)
- 2026Generalization Bounds of Stochastic Gradient Descent in Homogeneous Neural Networks26 Feb 2026
- 2026The Implicit Bias of Adam and Muon on Smooth Homogeneous Neural Networks18 Feb 2026
- 2026The Impact of Anisotropic Covariance Structure on the Training Dynamics and Generalization Error of Linear Networks11 Jan 2026
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025Embedding principle of homogeneous neural network for classification problem18 May 2025
- 2025The late-stage training dynamics of (stochastic) subgradient descent on homogeneous neural networks8 Feb 2025
- 2025Grokking at the Edge of Numerical Stability8 Jan 2025
- 2024Flavors of Margin: Implicit Bias of Steepest Descent in Homogeneous Neural Networks29 Oct 2024
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networks22 Sep 2024
- 2024Large Stepsize Gradient Descent for Non-Homogeneous Two-Layer Networks: Margin Improvement and Fast Optimization12 Jun 2024
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learning10 Jun 2024
12 newest of 25 All 25 (JSON)