Back to the Basics: Revisiting Out-of-Distribution Detection Baselines

Code

• github.com/cleanlab/cleanlab
  OfficialIn paperpytorch★ 11,379
• github.com/cleanlab/ood-detection-benchmarks
  OfficialIn papermxnet★ 8

Abstract

We study simple methods for out-of-distribution (OOD) image detection that are compatible with any already trained classifier, relying on only its predictions or learned representations. Evaluating the OOD detection performance of various methods when utilized with ResNet-50 and Swin Transformer models, we find methods that solely consider the model's predictions can be easily outperformed by also considering the learned representations. Based on our analysis, we advocate for a dead-simple approach that has been neglected in other studies: simply flag as OOD images whose average distance to their K nearest neighbors is large (in the representation space of an image classifier trained on the in-distribution data).

Tasks

Reproductions