arXiv cs.AI· Donghwan Kim, Hyunsoo Yoon·· 14 小时前AI 评分27
MahaVar:基于神经坍缩下类间马氏距离方差的 OOD 检测方法
MahaVar: OOD Detection via Class-wise Mahalanobis Distance Variance under Neural Collapse
AI 导读
MahaVar 是一种事后 OOD 检测器,在马氏距离基础上加入类间距离方差项,利用 ID 样本类间马氏距离方差高、OOD 样本方差低的特点区分二者。该方法从神经坍缩几何给出理论依据,在 OpenOOD v1.5 协议下于 CIFAR-100 和 ImageNet 上取得 SOTA,AUROC 与 FPR@95 均优于现有基于马氏距离的方法。
正文 · 原文
Abstract:Out-of-distribution (OOD) detection is a critical component for ensuring the reliability of deep neural networks in safety-critical applications. In this work, we present a key empirical observation: for in-distribution (ID) samples, class-wise Mahalanobis distances exhibit a pronounced sharp minimum structure, where the distance to the nearest class is small while distances to all other classes remain large, resulting in high variance across classes. In contrast, OOD samples tend to exhibit a less pronounced sharp minimum structure, producing comparatively lower variance across classes. We further provide a theoretical analysis grounding this observation in Neural Collapse geometry: under relaxed Neural Collapse assumptions on within-class compactness and inter-class separation, ID samples are shown to structurally exhibit high class-wise distance variance, offering a theoretical basis for its use as an OOD score. Motivated by this observation and its theoretical backing, we propose MahaVar, a simple and effective post-hoc OOD detector that augments the Mahalanobis distance with a class-wise distance variance term. Following the OpenOOD v1.5 benchmark protocol, MahaVar achieves state-of-the-art performance on CIFAR-100 and ImageNet, with consistent improvements in both AUROC and FPR@95 over existing Mahalanobis-based methods across all benchmarks.
| Comments: | NeurIPS 2026 |
| Subjects: | Machine Learning (cs.LG); Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2605.14413 [cs.LG] |
| (or arXiv:2605.14413v2 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2605.14413 arXiv-issued DOI via DataCite |
Submission history
From: Donghwan Kim [view email]
[v1]
Thu, 14 May 2026 05:58:19 UTC (2,335 KB)
[v2]
Thu, 29 Oct 2026 07:36:48 UTC (2,311 KB)
来源:arXiv cs.AI · arxiv.org