Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Exploring the Limits of Out-of-Distribution Detection

About

Near out-of-distribution detection (OOD) is a major challenge for deep neural networks. We demonstrate that large-scale pre-trained transformers can significantly improve the state-of-the-art (SOTA) on a range of near OOD tasks across different data modalities. For instance, on CIFAR-100 vs CIFAR-10 OOD detection, we improve the AUROC from 85% (current SOTA) to more than 96% using Vision Transformers pre-trained on ImageNet-21k. On a challenging genomics OOD detection benchmark, we improve the AUROC from 66% to 77% using transformers and unsupervised pre-training. To further improve performance, we explore the few-shot outlier exposure setting where a few examples from outlier classes may be available; we show that pre-trained transformers are particularly well-suited for outlier exposure, and that the AUROC of OOD detection on CIFAR-100 vs CIFAR-10 can be improved to 98.7% with just 1 image per OOD class, and 99.46% with 10 images per OOD class. For multi-modal image-text pre-trained transformers such as CLIP, we explore a new way of using just the names of outlier classes as a sole source of information without any accompanying images, and show that this outperforms previous SOTA on standard vision OOD benchmark tasks.

Stanislav Fort, Jie Ren, Balaji Lakshminarayanan• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)--
3518
Image ClassificationCIFAR-10 (test)--
906
Out-of-Distribution DetectionSUN OOD with ImageNet-1k In-distribution (test)
FPR@9538.56
159
Out-of-Distribution DetectionPlaces with ImageNet-1k OOD In-distribution (test)
FPR9555.14
99
Out-of-Distribution DetectionCIFAR-10 vs CIFAR-100 (test)--
93
Image ClassificationImageNet-100--
84
OOD DetectionImageNet-1K OOD (Average: OpenImage-O, Texture, iNaturalist, ImageNet-O) 1.0 (test)
AUROC88.96
61
Out-of-Distribution DetectionCIFAR100 (test)
AUROC77.7
57
Out-of-Distribution DetectionImageNet 1k V2 (test)
FPR@9528.02
40
Out-of-Distribution DetectionCIFAR10 (ID) vs SVHN (OOD)
AUROC99
37
Showing 10 of 49 rows

Other info

Code

Follow for update