Perturbation Analysis of Neural Collapse
Not on the map. Why?
Abstract (arXiv)
Training deep neural networks for classification often includes minimizing the training loss beyond the zero training error point. In this phase of training, a "neural collapse" behavior has been observed: the variability of features (outputs of the penultimate layer) of within-class samples decreases and the mean features of different classes approach a certain tight frame structure. Recent works analyze this behavior via idealized unconstrained features models where all the minimizers exhibit exact collapse. However, with practical networks and datasets, the features typically do not reach exact collapse, e.g., because deep layers cannot arbitrarily modify intermediate features that are far from being collapsed. In this paper, we propose a richer model that can capture this phenomenon by forcing the features to stay in the vicinity of a predefined features matrix (e.g., intermediate features). We explore the model in the small vicinity case via perturbation analysis and establish results that cannot be obtained by the previously studied models. For example, we prove reduction in the within-class variability of the optimized features compared to the predefined input features (via analyzing gradient flow on the "central-path" with minimal assumptions), analyze the minimizers in the near-collapse regime, and provide insights on the effect of regularization hyperparameters on the closeness to collapse. We support our theory with experiments in practical deep learning settings.
Key facts. Cited by 34 papers in the map (data through 7 Oct 2026). Builds on 25 of its 33 references that are in the map. First posted on arXiv 29 Oct 2022.
Builds on
25 of 33 references.
References · 25
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018Neural Ordinary Differential Equationscited by 3,733
- 2016Identity Mappings in Deep Residual Networkscited by 1,960
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2019Deep Equilibrium Modelscited by 603
- 2017Train longer, generalize better: closing the generalization gap in large batch training of neural networkscited by 278
- 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
- 2018Nonconvex Optimization Meets Low-Rank Matrix Factorization: An Overviewcited by 159
- 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 156
- 2017The Power of Interpolation: Understanding the Effectiveness of SGD in Modern Over-parametrized Learningcited by 150
- 2020Neural collapse with unconstrained featurescited by 140
13 more
- 2020Evaluation of Neural Architectures Trained with Square Loss vs Cross-Entropy in Classification Taskscited by 137
- 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 101
- 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
- 2018Does data interpolation contradict statistical optimality?cited by 95
- 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
- 2020Revealing the Structure of Deep Neural Networks via Convex Dualitycited by 54
- 2020On the emergence of simplex symmetry in the final and penultimate layers of neural network classifierscited by 51
- 2021Continuous vs. Discrete Optimization of Deep Neural Networkscited by 34
- 2022On the Implicit Bias Towards Minimal Depth of Deep Neural Networkscited by 12
Built on by
Citing papers per year: 2023: 8; 2024: 15; 2025: 4; 2026: 7.
34 papers.
Sort:
By area:
- 2024Unintentional Unalignment: Likelihood Displacement in Direct Preference Optimizationcited by 66
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2023Generalized Neural Collapse for a Large Number of Classescited by 39
- 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 36
- 2024Tracing Representation Progression: Analyzing and Enhancing Layer-Wise Similaritycited by 36
- 2024Average gradient outer product as a mechanism for deep neural collapsecited by 21
- 2024The Prevalence of Neural Collapse in Neural Multivariate Regressioncited by 21
- 2023The Law of Parsimony in Gradient Descent for Learning Deep Linear Networkscited by 19
- 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
22 more
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 16
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 14
- 2024Progressive Feedforward Collapse of ResNet Trainingcited by 12
- 2024Cross Entropy versus Label Smoothing: A Neural Collapse Perspectivecited by 11
- 2024Unifying Low Dimensional Spectra in Deep Learningcited by 11
- 2023Quantifying the Variability Collapse of Neural Networkscited by 9
- 2024Beyond Unconstrained Features: Neural Collapse for Shallow Neural Networks with General Datacited by 9
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 9
- 2023On the Robustness of Neural Collapse and the Neural Collapse of Robustnesscited by 8
- 2024Can Kernel Methods Explain How the Data Affects Neural Collapse?cited by 4
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergencecited by 4
- 2024Preventing Collapse in Contrastive Learning with Orthonormal Prototypes (CLOP)cited by 2
- 2024SGW-based Multi-Task Learning in Vision Taskscited by 2
- 2026Neural Collapse Dynamics: Depth, Activation, Regularisation, and Feature Norm Thresholdcited by 2
- 2025Robust Weight Imprinting: Insights from Neural Collapse and Proxy-Based Aggregationcited by 0
- 2025Enhancing Conformal Prediction via Class Similaritycited by 0
- 2026How Label Imbalance Shapes Geometry: A General Spectral Analysis of Multi-Label Neural Collapsecited by 0
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codescited by 0
- 2026The role of class encoding in neural collapsecited by 0
- 2026Optical-Guided Neural Collapse for SAR Few-Shot Class Incremental Learningcited by 0
- 2026Lifelong Representations: A Survey on Continual Self-Supervised Learning for Vision Modelscited by 0
- 2026Understanding Head Geometry and Dynamics in Federated Regression through a Natural Solution Selection Rule: An Unconstrained Feature Model Analysiscited by 0
- 2026Understanding Head Geometry and Dynamics in Federated Regression through a Natural Solution Selection Rule: An Unconstrained Feature Model Analysis30 Sep 2026
- 2026Lifelong Representations: A Survey on Continual Self-Supervised Learning for Vision Models8 Jul 2026
- 2026Optical-Guided Neural Collapse for SAR Few-Shot Class Incremental Learning3 Jun 2026
- 2026The role of class encoding in neural collapse29 May 2026
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codes21 May 2026
- 2026How Label Imbalance Shapes Geometry: A General Spectral Analysis of Multi-Label Neural Collapse3 May 2026
- 2026Neural Collapse Dynamics: Depth, Activation, Regularisation, and Feature Norm Threshold31 Mar 2026
- 2025Enhancing Conformal Prediction via Class Similarity24 Nov 2025
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergence25 Sep 2025
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025Robust Weight Imprinting: Insights from Neural Collapse and Proxy-Based Aggregation18 Mar 2025
- 2024Unintentional Unalignment: Likelihood Displacement in Direct Preference Optimization11 Oct 2024
22 more
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
- 2024SGW-based Multi-Task Learning in Vision Tasks3 Oct 2024
- 2024The Prevalence of Neural Collapse in Neural Multivariate Regression6 Sep 2024
- 2024Beyond Unconstrained Features: Neural Collapse for Shallow Neural Networks with General Data3 Sep 2024
- 2024Tracing Representation Progression: Analyzing and Enhancing Layer-Wise Similarity20 Jun 2024
- 2024Can Kernel Methods Explain How the Data Affects Neural Collapse?4 Jun 2024
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Models28 May 2024
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?23 May 2024
- 2024Progressive Feedforward Collapse of ResNet Training2 May 2024
- 2024Unifying Low Dimensional Spectra in Deep Learning9 Apr 2024
- 2024Preventing Collapse in Contrastive Learning with Orthonormal Prototypes (CLOP)27 Mar 2024
- 2024Average gradient outer product as a mechanism for deep neural collapse21 Feb 2024
- 2024Cross Entropy versus Label Smoothing: A Neural Collapse Perspective6 Feb 2024
- 2024Neural Collapse for Cross-entropy Class-Imbalanced Learning with Unconstrained ReLU Feature Model4 Jan 2024
- 2023On the Robustness of Neural Collapse and the Neural Collapse of Robustness13 Nov 2023
- 2023Generalized Neural Collapse for a Large Number of Classes9 Oct 2023
- 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Data18 Sep 2023
- 2023A Neural Collapse Perspective on Feature Evolution in Graph Neural Networks4 Jul 2023
- 2023Quantifying the Variability Collapse of Neural Networks6 Jun 2023
- 2023The Law of Parsimony in Gradient Descent for Learning Deep Linear Networks1 Jun 2023
- 2023Neural (Tangent Kernel) Collapse25 May 2023
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Model22 May 2023
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2023Generalized Neural Collapse for a Large Number of Classescited by 39
- 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 36
- 2024Tracing Representation Progression: Analyzing and Enhancing Layer-Wise Similaritycited by 36
- 2024Average gradient outer product as a mechanism for deep neural collapsecited by 21
- 2024The Prevalence of Neural Collapse in Neural Multivariate Regressioncited by 21
- 2023The Law of Parsimony in Gradient Descent for Learning Deep Linear Networkscited by 19
- 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
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 16
12 most cited of 33 All 33 (JSON)
- 2026Understanding Head Geometry and Dynamics in Federated Regression through a Natural Solution Selection Rule: An Unconstrained Feature Model Analysis30 Sep 2026
- 2026Lifelong Representations: A Survey on Continual Self-Supervised Learning for Vision Models8 Jul 2026
- 2026Optical-Guided Neural Collapse for SAR Few-Shot Class Incremental Learning3 Jun 2026
- 2026The role of class encoding in neural collapse29 May 2026
- 2026The Implicit Bias of Depth: From Neural Collapse to Softmax Codes21 May 2026
- 2026How Label Imbalance Shapes Geometry: A General Spectral Analysis of Multi-Label Neural Collapse3 May 2026
- 2026Neural Collapse Dynamics: Depth, Activation, Regularisation, and Feature Norm Threshold31 Mar 2026
- 2025Enhancing Conformal Prediction via Class Similarity24 Nov 2025
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergence25 Sep 2025
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025Robust Weight Imprinting: Insights from Neural Collapse and Proxy-Based Aggregation18 Mar 2025
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
12 newest of 33 All 33 (JSON)