用合成数据提升图像分类模型对分布外样本的检测能力
BootOOD: Self-Supervised Out-of-Distribution Detection via Synthetic Sample Exposure under Neural Collapse
- 仅用正常数据生成伪分布外样本,通过特征范数差异区分异常
- 在CIFAR-10/100、ImageNet-200上性能超越多数现有方法
- 适合部署于安全关键场景,无需额外异常数据训练
分布外(OOD)检测对安全敏感环境中的图像分类至关重要,但现有方法在分布外样本与正常类别语义相近时表现不佳。本文提出BootOOD,一种完全自监督的OOD检测框架,仅依赖正常数据进行训练,专门应对语义挑战性强的分布外样本。该方法通过简单变换正常特征生成伪分布外特征,并利用神经坍缩(Neural Collapse, NC)特性:正常特征紧密聚集于类别均值且具有稳定特征范数。不同于以往将分布外特征约束在与坍缩类别均值正交子空间的方法,BootOOD引入轻量级辅助头,基于特征范数进行半径分类。此设计将分布外检测与主分类器解耦,放宽要求:分布外样本学习为比正常特征更小的范数,当正常与分布外语义接近时更易实现。在CIFAR-10、CIFAR-100和ImageNet-200上的实验表明,BootOOD优于多数后处理方法,超越无需异常暴露的训练方法,并与最先进的异常暴露方法相当,同时保持或提升正常分类准确率。
原文摘要 · Abstract (English)
Out-of-distribution (OOD) detection is critical for deploying image classifiers in safety-sensitive environments, yet existing detectors often struggle when OOD samples are semantically similar to the in-distribution (ID) classes. We present BootOOD, a fully self-supervised OOD detection framework that bootstraps exclusively from ID data and is explicitly designed to handle semantically challenging OOD samples. BootOOD synthesizes pseudo-OOD features through simple transformations of ID representations and leverages Neural Collapse (NC), where ID features cluster tightly around class means with consistent feature norms. Unlike prior approaches that aim to constrain OOD features into subspaces orthogonal to the collapsed ID means, BootOOD introduces a lightweight auxiliary head that performs radius-based classification on feature norms. This design decouples OOD detection from the primary classifier and imposes a relaxed requirement: OOD samples are learned to have smaller feature norms than ID features, which is easier to satisfy when ID and OOD are semantically close. Experiments on CIFAR-10, CIFAR-100, and ImageNet-200 show that BootOOD outperforms prior post-hoc methods, surpasses training-based methods without outlier exposure, and is competitive with state-of-the-art outlier-exposure approaches while maintaining or improving ID accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。