Neural Collapse Dynamics: Depth, Activation, Regularisation, and Feature Norm Threshold
Not on the map. Why?
Abstract (arXiv)
Neural collapse (NC) -- the convergence of penultimate-layer features to a simplex equiangular tight frame -- is well understood at equilibrium, but the dynamics governing its onset remain poorly characterised. We identify a simple and predictive regularity: NC occurs when the mean feature norm reaches a model-dataset-specific critical value, fn*, that is largely invariant to training conditions. This value concentrates tightly within each (model, dataset) pair (CV < 8%); training dynamics primarily affect the rate at which fn approaches fn*, rather than the value itself. In standard training trajectories, the crossing of fn below fn* consistently precedes NC onset, providing a practical predictor with a mean lead time of 62 epochs (MAE 24 epochs). A direct intervention experiment confirms fn* is a stable attractor of the gradient flow -- perturbations to feature scale are self-corrected during training, with convergence to the same value regardless of direction (p>0.2). Completing the (architecture)x(dataset) grid reveals the paper's strongest result: ResNet-20 on MNIST gives fn* = 5.867 -- a +458% architecture effect versus only +68% on CIFAR-10. The grid is strongly non-additive; fn* cannot be decomposed into independent architecture and dataset contributions. Four structural regularities emerge: (1) depth has a non-monotonic effect on collapse speed; (2) activation jointly determines both collapse speed and fn*; (3) weight decay defines a three-regime phase diagram -- too little slows, an optimal range is fastest, and too much prevents collapse; (4) width monotonically accelerates collapse while shifting fn* by at most 13%. These results establish feature-norm dynamics as an actionable diagnostic for predicting NC timing, suggesting that norm-threshold behaviour is a general mechanism underlying delayed representational reorganisation in deep networks.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 43 of its 59 references that are in the map. First posted on arXiv 31 Mar 2026.
Builds on
43 of 59 references.
References · 43
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2017Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithmscited by 4,349
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2016Gaussian Error Linear Units (GELUs)cited by 3,502
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,621
- 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 950
- 2023Progress measures for grokking via mechanistic interpretabilitycited by 839
- 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
- 2022Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasetscited by 629
31 more
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
- 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
- 2022Omnigrok: Grokking Beyond Algorithmic Datacited by 146
- 2020Neural collapse with unconstrained featurescited by 142
- 2023Neural Collapse Inspired Feature-Classifier Alignment for Few-Shot Class Incremental Learningcited by 116
- 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
- 2021On the Role of Neural Collapse in Transfer Learningcited by 100
- 2022Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?cited 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
- 2017Implicit Regularization in Deep Learningcited by 90
- 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
- 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
- 2022Limitations of Neural Collapse for Understanding Generalization in Deep Learningcited by 63
- 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
- 2022Neural Collapse Inspired Attraction-Repulsion-Balanced Loss for Imbalanced Learningcited by 43
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
- 2022Perturbation Analysis of Neural Collapsecited by 35
- 2022Understanding and Improving Transfer Learning of Deep Models via Neural Collapsecited by 24
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 16
- 2024The Persistence of Neural Collapse Despite Low-Rank Biascited by 13
- 2024Progressive Feedforward Collapse of ResNet Trainingcited by 12
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 10
- 2023Towards Understanding Neural Collapse: The Effects of Batch Normalization and Weight Decaycited by 9
- 2023A Study of Neural Collapse Phenomenon: Grassmannian Frame, Symmetry and Generalizationcited by 7
- 2026A Systematic Empirical Study of Grokking: Depth, Architecture, Activation, and Regularizationcited by 6
- 2025Neural Collapse Beyond the Unconstrained Features Model: Landscape, Dynamics, and Generalization in the Mean-Field Regimecited by 4
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergencecited by 4
- 2026Optimizer choice matters for the emergence of Neural Collapsecited by 1
Built on by
Citing papers per year: 2026: 2.
2 papers.