Provable Emergence of Deep Neural Collapse and Low-Rank Bias in L2-Regularized Nonlinear Networks
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
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2019Implicit Regularization in Deep Matrix Factorizationcited by 402
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
- 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 156
- 2021The Low-Rank Simplicity Bias in Deep Networkscited by 150
- 2020Neural collapse with unconstrained featurescited by 140
- 2022On the Optimization Landscape of Neural Collapse under MSE Loss: Global Optimality with Unconstrained Featurescited by 97
- 2020Traces of Class/Cross-Class Structure Pervade Deep Learning Spectracited by 94
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
- 2019Learning deep linear neural networks: Riemannian gradient flows and convergence to global minimizerscited by 61
23 more
- 2021Pufferfish: Communication-efficient Models At No Extra Costcited by 61
- 2021Initialization and Regularization of Factorized Neural Layerscited by 58
- 2019Learning Filter Basis for Convolutional Neural Network Compressioncited by 57
- 2021Analytic Insights into Structure and Rank of Neural Network Hessian Mapscited by 52
- 2020On the emergence of simplex symmetry in the final and penultimate layers of neural network classifierscited by 51
- 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
- 2022Rank Diminishing in Deep Neural Networkscited by 47
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2022Low-rank lottery tickets: finding efficient low-rank neural networks via matrix differential equationscited by 42
- 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
- 2022Training invariances and the low-rank phenomenon: beyond linear networkscited by 33
- 2023Implicit bias of SGD in L2-regularized linear DNNs: One-way jumps from high to low rankcited by 28
- 2022The Asymmetric Maximum Margin Bias of Quasi-Homogeneous Neural Networkscited by 25
- 2020The interplay between randomness and structure during learning in RNNscited by 20
- 2023Neural Collapse in the Intermediate Hidden Layers of Classification Neural Networkscited by 20
- 2023A Neural Collapse Perspective on Feature Evolution in Graph Neural Networkscited by 19
- 2024Neural Collapse for Cross-entropy Class-Imbalanced Learning with Unconstrained ReLU Feature Modelcited by 19
- 2023Neural (Tangent Kernel) Collapsecited by 18
- 2023Bottleneck Structure in Learned Features: Low-Dimension vs Regularity Tradeoffcited by 17
- 2022The Role of Linear Layers in Nonlinear Interpolating Networkscited by 15
- 2023Low Tensor Rank Learning of Neural Dynamicscited by 12
- 2021A gradient flow equation for optimal control problems with end-point costcited by 6
Built on by
Citing papers per year: 2024: 6; 2025: 0; 2026: 4.
10 papers.
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 36
- 2024Approaching Deep Learning through the Spectral Dynamics of Weightscited by 29
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 14
- 2024The Persistence of Neural Collapse Despite Low-Rank Biascited by 12
- 2024Transformer Block Coupling and its Correlation with Generalization in LLMscited by 9
- 2024Revealing the Utilized Rank of Subspaces of Learning in Neural Networkscited by 4
- 2026Beyond Neural Collapse: Task-Intrinsic Geometry Governs Neural Representations in Modular Arithmeticcited by 2
- 2026Does Weight Decay Enhance Training Stability?cited by 1
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codescited by 0
- 2026Gradient Smoothing: Coupling Layer-wise Updates for Improved Optimizationcited by 0
- 2026Gradient Smoothing: Coupling Layer-wise Updates for Improved Optimization29 Jun 2026
- 2026Beyond Neural Collapse: Task-Intrinsic Geometry Governs Neural Representations in Modular Arithmetic8 Jun 2026
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codes21 May 2026
- 2026Does Weight Decay Enhance Training Stability?15 May 2026
- 2024The Persistence of Neural Collapse Despite Low-Rank Bias30 Oct 2024
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
- 2024Approaching Deep Learning through the Spectral Dynamics of Weights21 Aug 2024
- 2024Transformer Block Coupling and its Correlation with Generalization in LLMs10 Jul 2024
- 2024Revealing the Utilized Rank of Subspaces of Learning in Neural Networks5 Jul 2024
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Models28 May 2024