ATOM: Robustifying Out-of-Distribution Detection Using Outlier Mining

被引:54
作者
Chen, Jiefeng [1 ]
Li, Yixuan [1 ]
Wu, Xi [2 ]
Liang, Yingyu [1 ]
Jha, Somesh [1 ]
机构
[1] Univ Wisconsin, Dept Comp Sci, 1210 W Dayton St, Madison, WI 53706 USA
[2] Google, Madison, WI USA
来源
MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2021: RESEARCH TRACK, PT III | 2021年 / 12977卷
基金
美国国家科学基金会;
关键词
Out-of-distribution detection; Outlier mining; Robustness;
D O I
10.1007/978-3-030-86523-8_26
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Detecting out-of-distribution (OOD) inputs is critical for safely deploying deep learning models in an open-world setting. However, existing OOD detection solutions can be brittle in the open world, facing various types of adversarial OOD inputs. While methods leveraging auxiliary OOD data have emerged, our analysis on illuminative examples reveals a key insight that the majority of auxiliary OOD examples may not meaningfully improve or even hurt the decision boundary of the OOD detector, which is also observed in empirical results on real data. In this paper, we provide a theoretically motivated method, Adversarial Training with informative Outlier Mining (ATOM), which improves the robustness of OOD detection. We show that, by mining informative auxiliary OOD data, one can significantly improve OOD detection performance, and somewhat surprisingly, generalize to unseen adversarial attacks. ATOM achieves state-of-the-art performance under a broad family of classic and adversarial OOD evaluation tasks. For example, on the CIFAR-10 in-distribution dataset, ATOM reduces the FPR (at TPR 95%) by up to 57.99% under adversarial OOD inputs, surpassing the previous best baseline by a large margin.
引用
收藏
页码:430 / 445
页数:16
相关论文
共 52 条
[1]  
[Anonymous], 2018, 6 INT C LEARN REPR I
[2]  
[Anonymous], 1995, Learning and example selection for object and pattern detection
[3]  
[Anonymous], 2017, P INT C LEARNING REP
[4]  
Athalye A, 2018, PR MACH LEARN RES, V80
[5]  
Bendale A, 2015, PROC CVPR IEEE, P1893, DOI 10.1109/CVPR.2015.7298799
[6]  
Bevandi<prime>c P., 2018, ARXIV PREPRINT ARXIV
[7]  
Biggio Battista, 2013, Machine Learning and Knowledge Discovery in Databases. European Conference, ECML PKDD 2013. Proceedings: LNCS 8190, P387, DOI 10.1007/978-3-642-40994-3_25
[8]  
Carmon Y, 2019, 33 C NEURAL INFORM P, V32
[9]  
Chrabaszcz P., 2017, CORR
[10]   Describing Textures in the Wild [J].
Cimpoi, Mircea ;
Maji, Subhransu ;
Kokkinos, Iasonas ;
Mohamed, Sammy ;
Vedaldi, Andrea .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :3606-3613