Out-of-Distribution Detection in Long-Tailed Recognition with Calibrated Outlier Class Learning
This addresses a critical issue for machine learning systems in real-world scenarios with imbalanced data, though it is incremental as it builds on prior OOD detection methods.
The paper tackles the problem of out-of-distribution (OOD) detection in long-tailed recognition, where existing methods fail due to confusion between OOD samples and head/tail classes, and introduces a calibrated outlier class learning (COCL) approach that improves OOD detection performance and in-distribution classification accuracy on benchmarks like CIFAR10-LT, CIFAR100-LT, and ImageNet-LT.
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 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.