arXiv:2503.00745eess.IVcs.CV2025-03被引 1

用最短路径优化扩散模型,让医学图像增强更快更准。

Geodesic Diffusion Models for Efficient Medical Image Enhancement

  • 基于费雪-罗指定理设计最优噪声路径,缩短生成过程
  • 6步采样即达顶尖效果,训练速度提升20-30倍
  • 适合医疗影像实时处理,代码开源可复现

扩散模型通过学习反向过程生成数据,其性能依赖于预设的噪声调度。从几何角度看,每个调度对应概率空间中从数据分布到高斯先验的唯一轨迹。然而,现有模型采用经验性调度,效率低下,需大量中间步骤,导致训练和采样计算成本高。为此,本文推导出基于费雪-罗定理的测地线噪声调度族,对应概率空间中最短路径。基于此,提出测地线扩散模型(GDM),显著降低分布变换所需能量,提升训练与采样效率。该效率支持条件生成中从中间分布开始采样,仅用6步即达当前最佳性能。在两类医学图像增强任务——CT去噪与MRI超分辨率上验证,结果表明:相比DDPM,GDM训练时间减少20-30倍,采样加速160-170倍;相比Fast-DDPM,训练时间减少4-6倍,采样加速1.6倍。这些优势推动GDM在高效模型开发与临床实时应用中的使用。代码已公开于 https://github.com/mirthAI/GDM-VE。

原文摘要 · Abstract (English)

Diffusion models generate data by learning to reverse a forward process, where samples are progressively perturbed with Gaussian noise according to a predefined noise schedule. From a geometric perspective, each noise schedule corresponds to a unique trajectory in probability space from the data distribution to a Gaussian prior. However, prior diffusion models rely on empirically chosen schedules that may not be optimal. This inefficiency necessitates many intermediate time steps, resulting in high computational costs during both training and sampling. To address this, we derive a family of geodesic noise schedules corresponding to the shortest paths in probability space under the Fisher-Rao metric. Based on these schedules, we propose Geodesic Diffusion Models (GDMs), which significantly improve training and sampling efficiency by minimizing the energy required to transform between probability distributions. This efficiency further enables sampling to start from an intermediate distribution in conditional image generation, achieving state-of-the-art results with as few as 6 steps. We evaluated GDM on two medical image enhancement tasks: CT image denoising and MRI image super-resolution. Experimental results show that GDM achieved state-of-the-art performance while reducing training time by 20- to 30-fold compared to Denoising Diffusion Probabilistic Models (DDPMs) and 4- to 6-fold compared to Fast-DDPM, and accelerating sampling by 160- to 170-fold and 1.6-fold, respectively. These gains support the use of GDM for efficient model development and real-time clinical applications. Our code is publicly available at: https://github.com/mirthAI/GDM-VE.

扩散模型医学图像高效生成测地线

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