arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.12223eess.IVcs.CV

用于医学图像翻译的级联多路径捷径扩散模型

Cascaded Multi-path Shortcut Diffusion Model for Medical Image Translation

Yale University(耶鲁大学);University of California Irvine(加州大学欧文分校);University of Science and Technology of China(中国科学技术大学);Case Western Reserve University(凯斯西储大学);Northwestern University(西北大学);Yale School of Medicine(耶鲁医学院)

Yinchi Zhou, Tianqi Chen, Jun Hou, Huidong Xie, Nicha C. Dvornek, S. Kevin Zhou, David L. Wilson, James S. Duncan, Chi Liu, Bo Zhou

AI总结:

针对现有医学图像翻译方法不稳定、缺乏不确定性估计的问题,提出结合条件GAN与多路径捷径扩散的级联模型CMDM,在三个数据集上实现了媲美SOTA的翻译质量与可靠的不确定性估计。

摘要:

图像到图像翻译是医学影像处理中的关键组成部分,在多种成像模态和临床场景中具有广泛应用。现有方法包括生成对抗网络(GANs)和扩散模型(DMs),这类方法能生成真实感结果,但存在训练不稳定、缺乏不确定性估计的问题。尽管GAN和DM方法各自已在医学图像翻译任务中展现出能力,但将二者结合以进一步提升翻译性能、实现不确定性估计的潜力仍未得到充分探索。在本研究中,我们针对这些挑战提出了级联多路径捷径扩散模型(Cascade Multi-path Shortcut Diffusion Model, CMDM),用于实现高质量医学图像翻译与不确定性估计。为减少所需迭代次数并保证性能鲁棒性,我们的方法首先生成由条件GAN得到的先验图像,用于后续步骤中DM的高效反向翻译。此外,该方法采用多路径捷径扩散策略优化翻译结果并估计不确定性,级联流水线还通过级联间的残差平均进一步提升翻译质量。我们收集了三个不同的医学图像数据集,每个数据集包含两个子任务,以测试所提方法的泛化性。实验结果表明,CMDM能够生成与当前最优方法相当的高质量翻译结果,同时提供与翻译误差高度相关的合理不确定性估计。

英文摘要:

Image-to-image translation is a vital component in medical imaging processing, with many uses in a wide range of imaging modalities and clinical scenarios. Previous methods include Generative Adversarial Networks (GANs) and Diffusion Models (DMs), which offer realism but suffer from instability and lack uncertainty estimation. Even though both GAN and DM methods have individually exhibited their capability in medical image translation tasks, the potential of combining a GAN and DM to further improve translation performance and to enable uncertainty estimation remains largely unexplored. In this work, we address these challenges by proposing a Cascade Multi-path Shortcut Diffusion Model (CMDM) for high-quality medical image translation and uncertainty estimation. To reduce the required number of iterations and ensure robust performance, our method first obtains a conditional GAN-generated prior image that will be used for the efficient reverse translation with a DM in the subsequent step. Additionally, a multi-path shortcut diffusion strategy is employed to refine translation results and estimate uncertainty. A cascaded pipeline further enhances translation quality, incorporating residual averaging between cascades. We collected three different medical image datasets with two sub-tasks for each dataset to test the generalizability of our approach. Our experimental results found that CMDM can produce high-quality translations comparable to state-of-the-art methods while providing reasonable uncertainty estimations that correlate well with the translation error.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑