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

Feature Space Singularity for Out-of-Distribution Detection

About

Out-of-Distribution (OoD) detection is important for building safe artificial intelligence systems. However, current OoD detection methods still cannot meet the performance requirements for practical deployment. In this paper, we propose a simple yet effective algorithm based on a novel observation: in a trained neural network, OoD samples with bounded norms well concentrate in the feature space. We call the center of OoD features the Feature Space Singularity (FSS), and denote the distance of a sample feature to FSS as FSSD. Then, OoD samples can be identified by taking a threshold on the FSSD. Our analysis of the phenomenon reveals why our algorithm works. We demonstrate that our algorithm achieves state-of-the-art performance on various OoD detection benchmarks. Besides, FSSD also enjoys robustness to slight corruption in test data and can be further enhanced by ensembling. These make FSSD a promising algorithm to be employed in real world. We release our code at \url{https://github.com/megvii-research/FSSD_OoD_Detection}.

Haiwen Huang, Zhihan Li, Lulu Wang, Sishuo Chen, Bin Dong, Xinyu Zhou• 2020

Related benchmarks

TaskDatasetResultRank
Glaucoma Classificationretinal Glaucoma dataset (test)
Accuracy0.859
28
OOD DetectionRetinal Glaucoma images REFUGE (test)
AUROC76.42
28
Out-of-Distribution DetectionCIFAR10 vs. SVHN
AUROC99.5
12
Out-of-Distribution DetectionFMNIST vs. MNIST
AUROC (%)99.6
11
Out-of-Distribution DetectionRadar Facial Authentication Dataset ID OOD (test)
AUROC0.6192
10
Out-of-Distribution DetectionImageNet dogs vs. non-dogs
AUROC93.1
7
Out-of-Distribution DetectionCelebA non-blurry vs. blurry
AUROC78.3
7
Out-of-Distribution DetectionMS-1M vs. IJB-C
AUROC0.867
7
Out-of-Distribution DetectionBacteria Genome
AUROC74.8
6
Showing 9 of 9 rows

Other info

Code

Follow for update