arXiv:2606.24313cs.AI2026-06被引 2

用4步扩散模型高效生成高质量MRI图像,助力医疗影像多模态分析。

Prob-BBDM: a Probabilistic Brownian Bridge Diffusion Model for MRI sequence image-to-image translation

论文配图:Prob-BBDM: a Probabilistic Brownian Bridge Diffusion Model for MRI sequence image-to-image translation
图 1 · 摘自论文原文
  • 基于布朗桥扩散机制,融合变分编码引导提升生成质量。
  • 在BraTS数据集上达到88.46% SSIM、26.09 dB PSNR,4步完成推理。
  • 生成图像可直接用于肿瘤分割,Dice达88.71%,适合临床实用。

基于布朗桥扩散模型(BBDM)的新型图像到图像翻译方法Prob-BBDM,可从2D轴向切片合成磁共振成像(MRI)序列。该方法结合变分编码器引导的扩散机制,利用概率图像分布提升合成质量。在BraTS 2021数据集上,Prob-BBDM在多个翻译任务中表现优异,最高达88.46% SSIM和26.09 dB PSNR,且仅需4步扩散过程,计算效率高。在第三方外部数据集上验证了其跨域泛化能力。进一步评估显示,合成图像输入预训练分割模型后,肿瘤分割获得88.71% Dice分数和3.49 mm HD95,证明其保留关键诊断信息。结果表明,Prob-BBDM在高质量、高效、通用的MRI合成方面具有潜力,为医学图像翻译提供新路径。

原文摘要 · Abstract (English)

AI-driven image-to-image synthesis is rapidly advancing, with growing applications in medical imaging. Multi-modal image analysis plays a crucial role in optimizing examination quality, yet acquiring multiple imaging modalities in clinical settings remains resource-intensive and time-consuming, especially for 3D imaging. To address this challenge, we propose a novel image-to-image translation model based on Brownian Bridge Diffusion Models (BBDM), which synthesizes magnetic resonance imaging (MRI) sequences from 2D axial slices. Our approach integrates a variational encoder-guided diffusion mechanism, leveraging probabilistic image distributions to enhance synthesis quality. Evaluated on the BraTS 2021 dataset, our Probabilistic-BBDM (Prob-BBDM) achieves superior performance across multiple translation tasks, reaching up to 88.46% SSIM and 26.09 dB PSNR, with consistent improvements over baselines. Notably, our diffusion process requires only 4 steps, making it computationally efficient while maintaining high-quality synthesis. To further validate generalizability, we test Prob-BBDM on an external third-party dataset, demonstrating consistent performance across domains. Additionally, we assess the clinical utility of the synthesized slices by using them as input to a pre-trained segmentation model. Tumor segmentation yields a Dice score of 88.71% and an HD95 of 3.49 mm, confirming that the synthesized slices preserve critical diagnostic information. These results highlight the potential of Prob-BBDM for high-quality, efficient, and generalizable MRI synthesis, offering a promising step toward improved medical image translation.

MRI生成扩散模型图像翻译医疗影像

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