动态追踪特征空间判别轴,提升流式数据下分布外检测性能。
Tracking the Discriminative Axis: Dual Prototypes for Test-Time OOD Detection Under Covariate Shift
- 在线追踪ID与OOD双原型,适应环境变化的判别轴。
- 在ImageNet-C vs. Textures-C上提升15.32% AUROC,FPR降低49.15个百分点。
- 适用于持续变化环境下需实时检测异常输入的系统。
为保障深度学习系统的可靠部署,分布外(OOD)检测至关重要。现实场景中,测试数据常以随时间演化的混合样本形式出现,且受协变量偏移影响,ID与OOD样本均受相同协变量因素影响。现有方法多假设ID分布恒定,但此假设在该场景下失效,导致性能严重下降。我们实证发现,即使存在协变量偏移,协变量偏移后的ID(csID)与OOD(csOOD)样本在特征空间仍可沿判别轴分离。基于此,提出DART方法,在测试时动态追踪双原型(分别代表ID和OOD),以恢复漂移的判别轴,并引入多层融合与翻转校正增强鲁棒性。在涵盖多种挑战性基准的数据集上进行实验,所有数据集均施加15种常见噪声类型,严重程度为5级,结果表明该方法显著提升性能:相比主流基线,在ImageNet-C vs. Textures-C上实现15.32百分点的AUROC增益和49.15百分点的FPR@95TPR降低。
原文摘要 · Abstract (English)
For reliable deployment of deep-learning systems, out-of-distribution (OOD) detection is indispensable. In the real world, where test-time inputs often arrive as streaming mixtures of in-distribution (ID) and OOD samples under evolving covariate shifts, OOD samples are domain-constrained and bounded by the environment, and both ID and OOD are jointly affected by the same covariate factors. Existing methods typically assume a stationary ID distribution, but this assumption breaks down in such settings, leading to severe performance degradation. We empirically discover that, even under covariate shift, covariate-shifted ID (csID) and OOD (csOOD) samples remain separable along a discriminative axis in feature space. Building on this observation, we propose DART, a test-time, online OOD detection method that dynamically tracks dual prototypes -- one for ID and the other for OOD -- to recover the drifting discriminative axis, augmented with multi-layer fusion and flip correction for robustness. Extensive experiments on a wide range of challenging benchmarks, where all datasets are subjected to 15 common corruption types at severity level 5, demonstrate that our method significantly improves performance, yielding 15.32 percentage points (pp) AUROC gain and 49.15 pp FPR@95TPR reduction on ImageNet-C vs. Textures-C compared to established baselines. These results highlight the potential of the test-time discriminative axis tracking for dependable OOD detection in dynamically changing environments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。