arXiv:2507.23411cs.CV2025-07中稿 · Uncertainty for Sa…

用扩散轨迹检测医学影像异常,快且准。

Out-of-Distribution Detection in Medical Imaging via Diffusion Trajectories

  • 不依赖重建,用扩散过程轨迹曲率评分
  • 仅5步扩散即可准确识别异常,性能提升超18%
  • 预训练模型跨任务通用,适合实时诊断应用

在医学影像中,无监督的分布外(OOD)检测为识别极低发生率病灶提供了一种高效方案。与有监督方法相比,该方法无需标签,对数据不平衡具有天然鲁棒性。现有生成式方法多依赖似然估计或重构误差,但计算成本高、可靠性差,且当正常数据变化时需重新训练。为此,我们提出一种无需重构的OOD检测方法,利用基于Stein得分的去噪扩散模型(SBDDM)的前向扩散轨迹。通过估计的Stein得分捕捉轨迹曲率,实现仅需5次扩散步骤的精准异常评分。一个在大规模语义对齐医学数据集上预训练的SBDDM,可有效泛化至多个近域和远域OOD基准,达到当前最优性能,同时显著降低推理开销。相较于现有方法,其在近域和远域检测上分别实现最高达10.43%和18.10%的相对提升,是实现实时、可靠辅助诊断的实用模块。

原文摘要 · Abstract (English)

In medical imaging, unsupervised out-of-distribution (OOD) detection offers an attractive approach for identifying pathological cases with extremely low incidence rates. In contrast to supervised methods, OOD-based approaches function without labels and are inherently robust to data imbalances. Current generative approaches often rely on likelihood estimation or reconstruction error, but these methods can be computationally expensive, unreliable, and require retraining if the inlier data changes. These limitations hinder their ability to distinguish nominal from anomalous inputs efficiently, consistently, and robustly. We propose a reconstruction-free OOD detection method that leverages the forward diffusion trajectories of a Stein score-based denoising diffusion model (SBDDM). By capturing trajectory curvature via the estimated Stein score, our approach enables accurate anomaly scoring with only five diffusion steps. A single SBDDM pre-trained on a large, semantically aligned medical dataset generalizes effectively across multiple Near-OOD and Far-OOD benchmarks, achieving state-of-the-art performance while drastically reducing computational cost during inference. Compared to existing methods, SBDDM achieves a relative improvement of up to 10.43% and 18.10% for Near-OOD and Far-OOD detection, making it a practical building block for real-time, reliable computer-aided diagnosis.

医学影像异常检测扩散模型OOD检测

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。