Inducing Neural Collapse to a Fixed Hierarchy-Aware Frame for Reducing Mistake Severity
Not on the map. Why?
Abstract (arXiv)
There is a recently discovered and intriguing phenomenon called Neural Collapse: at the terminal phase of training a deep neural network for classification, the within-class penultimate feature means and the associated classifier vectors of all flat classes collapse to the vertices of a simplex Equiangular Tight Frame (ETF). Recent work has tried to exploit this phenomenon by fixing the related classifier weights to a pre-computed ETF to induce neural collapse and maximize the separation of the learned features when training with imbalanced data. In this work, we propose to fix the linear classifier of a deep neural network to a Hierarchy-Aware Frame (HAFrame), instead of an ETF, and use a cosine similarity-based auxiliary loss to learn hierarchy-aware penultimate features that collapse to the HAFrame. We demonstrate that our approach reduces the mistake severity of the model's predictions while maintaining its top-1 accuracy on several datasets of varying scales with hierarchies of heights ranging from 3 to 12. Code: https://github.com/ltong1130ztr/HAFrame
Key facts. Cited by 13 papers in the map (data through 7 Oct 2026). Builds on 32 of its 48 references that are in the map. First posted on arXiv 10 Mar 2023.
Builds on
32 of 48 references.
References · 32
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,451
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
- 2016Gaussian Error Linear Units (GELUs)cited by 3,431
- 2016Wide Residual Networkscited by 3,057
- 2013Fine-Grained Visual Classification of Aircraftcited by 1,701
- 2015Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs)cited by 1,061
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2018Meta-Learning for Semi-Supervised Few-Shot Classificationcited by 456
- 2021Exploring Deep Neural Networks via Layer-Peeled Model: Minority Collapse in Imbalanced Trainingcited by 166
20 more
- 2013Zero-Shot Learning by Convex Combination of Semantic Embeddingscited by 135
- 2019Making Better Mistakes: Leveraging Class Hierarchies with Deep Networkscited by 112
- 2022Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?cited by 100
- 2017Do Convolutional Neural Networks Learn Class Hierarchy?cited by 61
- 2021PASS: An ImageNet replacement for self-supervised pretraining without humanscited by 51
- 2023Understanding Imbalanced Semantic Segmentation Through Neural Collapsecited by 46
- 2018Fix your classifier: the marginal value of training the last weight layercited by 43
- 2018Hierarchy-based Image Embeddings for Semantic Image Retrievalcited by 41
- 2018Dynamic-structured Semantic Propagation Networkcited by 38
- 2020Solving Long-tailed Recognition with Deep Realistic Taxonomic Classifiercited by 34
- 2023Inducing Neural Collapse in Deep Long-tailed Learningcited by 28
- 2018Hierarchical Novelty Detection for Visual Object Recognitioncited by 26
- 2022Learning Hierarchy Aware Features for Reducing Mistake Severitycited by 26
- 2018Integrating domain knowledge: using hierarchies to improve deep classifierscited by 25
- 2021Regular Polytope Networkscited by 24
- 2021No Cost Likelihood Manipulation at Test Time for Making Better Mistakes in Deep Networkscited by 21
- 2022Skin Lesion Recognition with Class-Hierarchy Regularized Hyperbolic Embeddingscited by 19
- 2023Maximally Compact and Separated Features with Regular Polytope Networkscited by 5
- 2021Embedding Semantic Hierarchy in Discrete Optimal Transport for Risk Minimizationcited by 4
- 2021Connecting Sphere Manifolds Hierarchically for Regularizationcited by 2
Built on by
Citing papers per year: 2023: 3; 2024: 2; 2025: 5; 2026: 3.
13 papers.
Sort:
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 36
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 9
- 2023Symmetric Neural-Collapse Representations with Supervised Contrastive Loss: The Impact of ReLU and Batchingcited by 5
- 2024Out-of-Distribution Detection with Prototypical Outlier Proxycited by 4
- 2025ProHOC: Probabilistic Hierarchical Out-of-Distribution Classification via Multi-Depth Networkscited by 4
- 2025Making Better Mistakes in CLIP-Based Zero-Shot Classification with Hierarchy-Aware Language Promptscited by 2
- 2023It's an Alignment, Not a Trade-off: Revisiting Bias and Variance in Deep Modelscited by 1
- 2026Semi-Supervised Hierarchical Open-Set Classificationcited by 1
- 2025Hier-COS: Making Deep Features Hierarchy-aware via Composition of Orthogonal Subspacescited by 0
- 2025Imitate Optimal Policy: Prevail and Induce Action Collapse in Policy Gradientcited by 0
- 2026Online Continual Learning with Dynamic Label Hierarchiescited by 0
- 2026RST: certifying and constructing prescribed information in variational autoencoders18 May 2026
- 2026Online Continual Learning with Dynamic Label Hierarchies12 May 2026
- 2026Semi-Supervised Hierarchical Open-Set Classification23 Jan 2026
- 2025Imitate Optimal Policy: Prevail and Induce Action Collapse in Policy Gradient2 Sep 2025
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025ProHOC: Probabilistic Hierarchical Out-of-Distribution Classification via Multi-Depth Networks27 Mar 2025
- 2025Hier-COS: Making Deep Features Hierarchy-aware via Composition of Orthogonal Subspaces10 Mar 2025
- 2025Making Better Mistakes in CLIP-Based Zero-Shot Classification with Hierarchy-Aware Language Prompts4 Mar 2025
- 2024Out-of-Distribution Detection with Prototypical Outlier Proxy22 Dec 2024
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Models28 May 2024
- 2023It's an Alignment, Not a Trade-off: Revisiting Bias and Variance in Deep Models13 Oct 2023
- 2023Symmetric Neural-Collapse Representations with Supervised Contrastive Loss: The Impact of ReLU and Batching13 Jun 2023
1 more
- 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Modelscited by 36
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformerscited by 9
- 2023Symmetric Neural-Collapse Representations with Supervised Contrastive Loss: The Impact of ReLU and Batchingcited by 5
- 2024Out-of-Distribution Detection with Prototypical Outlier Proxycited by 4
- 2025ProHOC: Probabilistic Hierarchical Out-of-Distribution Classification via Multi-Depth Networkscited by 4
- 2025Making Better Mistakes in CLIP-Based Zero-Shot Classification with Hierarchy-Aware Language Promptscited by 2
- 2023It's an Alignment, Not a Trade-off: Revisiting Bias and Variance in Deep Modelscited by 1
- 2026Semi-Supervised Hierarchical Open-Set Classificationcited by 1
- 2025Hier-COS: Making Deep Features Hierarchy-aware via Composition of Orthogonal Subspacescited by 0
- 2025Imitate Optimal Policy: Prevail and Induce Action Collapse in Policy Gradientcited by 0
- 2026Online Continual Learning with Dynamic Label Hierarchiescited by 0
12 most cited of 13 All 13 (JSON)
- 2026RST: certifying and constructing prescribed information in variational autoencoders18 May 2026
- 2026Online Continual Learning with Dynamic Label Hierarchies12 May 2026
- 2026Semi-Supervised Hierarchical Open-Set Classification23 Jan 2026
- 2025Imitate Optimal Policy: Prevail and Induce Action Collapse in Policy Gradient2 Sep 2025
- 2025Neural Collapse is Globally Optimal in Deep Regularized ResNets and Transformers21 May 2025
- 2025ProHOC: Probabilistic Hierarchical Out-of-Distribution Classification via Multi-Depth Networks27 Mar 2025
- 2025Hier-COS: Making Deep Features Hierarchy-aware via Composition of Orthogonal Subspaces10 Mar 2025
- 2025Making Better Mistakes in CLIP-Based Zero-Shot Classification with Hierarchy-Aware Language Prompts4 Mar 2025
- 2024Out-of-Distribution Detection with Prototypical Outlier Proxy22 Dec 2024
- 2024Linguistic Collapse: Neural Collapse in (Large) Language Models28 May 2024
- 2023It's an Alignment, Not a Trade-off: Revisiting Bias and Variance in Deep Models13 Oct 2023
- 2023Symmetric Neural-Collapse Representations with Supervised Contrastive Loss: The Impact of ReLU and Batching13 Jun 2023
12 newest of 13 All 13 (JSON)