arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2310.08040cs.LG

SEE-OoD:用于增强分布外检测的监督式探索

SEE-OoD: Supervised Exploration For Enhanced Out-of-Distribution Detection

University of Michigan(密歇根大学);University of Michigan Transportation Research Institute(密歇根大学交通研究所);American University of Beirut(贝鲁特美国大学)

Xiaoyang Song, Wenbo Sun, Maher Nouiehed, Raed Al Kontar, Judy Jin

AI总结:

针对现有OoD检测方法过拟合或特征重叠失效的问题,提出基于Wasserstein分数的生成对抗训练方案,在有限OoD样本监督下同时进行数据增强与探索,提升检测精度并具备良好泛化性。

摘要:

当前的分布外(OoD)检测技术主要依赖于在训练阶段量化预测不确定性并引入模型正则化,使用真实或合成的OoD样本。然而,利用真实OoD样本的方法缺乏探索性,容易对现有OoD样本过拟合。而合成样本通常基于从训练数据中提取的特征生成,当训练数据与OoD数据在特征空间中高度重叠时,其效果会大打折扣。在这项工作中,我们提出了一种基于Wasserstein分数的生成对抗训练方案,以提升OoD检测精度。该方案首次在有限OoD样本的监督下,同时进行数据增强和探索。具体而言,生成器探索OoD空间,并利用判别器的反馈生成合成OoD样本;判别器则使用预定义的Wasserstein分数,同时利用观测样本和合成样本进行OoD检测。我们提供了理论保证,证明在经验设置下,通过对抗训练可以统计性地达到我们生成方案的最优解。随后,我们证明了所提方法在多种计算机视觉数据集上优于现有最先进技术,并对未见过的OoD数据展现出优越的泛化能力。

英文摘要:

Current techniques for Out-of-Distribution (OoD) detection predominantly rely on quantifying predictive uncertainty and incorporating model regularization during the training phase, using either real or synthetic OoD samples. However, methods that utilize real OoD samples lack exploration and are prone to overfit the OoD samples at hand. Whereas synthetic samples are often generated based on features extracted from training data, rendering them less effective when the training and OoD data are highly overlapped in the feature space. In this work, we propose a Wasserstein-score-based generative adversarial training scheme to enhance OoD detection accuracy, which, for the first time, performs data augmentation and exploration simultaneously under the supervision of limited OoD samples. Specifically, the generator explores OoD spaces and generates synthetic OoD samples using feedback from the discriminator, while the discriminator exploits both the observed and synthesized samples for OoD detection using a predefined Wasserstein score. We provide theoretical guarantees that the optimal solutions of our generative scheme are statistically achievable through adversarial training in empirical settings. We then demonstrate that the proposed method outperforms state-of-the-art techniques on various computer vision datasets and exhibits superior generalizability to unseen OoD data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑