← Back to Computer Vision
cs.CV

Can diffusion models spot when medical images look wrong?

Alireza Kheirandish, Jihoon Hong, Sara Fridovich-Keil

May 29, 2026

Medical imaging and other inverse problems often encounter out-of-distribution data—like CT scans with tumors when trained only on healthy livers. Existing anomaly detection requires examples of the abnormality or operates only on full images. This work uses the KL divergence between a diffusion model's prior and posterior to flag both whole-image and localized anomalies without calibration data. The method detected tumor-bearing livers and generalizes across diffusion architectures, datasets, and imaging modalities.
Published as KLIP: localized distribution shift detection via KL-divergence with diffusion priors in Inverse Problems arXiv:2605.31596
Read the original paper →