EAT: Towards Long-Tailed Out-of-Distribution Detection

Abstract

Despite recent advancements in out-of-distribution (OOD) detection, most current studies assume a class-balanced in-distribution training dataset, which is rarely the case in real-world scenarios. This paper addresses the challenging task of long-tailed OOD detection, where the in-distribution data follows a long-tailed class distribution. The main difficulty lies in distinguishing OOD data from samples belonging to the tail classes, as the ability of a classifier to detect OOD instances is not strongly correlated with its accuracy on the in-distribution classes. To overcome this issue, we propose two simple ideas: (1) Expanding the in-distribution class space by introducing multiple abstention classes. This approach allows us to build a detector with clear decision boundaries by training on OOD data using virtual labels. (2) Augmenting the context-limited tail classes by overlaying images onto the context-rich OOD data. This technique encourages the model to pay more attention to the discriminative features of the tail classes. We provide a clue for separating in-distribution and OOD data by analyzing gradient noise. Through extensive experiments, we demonstrate that our method outperforms the current state-of-the-art on various benchmark datasets. Moreover, our method can be used as an add-on for existing long-tail learning approaches, significantly enhancing their OOD detection performance. Code is available at: https://github.com/Stomach-ache/Long-Tailed-OOD-Detection.

Cite

Text

Wei et al. "EAT: Towards Long-Tailed Out-of-Distribution Detection." AAAI Conference on Artificial Intelligence, 2024. doi:10.1609/AAAI.V38I14.29508

Markdown

[Wei et al. "EAT: Towards Long-Tailed Out-of-Distribution Detection." AAAI Conference on Artificial Intelligence, 2024.](https://mlanthology.org/aaai/2024/wei2024aaai-eat/) doi:10.1609/AAAI.V38I14.29508

BibTeX

@inproceedings{wei2024aaai-eat,
  title     = {{EAT: Towards Long-Tailed Out-of-Distribution Detection}},
  author    = {Wei, Tong and Wang, Bo-Lin and Zhang, Min-Ling},
  booktitle = {AAAI Conference on Artificial Intelligence},
  year      = {2024},
  pages     = {15787-15795},
  doi       = {10.1609/AAAI.V38I14.29508},
  url       = {https://mlanthology.org/aaai/2024/wei2024aaai-eat/}
}