mlsanityData to 14 Oct 2026

The Implicit Bias of Depth: From Neural Collapse to Softmax Codes

arXiv 2605.23087 ↗First posted 21 May 2026cs.LGcited by 0

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

  1. 2016Understanding deep learning requires rethinking generalizationcited by 1,674
  2. 2013Exact solutions to the nonlinear dynamics of learning in deep linear neural networkscited by 763
  3. 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
  4. 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
  5. 2019Implicit Regularization in Deep Matrix Factorizationcited by 405
  6. 2017Implicit Regularization in Matrix Factorizationcited by 322
  7. 2019Gradient Descent Maximizes the Margin of Homogeneous Neural Networkscited by 320
  8. 2018On the Optimization of Deep Networks: Implicit Acceleration by Overparameterizationcited by 284
  9. 2018Characterizing Implicit Bias in Terms of Optimization Geometrycited by 266
  10. 2018Implicit Bias of Gradient Descent on Linear Convolutional Networkscited by 251
  11. 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
  12. 2018A mathematical theory of semantic development in deep neural networkscited by 186
38 more
  1. 2018A Convergence Analysis of Gradient Descent for Deep Linear Neural Networkscited by 184
  2. 2018Gradient descent aligns the layers of deep linear networkscited by 182
  3. 2020Directional convergence and alignment in deep learningcited by 178
  4. 2021Exploring Deep Neural Networks via Layer-Peeled Model: Minority Collapse in Imbalanced Trainingcited by 166
  5. 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 158
  6. 2021The Low-Rank Simplicity Bias in Deep Networkscited by 154
  7. 2020Towards Resolving the Implicit Bias of Gradient Descent for Matrix Factorization: Greedy Low-Rank Learningcited by 150
  8. 2020Neural collapse with unconstrained featurescited by 142
  9. 2019Implicit Regularization of Discrete Gradient Dynamics in Linear Neural Networkscited by 124
  10. 2020Implicit Regularization in Deep Learning May Not Be Explainable by Normscited by 119
  11. 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
  12. 2021On the Role of Neural Collapse in Transfer Learningcited by 100
  13. 2022On the Optimization Landscape of Neural Collapse under MSE Loss: Global Optimality with Unconstrained Featurescited by 97
  14. 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
  15. 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
  16. 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
  17. 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
  18. 2023Transformers as Support Vector Machinescited by 74
  19. 2022Limitations of Neural Collapse for Understanding Generalization in Deep Learningcited by 63
  20. 2019Learning deep linear neural networks: Riemannian gradient flows and convergence to global minimizerscited by 61
  21. 2020Gradient descent follows the regularization path for general lossescited by 59
  22. 2022Implicit Regularization Towards Rank Minimization in ReLU Networkscited by 53
  23. 2024From Lazy to Rich: Exact Learning Dynamics in Deep Linear Networkscited by 49
  24. 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
  25. 2020Gradient Descent for Deep Matrix Factorization: Dynamics and Implicit Bias towards Low Rankcited by 42
  26. 2023Implicit Bias of Gradient Descent for Logistic Regression at the Edge of Stabilitycited by 42
  27. 2024Get rich quick: exact solutions reveal how unbalanced initializations promote rapid feature learningcited by 42
  28. 2025Implicit Bias of Spectral Descent and Muon on Multiclass Separable Datacited by 41
  29. 2023Neural Collapse in Deep Linear Networks: From Balanced to Imbalanced Datacited by 40
  30. 2023Generalized Neural Collapse for a Large Number of Classescited by 39
  31. 2022Implicit Bias of Large Depth Networks: a Notion of Rank for Nonlinear Functionscited by 38
  32. 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
  33. 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 38
  34. 2022A Law of Data Separation in Deep Learningcited by 36
  35. 2023(S)GD over Diagonal Linear Networks: Implicit Regularisation, Large Stepsizes and Edge of Stabilitycited by 36
  36. 2024The Implicit Bias of Adam on Separable Datacited by 36
  37. 2022Perturbation Analysis of Neural Collapsecited by 35
  38. 2021On Margin Maximization in Linear and ReLU Networkscited by 34

50 most cited of 85 All 85 (JSON)

Built on by

Not cited yet.