Neural Collapse by Design: Learning Class Prototypes on the Hypersphere
Not on the map. Why?
Abstract (arXiv)
Supervised classification has a theoretical optimum, Neural Collapse (NC), yet neither of its two dominant paradigms reaches it in practice. Cross entropy (CE) leaves radial degrees of freedom unconstrained and converges to a degenerate geometry, while supervised contrastive learning (SCL) drives features toward NC during pretraining but discards this structure in a post hoc linear probing phase. We show that both paradigms are different appearances of the same method that contrasts prototypes on the unit hypersphere, and that closing the gap requires fixing each at its point of failure. From the CE side, we propose NTCE and NONL, two normalized losses that import contrastive optimization's missing ingredients into classifier learning: a large effective negative set and decoupled alignment and uniformity terms. From the SCL side, we prove that SCL's objective already optimizes throughout training for a principled classifier whose weights are the class mean embeddings, making linear probing both redundant and harmful. Empirically, on four benchmarks including ImageNet-1K, NTCE and NONL surpass CE accuracy, closely approximate NC (≥ 95%), and match CE's converged NC on 4/5 metrics in under 7.5% of its iterations, while SCL with fixed prototypes matches linear probing without the hours-long classifier training phase. The learned geometry yields +5.5% mean relative improvement in transfer learning, up to +8.7% under severe class imbalance, and improved robustness to corruptions on ImageNet-C. Our work recasts supervised learning as prototype learning on the hypersphere, with NC reached by design.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 47 of its 58 references that are in the map. First posted on arXiv 19 May 2026.
Builds on
47 of 58 references.
References · 47
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2017Prototypical Networks for Few-shot Learningcited by 3,015
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
- 2020Supervised Contrastive Learningcited by 2,829
- 2018Spectral Normalization for Generative Adversarial Networkscited by 1,636
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,405
- 2018CosFace: Large Margin Cosine Loss for Deep Face Recognitioncited by 971
- 2017SphereFace: Deep Hypersphere Embedding for Face Recognitioncited by 737
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
- 2017Spectrally-normalized margin bounds for neural networkscited by 686
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2020Long-tail learning via logit adjustmentcited by 618
35 more
- 2016Weight Normalization: A Simple Reparameterization to Accelerate Training of Deep Neural Networkscited by 543
- 2017A PAC-Bayesian Approach to Spectrally-Normalized Margin Bounds for Neural Networkscited by 314
- 2017NormFace: L2 Hypersphere Embedding for Face Verificationcited by 237
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
- 2016Associative Embedding: End-to-End Learning for Joint Detection and Groupingcited by 212
- 2017Formal Guarantees on the Robustness of a Classifier against Adversarial Manipulationcited by 185
- 2018MMA Training: Direct Input Space Margin Maximization through Adversarial Trainingcited by 172
- 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 158
- 2021Decoupled Contrastive Learningcited by 143
- 2017Semantic Instance Segmentation with a Discriminative Loss Functioncited by 118
- 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
- 2017Orthogonal Weight Normalization: Solution to Optimization over Multiple Dependent Stiefel Manifolds in Deep Neural Networkscited by 92
- 2019Instance Segmentation by Jointly Optimizing Spatial Embeddings and Clustering Bandwidthcited by 89
- 2016Robustness of classifiers: from adversarial to random noisecited by 88
- 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
- 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
- 2020Why Do Better Loss Functions Lead to Less Transferable Features?cited by 73
- 2021Dissecting Supervised Contrastive Learningcited by 64
- 2020Unsupervised Feature Learning by Cross-Level Instance-Group Discriminationcited by 51
- 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2022Perfectly Balanced: Improving Transfer and Robustness of Supervised Contrastive Learningcited by 40
- 2023Matrix Information Theory for Self-Supervised Learningcited by 21
- 2024Neural Collapse versus Low-rank Bias: Is Deep Neural Collapse Really Optimal?cited by 17
- 2024Guiding Neural Collapse: Optimising Towards the Nearest Simplex Equiangular Tight Framecited by 14
- 2025Controlling Neural Collapse Enhances Out-of-Distribution Detection and Transfer Learningcited by 11
- 2023Engineering the Neural Collapse Geometry of Supervised-Contrastive Losscited by 8
- 2023Equiangular Basis Vectorscited by 7
- 2024Bridging Mini-Batch and Asymptotic Analysis in Contrastive Learning: From InfoNCE to Kernel-Based Lossescited by 7
- 2023Symmetric Neural-Collapse Representations with Supervised Contrastive Loss: The Impact of ReLU and Batchingcited by 5
- 2024Unveiling the Dynamics of Information Interplay in Supervised Learningcited by 3
- 2024Neural Collapse Inspired Knowledge Distillationcited by 2
Built on by
Not cited yet.