Controlling Neural Collapse Enhances Out-of-Distribution Detection and Transfer Learning
Md Yousuf Harun, Jhair Gallardo, Christopher Kanan
摘要
Out-of-distribution (OOD) detection and OOD generalization are widely studied in Deep Neural Networks (DNNs), yet their relationship remains poorly understood. We empirically show that the degree of Neural Collapse (NC) in a network layer is inversely related with these objectives: stronger NC improves OOD detection but degrades generalization, while weaker NC enhances generalization at the cost of detection. This trade-off suggests that a single feature space cannot simultaneously achieve both tasks. To address this, we develop a theoretical framework linking NC to OOD detection and generalization. We show that entropy regularization mitigates NC to improve generalization, while a fixed Simplex Equiangular Tight Frame (ETF) projector enforces NC for better detection. Based on these insights, we propose a method to control NC at different DNN layers. In experiments, our method excels at both tasks across OOD datasets and DNN architectures. Code for our experiments is available: here.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper5
- Diagnosing Generalization Failures from Representational Geometry MarkersChi-Ning Chou, Artem Kirsanov, Yao-Yuan Yang, SueYeon ChungICLR 2026 · 被引用 2 次
- Optimizer Choice Matters For The Emergence of Neural CollapseJim Zhao, Tin Sum Cheng, Wojciech Masarczyk, Aurelien LucchiICLR 2026 · 被引用 1 次
- GEPC: Group-Equivariant Posterior Consistency for Out-of-Distribution Detection in Diffusion ModelsRouzoumka Yadang Alexis, Jean Pinsolle, Eugénie TERREAUX, christele morisseau 等ICML 2026 · 被引用 1 次
- Cluster-Aware Neural Collapse Prompt Tuning for Long-Tailed Generalization of Vision-Language ModelsBoyang Guo, Liang Li, Lin Peng, Yuhan Gao 等CVPR 2026
- Neural Collapse by Design: Learning Class Prototypes on the HyperspherePanagiotis Koromilas, Theodoros Giannakopoulos, Mihalis Nicolaou, Yannis PanagakisICML 2026
它引用的顶会 Paper42
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn 等ICLR 2021 · 被引用 21,477 次
- Bootstrap Your Own Latent - A New Approach to Self-Supervised LearningJean-Bastien Grill, Florian Strub, Florent Altché, Corentin Tallec 等NeurIPS 2020 · 被引用 9,171 次
- The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution GeneralizationDan Hendrycks, Steven Basart, Norman Mu, Saurav Kadavath 等ICCV 2021 · 被引用 2,294 次
- Energy-based Out-of-distribution DetectionWeitang Liu, Xiaoyun Wang, John D. Owens, Yixuan LiNeurIPS 2020 · 被引用 2,213 次
- Do Vision Transformers See Like Convolutional Neural Networks?Maithra Raghu, Thomas Unterthiner, Simon Kornblith, Chiyuan Zhang 等NeurIPS 2021 · 被引用 1,553 次
相关 Paper
- Detecting Out-of-Distribution Through the Lens of Neural CollapseLitian Liu, Yao QinCVPR 2025
- Neural Collapse in Deep Linear Networks: From Balanced to Imbalanced DataHien Dang, Tho Tran Huu, Stanley J. Osher, Hung Tran-The 等ICML 2023 · 被引用 44 次
- On the Optimization Landscape of Neural Collapse under MSE Loss: Global Optimality with Unconstrained FeaturesJinxin Zhou, Xiao Li, Tianyu Ding, Chong You 等ICML 2022 · 被引用 122 次
- Generalizing and Decoupling Neural Collapse via Hyperspherical Uniformity GapWeiyang Liu, Longhui Yu, Adrian Weller, Bernhard SchölkopfICLR 2023 · 被引用 3 次
- Neural Collapse for Cross-entropy Class-Imbalanced Learning with Unconstrained ReLU Features ModelHien Dang, Tho Tran Huu, Tan Minh Nguyen, Nhat HoICML 2024 · 被引用 19 次
