English

Class-wise Thresholding for Robust Out-of-Distribution Detection

Machine Learning 2022-07-04 v3 Artificial Intelligence

Abstract

We consider the problem of detecting OoD(Out-of-Distribution) input data when using deep neural networks, and we propose a simple yet effective way to improve the robustness of several popular OoD detection methods against label shift. Our work is motivated by the observation that most existing OoD detection algorithms consider all training/test data as a whole, regardless of which class entry each input activates (inter-class differences). Through extensive experimentation, we have found that such practice leads to a detector whose performance is sensitive and vulnerable to label shift. To address this issue, we propose a class-wise thresholding scheme that can apply to most existing OoD detection algorithms and can maintain similar OoD detection performance even in the presence of label shift in the test distribution.

Keywords

Cite

@article{arxiv.2110.15292,
  title  = {Class-wise Thresholding for Robust Out-of-Distribution Detection},
  author = {Matteo Guarrera and Baihong Jin and Tung-Wei Lin and Maria Zuluaga and Yuxin Chen and Alberto Sangiovanni-Vincentelli},
  journal= {arXiv preprint arXiv:2110.15292},
  year   = {2022}
}

Comments

12 pages, 7 figures, 7 tables

R2 v1 2026-06-24T07:16:26.674Z