The Implicit Bias of Depth: From Neural Collapse to Softmax Codes
Not on the map. Why?
Abstract (arXiv)
Neural collapse (NC) describes the structured geometry that emerges in the features and weights of trained classifiers. Recent theory suggests NC can be suboptimal in deep architectures, attributing this to an explicit low-rank bias from L2 regularization. We study the deep unconstrained feature model (UFM)-equivalent to a deep linear network with orthogonal inputs-trained without regularization, to isolate how gradient descent and depth alone shape NC. We show that depth induces an implicit low-rank bias: low-rank matrices propagate norm more efficiently through successive multiplications, promoting low-rank alternatives to NC. These alternatives, we argue, correspond to softmax codes: max-margin solutions previously found in width-bottlenecked networks. Analyzing training dynamics under spectral initialization, we identify an early-time repulsion among singular values that drives low-rank emergence, and characterize how depth shrinks NC's basin of attraction. Finally, we show that some effects act in the opposite direction: for randomly initialized networks, increasing width biases training toward higher-rank solutions. Our results provide the first asymptotic and dynamic characterization of implicit bias in deep UFMs trained with unregularized multiclass cross-entropy.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 85 of its 96 references that are in the map. First posted on arXiv 21 May 2026.
Builds on
85 of 96 references.
References · 85
- 2016Understanding deep learning requires rethinking generalizationcited by 1,674
- 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
- 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
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
- 2017Implicit Regularization in Matrix Factorizationcited by 322
- 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
- 2018On the Optimization of Deep Networks: Implicit Acceleration by Overparameterizationcited by 284
- 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 266
- 2018Implicit Bias of Gradient Descent on Linear Convolutional Networkscited by 251
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
- 2018A mathematical theory of semantic development in deep neural networkscited by 186
38 more
- 2018A Convergence Analysis of Gradient Descent for Deep Linear Neural Networkscited by 184
- 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
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
- 2020Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learningcited by 150
- 2020Neural collapse with unconstrained featurescited by 142
- 2019Implicit Regularization of Discrete Gradient Dynamics in Linear Neural Networkscited by 124
- 2020Implicit Regularization in Deep Learning May Not Be Explainable by Normscited by 119
- 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
- 2021On the Role of Neural Collapse in Transfer Learningcited by 100
- 2022On the Optimization Landscape of Neural Collapse under MSE Loss: Global Optimality with Unconstrained Featurescited by 97
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
- 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
- 2023Transformers as Support Vector Machinescited by 74
- 2022Limitations of Neural Collapse for Understanding Generalization in Deep Learningcited by 63
- 2019Learning deep linear neural networks: Riemannian gradient flows and convergence to global minimizerscited by 61
- 2020Gradient descent follows the regularization path for general lossescited by 59
- 2022Implicit Regularization Towards Rank Minimization in ReLU Networkscited by 53
- 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 49
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2020Gradient Descent for Deep Matrix Factorization: Dynamics and Implicit Bias towards Low Rankcited by 42
- 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
- 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
- 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 41
- 2023Neural Collapse in Deep Linear Networks: From Balanced to Imbalanced Datacited by 40
- 2023Generalized Neural Collapse for a Large Number of Classescited by 39
- 2022Implicit Bias of Large Depth Networks: a Notion of Rank for Nonlinear Functionscited by 38
- 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 38
- 2022A Law of Data Separation in Deep Learningcited by 36
- 2023(S)GD over Diagonal Linear Networks: Implicit Regularisation, Large Stepsizes and Edge of Stabilitycited by 36
- 2024The Implicit Bias of Adam on Separable Datacited by 36
- 2022Perturbation Analysis of Neural Collapsecited by 35
- 2021On Margin Maximization in Linear and ReLU Networkscited by 34
50 most cited of 85 All 85 (JSON)
Built on by
Not cited yet.