Space Alignment Matters: The Missing Piece for Inducing Neural Collapse in Long-Tailed Learning
DOI:
https://doi.org/10.1609/aaai.v40i31.39835Abstract
Recent studies on Neural Collapse (NC) reveal that, under class-balanced conditions, the class feature means and the classifier weights spontaneously align into a simplex equiangular tight frame (ETF). In long-tailed regimes, however, severe sample imbalance tends to prevent the emergence of the NC phenomenon, resulting in poor generalization performance.Current efforts predominantly seek to recover the ETF geometry by imposing constraints on features or classifier weights, yet overlook a critical problem: There is a pronounced misalignment between the feature and the classifier weight spaces. In this paper, we theoretically quantify the harm of such misalignment through an optimal error exponent analysis.Built on this insight, we propose three explicit alignment strategies that plug-and-play into existing long-tail methods without architectural change. Extensive experiments on the CIFAR-10-LT, CIFAR-100-LT, and ImageNet-LT datasets consistently boost examined baselines and achieve the state-of-the-art performances.Downloads
Published
2026-03-14
How to Cite
Wang, J., Gao, Z., & Xie, Z. (2026). Space Alignment Matters: The Missing Piece for Inducing Neural Collapse in Long-Tailed Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 40(31), 26302–26309. https://doi.org/10.1609/aaai.v40i31.39835
Issue
Section
AAAI Technical Track on Machine Learning VIII