mlsanityData to 14 Oct 2026

How Label Imbalance Shapes Geometry: A General Spectral Analysis of Multi-Label Neural Collapse

arXiv 2605.01897 ↗First posted 3 May 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

This work investigates the phenomenon of Neural Collapse (NC) in multi-label classification, extending its conceptual framework from multi-class learning to general correlated and imbalanced multi-label settings. Although recent studies have identified a ''tag-wise averaging'' structure for multi-label features, this view relies on implicit assumptions of label balance and combinatorial symmetry. Consequently, it fails to account for the geometrical distortions caused by intrinsic label correlations and data imbalance, which are common in practice. We resolve the multiplicity-one imbalance conjecture raised by Li et al. (2024), showing that higher-multiplicity prototypes obey a class-frequency-weighted synthesis rule rather than uniform averaging. To address this, we propose a rigorous spectral-control framework to analyze the terminal phase of multi-label learning under general imbalanced conditions. We introduce the label covariance spectrum kappam, a scalar controlling the distribution-dependent lower-bound geometry, derived from the second-order moment matrix of the label distribution. Contrary to the averaging perspective, our analysis reveals that the centered label covariance spectrum controls the stability of terminal geometry by quantifying the weakest centered inter-class contrast directions. We prove that the classical Tag-wise Averaging emerges only as a special case under perfect orthogonality. Numerical experiments on synthetic distributions validate our theoretical bounds. This work resolves the scaled-average aspect of the imbalance conjecture and establishes a unifying theoretical framework that extends Neural Collapse to complex, imbalanced multi-label settings.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 32 of its 44 references that are in the map. First posted on arXiv 3 May 2026.

Builds on

32 of 44 references.

References · 32

  1. 2014Microsoft COCO: Common Objects in Contextcited by 17,608
  2. 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 989
  3. 2019Decoupling Representation and Classifier for Long-Tailed Recognitioncited by 809
  4. 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 719
  5. 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 227
  6. 2021Exploring Deep Neural Networks via Layer-Peeled Model: Minority Collapse in Imbalanced Trainingcited by 166
  7. 2021Neural Collapse Under MSE Loss: Proximity to and Dynamics on the Central Pathcited by 158
  8. 2020Learning Diverse and Discriminative Representations via the Principle of Maximal Coding Rate Reductioncited by 142
  9. 2020Neural collapse with unconstrained featurescited by 142
  10. 2019Learning with Fenchel-Young Lossescited by 119
  11. 2022Extended Unconstrained Features Model for Exploring Deep Neural Collapsecited by 102
  12. 2020The Emerging Trends of Multi-Label Learningcited by 100
20 more
  1. 2021On the Role of Neural Collapse in Transfer Learningcited by 100
  2. 2022Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?cited by 100
  3. 2020Neural Networks are Convex Regularizers: Exact Polynomial-time Convex Optimization Formulations for Two-layer Networkscited by 98
  4. 2022On the Optimization Landscape of Neural Collapse under MSE Loss: Global Optimality with Unconstrained Featurescited by 97
  5. 2021An Unconstrained Layer-Peeled Perspective on Neural Collapsecited by 87
  6. 2021ReduNet: A White-box Deep Network from the Principle of Maximizing Rate Reductioncited by 84
  7. 2022Imbalance Trouble: Revisiting Neural-Collapse Geometrycited by 80
  8. 2021ML-Decoder: Scalable and Versatile Classification Headcited by 79
  9. 2022Are All Losses Created Equal: A Neural Collapse Perspectivecited by 78
  10. 2020Neural Collapse with Cross-Entropy Losscited by 52
  11. 2023Deep Neural Collapse Is Provably Optimal for the Deep Unconstrained Features Modelcited by 43
  12. 2023Neural Collapse in Deep Linear Networks: From Balanced to Imbalanced Datacited by 40
  13. 2023Neural Collapse for Unconstrained Feature Model under Cross-entropy Loss with Imbalanced Datacited by 38
  14. 2022Perturbation Analysis of Neural Collapsecited by 35
  15. 2023On the Implicit Geometry of Cross-Entropy Parameterizations for Label-Imbalanced Datacited by 23
  16. 2023A Neural Collapse Perspective on Feature Evolution in Graph Neural Networkscited by 19
  17. 2024Neural Collapse for Cross-entropy Class-Imbalanced Learning with Unconstrained ReLU Feature Modelcited by 19
  18. 2023Neural Collapse in Multi-label Learning with Pick-all-label Losscited by 14
  19. 2014Local Rademacher Complexity for Multi-label Learningcited by 5
  20. 2020Optimistic bounds for multi-output predictioncited by 5

Built on by

Not cited yet.