cs.CVMay 6, 2026

Wasserstein-Aligned Localisation for VLM-Based Distributional OOD Detection in Medical Imaging

Authors: Bernhard KainzJohanna P MuellerMatthew BaughCosmin Bercea

Organizations: Dept. AIBE, Friedrich–Alexander University Erlangen–Nürnberg, DE · Department of Computing, Imperial College London, UK · Technical University Munich, DE · Munich Center for Machine Learning (MCML), DE

Abstract

Zero-shot anomaly localisation via vision-language models (VLMs) offers a compelling approach for rare pathology detection, yet its performance is fundamentally limited by the absence of healthy anatomical context. We reformulate zero-shot localisation as a comparative inference problem in which anomalies are identified through structured comparison against reference distributions of normal anatomy. We introduce WALDO, a training-free framework grounded in optimal transport theory that enables comparative reasoning through: (i) entropy-weighted Sliced Wasserstein distances for anatomically-aware reference selection from DINOv2 patch distributions, (ii) Goldilocks zone sampling exploiting the non-monotonic relationship between reference similarity and localisation accuracy, and (iii) self-consistency aggregation via weighted non-maximum suppression. We theoretically analyse the Goldilocks effect through distributional divergence, and show that references with moderate similarity minimize a bias-variance trade-off in comparative visual reasoning. On the NOVA brain MRI benchmark, WALDO with Qwen2.5-VL-72B achieves 43.5±1.6%43.5_{\pm1.6}\% mAP@30 (95% CI: [40.4, 46.7]), representing a 19% relative improvement over zero-shot baselines. Cross-model evaluation shows consistent gains: GPT-4o achieves 32.0±6.5%32.0_{\pm6.5}\% and Qwen3-VL-32B achieves 32.0±6.6%32.0_{\pm6.6}\% mAP@30. Paired McNemar tests confirm statistical significance (p<0.01p<0.01). Source code is available at https://github.com/bkainz/WALDO_MICCAI26_demo .

Explore similar work

CardsList