mlsanityData to 7 Oct 2026

Provable Emergence of Deep Neural Collapse and Low-Rank Bias in L2-Regularized Nonlinear Networks

arXiv 2402.03991 ↗First posted 6 Feb 2024cs.LG cs.NA math.NAcited by 10

Not on the map. Why?

Abstract (arXiv)

We present a unified theoretical framework connecting the first property of Deep Neural Collapse (DNC1) to the emergence of implicit low-rank bias in nonlinear networks trained with L2 weight decay regularization. Our main contributions are threefold. First, we derive a quantitative relation between the Total Cluster Variation (TCV) of intermediate embeddings and the numerical rank of stationary weight matrices. In particular, we establish that, at any critical point, the distance from a weight matrix to the set of rank-K matrices is bounded by a constant times the TCV of earlier-layer features, scaled inversely with the weight-decay parameter. Second, we prove global optimality of DNC1 in a constrained representation-cost setting for both feedforward and residual architectures, showing that zero TCV across intermediate layers minimizes the representation cost under natural architectural constraints. Third, we establish a benign landscape property: for almost every interpolating initialization there exists a continuous, loss-decreasing path from the initialization to a globally optimal, DNC1-satisfying configuration. Our theoretical claims are validated empirically; numerical experiments confirm the predicted relations among TCV, singular-value structure, and weight decay. These results indicate that neural collapse and low-rank bias are intimately linked phenomena arising from the optimization geometry induced by weight decay.

Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 35 of its 54 references that are in the map. First posted on arXiv 6 Feb 2024.

Builds on

35 of 54 references.

References · 35

  1. 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
  2. 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
  3. 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
  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 156
  6. 2021The Low-Rank Simplicity Bias in Deep Networkscited by 150
  7. 2020Neural collapse with unconstrained featurescited by 140
  8. 2022On the Optimization Landscape of Neural Collapse under MSE Loss: Global Optimality with Unconstrained Featurescited by 97
  9. 2020Traces of Class/Cross-Class Structure Pervade Deep Learning Spectracited by 94
  10. 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
  11. 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
  12. 2019Learning deep linear neural networks: Riemannian gradient flows and convergence to global minimizerscited by 61
23 more
  1. 2021Pufferfish: Communication-efficient Models At No Extra Costcited by 61
  2. 2021Initialization and Regularization of Factorized Neural Layerscited by 58
  3. 2019Learning Filter Basis for Convolutional Neural Network Compressioncited by 57
  4. 2021Analytic Insights into Structure and Rank of Neural Network Hessian Mapscited by 52
  5. 2020On the emergence of simplex symmetry in the final and penultimate layers of neural network classifierscited by 51
  6. 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
  7. 2022Rank Diminishing in Deep Neural Networkscited by 47
  8. 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
  9. 2022Low-rank lottery tickets: finding efficient low-rank neural networks via matrix differential equationscited by 42
  10. 2022Implicit Bias of Large Depth Networks: a Notion of Rank for Nonlinear Functionscited by 38
  11. 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
  12. 2022Training invariances and the low-rank phenomenon: beyond linear networkscited by 33
  13. 2023Implicit bias of SGD in L2-regularized linear DNNs: One-way jumps from high to low rankcited by 28
  14. 2022The Asymmetric Maximum Margin Bias of Quasi-Homogeneous Neural Networkscited by 25
  15. 2020The interplay between randomness and structure during learning in RNNscited by 20
  16. 2023Neural Collapse in the Intermediate Hidden Layers of Classification Neural Networkscited by 20
  17. 2023A Neural Collapse Perspective on Feature Evolution in Graph Neural Networkscited by 19
  18. 2024Neural Collapse for Cross-entropy Class-Imbalanced Learning with Unconstrained ReLU Feature Modelcited by 19
  19. 2023Neural (Tangent Kernel) Collapsecited by 18
  20. 2023Bottleneck Structure in Learned Features: Low-Dimension vs Regularity Tradeoffcited by 17
  21. 2022The Role of Linear Layers in Nonlinear Interpolating Networkscited by 15
  22. 2023Low Tensor Rank Learning of Neural Dynamicscited by 12
  23. 2021A gradient flow equation for optimal control problems with end-point costcited by 6

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 6; 2025: 0; 2026: 4.

10 papers.