English

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

Computer Vision and Pattern Recognition 2026-05-07 v1

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 .

Keywords

Cite

@article{arxiv.2605.05161,
  title  = {Wasserstein-Aligned Localisation for VLM-Based Distributional OOD Detection in Medical Imaging},
  author = {Bernhard Kainz and Johanna P Mueller and Matthew Baugh and Cosmin Bercea},
  journal= {arXiv preprint arXiv:2605.05161},
  year   = {2026}
}

Comments

submitted to MICCAI 2026