Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Density of States Estimation for Out-of-Distribution Detection

About

Perhaps surprisingly, recent studies have shown probabilistic model likelihoods have poor specificity for out-of-distribution (OOD) detection and often assign higher likelihoods to OOD data than in-distribution data. To ameliorate this issue we propose DoSE, the density of states estimator. Drawing on the statistical physics notion of ``density of states,'' the DoSE decision rule avoids direct comparison of model probabilities, and instead utilizes the ``probability of the model probability,'' or indeed the frequency of any reasonable statistic. The frequency is calculated using nonparametric density estimators (e.g., KDE and one-class SVM) which measure the typicality of various model statistics given the training data and from which we can flag test points with low typicality as anomalous. Unlike many other methods, DoSE requires neither labeled data nor OOD examples. DoSE is modular and can be trivially applied to any existing, trained model. We demonstrate DoSE's state-of-the-art performance against other unsupervised OOD detectors on previously established ``hard'' benchmarks.

Warren R. Morningstar, Cusuh Ham, Andrew G. Gallagher, Balaji Lakshminarayanan, Alexander A. Alemi, Joshua V. Dillon• 2020

Related benchmarks

TaskDatasetResultRank
Out-of-Distribution DetectionCIFAR-10 vs SVHN (test)
AUROC97.3
146
Out-of-Distribution DetectionCIFAR-10 vs CIFAR-100 (test)
AUROC56.9
119
Out-of-Distribution DetectionCIFAR-10 (ID) vs SVHN (OOD) (test)
AUROC80.2
92
Out-of-Distribution DetectionCIFAR-10 (ID) vs Celeb-A (OOD)
AUROC97.6
79
Out-of-Distribution DetectionCELEBA (in-distribution)
AUROC (CIFAR-100)88.9
57
Out-of-Distribution DetectionSVHN (In) CelebA (Out) (test)
AUROC99.7
27
Out-of-Distribution DetectionCIFAR-10 In-Distribution
AUROC (SVHN)0.955
26
Out-of-Distribution DetectionSVHN In-Distribution
AUROC (CIFAR-10)99.9
18
Out-of-Distribution DetectionCelebA (In) CIFAR-10 (Out) (test)
AUROC0.717
18
Out-of-Distribution DetectionSVHN
AUROC vs C1096.7
9
Showing 10 of 14 rows

Other info

Follow for update