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

Out-of-Distribution Detection with a Single Unconditional Diffusion Model

About

Out-of-distribution (OOD) detection is a critical task in machine learning that seeks to identify abnormal samples. Traditionally, unsupervised methods utilize a deep generative model for OOD detection. However, such approaches require a new model to be trained for each inlier dataset. This paper explores whether a single model can perform OOD detection across diverse tasks. To that end, we introduce Diffusion Paths (DiffPath), which uses a single diffusion model originally trained to perform unconditional generation for OOD detection. We introduce a novel technique of measuring the rate-of-change and curvature of the diffusion paths connecting samples to the standard normal. Extensive experiments show that with a single model, DiffPath is competitive with prior work using individual models on a variety of OOD tasks involving different distributions. Our code is publicly available at https://github.com/clear-nus/diffpath.

Alvin Heng, Alexandre H. Thiery, Harold Soh• 2024

Related benchmarks

TaskDatasetResultRank
Out-of-Distribution DetectionFMNIST vs. MNIST
AUROC (%)99.1
11
OOD DetectionMERRA2
Accuracy98.9
6
OOD DetectionBrain
Accuracy78.9
6
OOD DetectionNS-MIX
Accuracy78.8
6
OOD DetectionWave
Accuracy86.4
6
OOD DetectionNS-PwC
Accuracy (ACC)98.8
6
Off-Manifold DetectionTemperature Forecasting Hotspots
AUROC0.989
4
Off-Manifold DetectionStyle Transfer vs KMNIST
AUROC0.929
4
Showing 8 of 8 rows

Other info

Follow for update