Generalized Neural Collapse for a Large Number of Classes
Not on the map. Why?
Abstract (arXiv)
Neural collapse provides an elegant mathematical characterization of learned last layer representations (a.k.a. features) and classifier weights in deep classification models. Such results not only provide insights but also motivate new techniques for improving practical deep models. However, most of the existing empirical and theoretical studies in neural collapse focus on the case that the number of classes is small relative to the dimension of the feature space. This paper extends neural collapse to cases where the number of classes are much larger than the dimension of feature space, which broadly occur for language models, retrieval systems, and face recognition applications. We show that the features and classifier exhibit a generalized neural collapse phenomenon, where the minimum one-vs-rest margins is maximized.We provide empirical study to verify the occurrence of generalized neural collapse in practical deep neural networks. Moreover, we provide theoretical study to show that the generalized neural collapse provably occurs under unconstrained feature model with spherical constraint, under certain technical conditions on feature dimension and number of classes.
Key facts. Cited by 39 papers in the map (data through 14 Oct 2026). Builds on 42 of its 75 references that are in the map. First posted on arXiv 9 Oct 2023.
Builds on
42 of 75 references.
References · 42
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,513
- 2020Exploring Simple Siamese Representation Learningcited by 2,866
- 2021Convex optimizationcited by 2,515
- 2020Improved Baselines with Momentum Contrastive Learningcited by 2,287
- 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 989
- 2018CosFace: Large Margin Cosine Loss for Deep Face Recognitioncited by 971
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
30 more
- 2020Contrastive Learning of Medical Visual Representations from Paired Images and Textcited by 641
- 2018Additive Margin Softmax for Face Verificationcited by 278
- 2017NormFace: L2 Hypersphere Embedding for Face Verificationcited by 237
- 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
- 2020Pre-training Tasks for Embedding-based Large-scale Retrievalcited by 153
- 2020Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reductioncited by 142
- 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
- 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
- 2021ReduNet: A White-box Deep Network from the Principle of Maximizing Rate Reductioncited by 84
- 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
- 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
- 2020On the emergence of simplex symmetry in the final and penultimate layers of neural network classifierscited by 52
- 2020Neural Collapse with Cross-Entropy Losscited by 52
- 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
- 2020Explicit regularization and implicit bias in deep network classifiers trained with the square losscited by 37
- 2022Perturbation Analysis of Neural Collapsecited by 35
- 2023Inducing Neural Collapse in Deep Long-tailed Learningcited by 28
- 2023Generalizing and Decoupling Neural Collapse via Hyperspherical Uniformity Gapcited by 14
- 2022Hidden State Variability of Pretrained Language Models Can Guide Computation Reduction for Transfer Learningcited by 13
- 2022Continual Learning by Modeling Intra-Class Variationcited by 12
- 2023A Study of Neural Collapse Phenomenon: Grassmannian Frame, Symmetry and Generalizationcited by 7
- 2022Learning Towards the Largest Marginscited by 5
Built on by
Citing papers per year: 2023: 2; 2024: 15; 2025: 11; 2026: 11.
39 papers.
Sort:
By area:
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 38
- 2024Explaining and Mitigating the Modality Gap in Contrastive Multimodal Learningcited by 26
- 2024Provably Optimal Memory Capacity for Modern Hopfield Models: Transformer-Compatible Dense Associative Memories as Spherical Codescited by 19
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 17
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 16
- 2023Neural Collapse in Multi-label Learning with Pick-all-label Losscited by 14
- 2024A Global Geometric Analysis of Maximal Coding Rate Reductioncited by 13
- 2024The Persistence of Neural Collapse Despite Low-Rank Biascited by 13
- 2024Cross Entropy versus Label Smoothing: A Neural Collapse Perspectivecited by 11
- 2024Unifying Low Dimensional Spectra in Deep Learningcited by 11
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 10
- 2024Control-oriented Clustering of Visual Latent Representationcited by 8
27 more
- 2024Cell-ontology guided transcriptome foundation modelcited by 7
- 2025DualCP: Rehearsal-Free Domain-Incremental Learning via Dual-Level Concept Prototypecited by 7
- 2024Task structure and nonlinearity jointly determine learned representational geometrycited by 6
- 2025From Compression to Expression: A Layerwise Analysis of In-Context Learningcited by 6
- 2025Consistent Supervised-Unsupervised Alignment for Generalized Category Discoverycited by 6
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergencecited by 4
- 2023Efficient Compression of Overparameterized Deep Models through Low-Dimensional Learning Dynamicscited by 3
- 2024All-around Neural Collapse for Imbalanced Classificationcited by 3
- 2024Understanding Representation of Deep Equilibrium Models from Neural Collapse Perspectivecited by 3
- 2025Mitigating Negative Flips via Margin Preserving Trainingcited by 3
- 2024Preventing Collapse in Contrastive Learning with Orthonormal Prototypes (CLOP)cited by 2
- 2025Neural Multivariate Regression: Qualitative Insights from the Unconstrained Feature Modelcited by 2
- 2025Neural Collapse under Gradient Flow on Shallow ReLU Networks for Orthogonally Separable Datacited by 2
- 2025Diagonalizing the Softmax: Hadamard Initialization for Tractable Cross-Entropy Dynamicscited by 2
- 2026Structure Before Collapse: Transient semantic geometry in next-token predictioncited by 2
- 2025BCE vs. CE in Deep Feature Learningcited by 1
- 2025Heads collapse, features stay: Why Replay needs big bufferscited by 1
- 2026Optimizer choice matters for the emergence of Neural Collapsecited by 1
- 2026Neural collapse in the orthoplex regimecited by 1
- 2026A Stationary (and Therefore Compatible) Representation is All You Needcited by 1
- 2026XtraLight-MedMamba for Classification of Neoplastic Tubular Adenomascited by 0
- 2026Supervised Classification Heads as Semantic Prototypes: Unlocking Vision-Language Alignment via Weight Recyclingcited 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
- 2026Resolving superposition in AI for interpretability and cross-modal alignment in patient-neuronal imagescited by 0
- 2026Neural Collapse Is Forbidden: Information Floors in Language Modelscited by 0
- 2026Approximate Homomorphisms and Convergent Representations in Transducerscited by 0
- 2026Approximate Homomorphisms and Convergent Representations in Transducers19 Aug 2026
- 2026Neural Collapse Is Forbidden: Information Floors in Language Models10 Jul 2026
- 2026Resolving superposition in AI for interpretability and cross-modal alignment in patient-neuronal images30 Jun 2026
- 2026Structure Before Collapse: Transient semantic geometry in next-token prediction25 Jun 2026
- 2026A Stationary (and Therefore Compatible) Representation is All You Need10 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
- 2026Supervised Classification Heads as Semantic Prototypes: Unlocking Vision-Language Alignment via Weight Recycling21 May 2026
- 2026Neural collapse in the orthoplex regime21 Mar 2026
- 2026Optimizer choice matters for the emergence of Neural Collapse18 Feb 2026
- 2026XtraLight-MedMamba for Classification of Neoplastic Tubular Adenomas4 Feb 2026
- 2025Heads collapse, features stay: Why Replay needs big buffers8 Dec 2025
27 more
- 2025Diagonalizing the Softmax: Hadamard Initialization for Tractable Cross-Entropy Dynamics3 Dec 2025
- 2025Mitigating Negative Flips via Margin Preserving Training11 Nov 2025
- 2025Neural Collapse under Gradient Flow on Shallow ReLU Networks for Orthogonally Separable Data24 Oct 2025
- 2025Explaining Grokking and Information Bottleneck through Neural Collapse Emergence25 Sep 2025
- 2025Consistent Supervised-Unsupervised Alignment for Generalized Category Discovery7 Jul 2025
- 2025From Compression to Expression: A Layerwise Analysis of In-Context Learning22 May 2025
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025Neural Multivariate Regression: Qualitative Insights from the Unconstrained Feature Model14 May 2025
- 2025BCE vs. CE in Deep Feature Learning9 May 2025
- 2025DualCP: Rehearsal-Free Domain-Incremental Learning via Dual-Level Concept Prototype23 Mar 2025
- 2024Explaining and Mitigating the Modality Gap in Contrastive Multimodal Learning10 Dec 2024
- 2024Understanding Representation of Deep Equilibrium Models from Neural Collapse Perspective30 Oct 2024
- 2024The Persistence of Neural Collapse Despite Low-Rank Bias30 Oct 2024
- 2024Provably Optimal Memory Capacity for Modern Hopfield Models: Transformer-Compatible Dense Associative Memories as Spherical Codes30 Oct 2024
- 2024Control-oriented Clustering of Visual Latent Representation7 Oct 2024
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapse7 Oct 2024
- 2024Cell-ontology guided transcriptome foundation model22 Aug 2024
- 2024All-around Neural Collapse for Imbalanced Classification14 Aug 2024
- 2024A Global Geometric Analysis of Maximal Coding Rate Reduction4 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
- 2024Unifying Low Dimensional Spectra in Deep Learning9 Apr 2024
- 2024Preventing Collapse in Contrastive Learning with Orthonormal Prototypes (CLOP)27 Mar 2024
- 2024Cross Entropy versus Label Smoothing: A Neural Collapse Perspective6 Feb 2024
- 2024Task structure and nonlinearity jointly determine learned representational geometry24 Jan 2024
- 2023Efficient Compression of Overparameterized Deep Models through Low-Dimensional Learning Dynamics8 Nov 2023
- 2023Neural Collapse in Multi-label Learning with Pick-all-label Loss24 Oct 2023
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 38
- 2024Explaining and Mitigating the Modality Gap in Contrastive Multimodal Learningcited by 26
- 2024Provably Optimal Memory Capacity for Modern Hopfield Models: Transformer-Compatible Dense Associative Memories as Spherical Codescited by 19
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 17
- 2024Wide Neural Networks Trained with Weight Decay Provably Exhibit Neural Collapsecited by 16
- 2023Neural Collapse in Multi-label Learning with Pick-all-label Losscited by 14
- 2024A Global Geometric Analysis of Maximal Coding Rate Reductioncited by 13
- 2024The Persistence of Neural Collapse Despite Low-Rank Biascited by 13
- 2024Cross Entropy versus Label Smoothing: A Neural Collapse Perspectivecited by 11
- 2024Unifying Low Dimensional Spectra in Deep Learningcited by 11
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 10
- 2024Control-oriented Clustering of Visual Latent Representationcited by 8
12 most cited of 36 All 36 (JSON)
- 2026Approximate Homomorphisms and Convergent Representations in Transducers19 Aug 2026
- 2026Neural Collapse Is Forbidden: Information Floors in Language Models10 Jul 2026
- 2026Resolving superposition in AI for interpretability and cross-modal alignment in patient-neuronal images30 Jun 2026
- 2026Structure Before Collapse: Transient semantic geometry in next-token prediction25 Jun 2026
- 2026A Stationary (and Therefore Compatible) Representation is All You Need10 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
- 2026Neural collapse in the orthoplex regime21 Mar 2026
- 2026Optimizer choice matters for the emergence of Neural Collapse18 Feb 2026
- 2025Heads collapse, features stay: Why Replay needs big buffers8 Dec 2025
- 2025Diagonalizing the Softmax: Hadamard Initialization for Tractable Cross-Entropy Dynamics3 Dec 2025
- 2025Mitigating Negative Flips via Margin Preserving Training11 Nov 2025
12 newest of 36 All 36 (JSON)