Out-of-Distribution Detection in Long-Tailed Recognition with Calibrated Outlier Class Learning

Authors

  • Wenjun Miao School of Computer Science and Engineering, Beihang University
  • Guansong Pang School of Computing and Information Systems, Singapore Management University
  • Xiao Bai School of Computer Science and Engineering, Beihang University State Key Laboratory of Software Development Environment, Jiangxi Research Institute, Beihang University
  • Tianqi Li School of Computer Science and Engineering, Beihang University
  • Jin Zheng School of Computer Science and Engineering, Beihang University State Key Laboratory of Virtual Reality Technology and Systems, Beihang University

DOI:

https://doi.org/10.1609/aaai.v38i5.28217

Keywords:

CV: Object Detection & Categorization, CV: Adversarial Attacks & Robustness, CV: Applications

Abstract

Existing out-of-distribution (OOD) methods have shown great success on balanced datasets but become ineffective in long-tailed recognition (LTR) scenarios where 1) OOD samples are often wrongly classified into head classes and/or 2) tail-class samples are treated as OOD samples. To address these issues, current studies fit a prior distribution of auxiliary/pseudo OOD data to the long-tailed in-distribution (ID) data. However, it is difficult to obtain such an accurate prior distribution given the unknowingness of real OOD samples and heavy class imbalance in LTR. A straightforward solution to avoid the requirement of this prior is to learn an outlier class to encapsulate the OOD samples. The main challenge is then to tackle the aforementioned confusion between OOD samples and head/tail-class samples when learning the outlier class. To this end, we introduce a novel calibrated outlier class learning (COCL) approach, in which 1) a debiased large margin learning method is introduced in the outlier class learning to distinguish OOD samples from both head and tail classes in the representation space and 2) an outlier-class-aware logit calibration method is defined to enhance the long-tailed classification confidence. Extensive empirical results on three popular benchmarks CIFAR10-LT, CIFAR100-LT, and ImageNet-LT demonstrate that COCL substantially outperforms existing state-of-the-art OOD detection methods in LTR while being able to improve the classification accuracy on ID data. Code is available at https://github.com/mala-lab/COCL.

Published

2024-03-24

How to Cite

Miao, W., Pang, G., Bai, X., Li, T., & Zheng, J. (2024). Out-of-Distribution Detection in Long-Tailed Recognition with Calibrated Outlier Class Learning. Proceedings of the AAAI Conference on Artificial Intelligence, 38(5), 4216-4224. https://doi.org/10.1609/aaai.v38i5.28217

Issue

Section

AAAI Technical Track on Computer Vision IV