EigenScore: OOD Detection using Covariance in Diffusion Models
By: Shirin Shoushtari , Yi Wang , Xiao Shi and more
Potential Business Impact:
Finds fake data that could trick smart computers.
Out-of-distribution (OOD) detection is critical for the safe deployment of machine learning systems in safety-sensitive domains. Diffusion models have recently emerged as powerful generative models, capable of capturing complex data distributions through iterative denoising. Building on this progress, recent work has explored their potential for OOD detection. We propose EigenScore, a new OOD detection method that leverages the eigenvalue spectrum of the posterior covariance induced by a diffusion model. We argue that posterior covariance provides a consistent signal of distribution shift, leading to larger trace and leading eigenvalues on OOD inputs, yielding a clear spectral signature. We further provide analysis explicitly linking posterior covariance to distribution mismatch, establishing it as a reliable signal for OOD detection. To ensure tractability, we adopt a Jacobian-free subspace iteration method to estimate the leading eigenvalues using only forward evaluations of the denoiser. Empirically, EigenScore achieves SOTA performance, with up to 5% AUROC improvement over the best baseline. Notably, it remains robust in near-OOD settings such as CIFAR-10 vs CIFAR-100, where existing diffusion-based methods often fail.
Similar Papers
Improving Out-of-Distribution Detection via Dynamic Covariance Calibration
CV and Pattern Recognition
Helps AI spot fake or wrong information.
Probability Density from Latent Diffusion Models for Out-of-Distribution Detection
Machine Learning (CS)
Helps AI know when it sees something new.
Evidential Spectrum-Aware Contrastive Learning for OOD Detection in Dynamic Graphs
Machine Learning (CS)
Finds weird changes in online connections.