NECO: NEural Collapse Based Out-of-distribution detection
Mouïn Ben Ammar
Nacim Belkhir
Sebastian Popescu
Antoine Manzanera
Gianni Franchi
[Paper]
[GitLab]



Example of neural collapse convergence on ID/OOD


Abstract

Detecting out-of-distribution (OOD) data is a critical challenge in machine learning due to model overconfidence, often without awareness of their epistemological limits. We hypothesize that "neural collapse", a phenomenon affecting in-distribution data for models trained beyond loss convergence, also influences OOD data. To benefit from this interplay, we introduce NECO, a novel post-hoc method for OOD detection, which leverages the geometric properties of “neural collapse” and of principal component spaces to identify OOD data. Our extensive experiments demonstrate that NECO achieves state-of-the-art results on both small and large-scale OOD detection tasks while exhibiting strong generalization capabilities across different network architectures. Furthermore, we provide a theoretical explanation for the effectiveness of our method in OOD detection.


Talk



Paper & Supplementary Material

Ammar, M. B., Belkhir, N., Popescu, S., Manzanera, A., & Franchi, G.
NECO: NEural Collapse Based Out-of-distribution detection
In ICLR, 2024.
(hosted on ArXiv)


[Slides]
[Bibtex]
[Code]