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

WDM:用于高分辨率医学图像合成的3D小波扩散模型

WDM: 3D Wavelet Diffusion Models for High-Resolution Medical Image Synthesis

University of Basel(巴塞尔大学)

Paul Friedrich, Julia Wolleb, Florentin Bieder, Alicia Durrer, Philippe C. Cattin

AI总结:

提出基于小波的3D扩散模型WDM,通过在小波分解图像上应用扩散过程,实现高分辨率医学图像合成,在单GPU上训练并在BraTS和LIDC-IDRI数据集上达到SOTA性能。

摘要:

由于CT或MR扫描的三维特性,医学图像的生成建模是一项极具挑战性的任务。现有方法大多采用分块、分片或级联生成技术,以将高维数据适配到有限的GPU内存中。然而,这些方法可能会引入伪影,并可能限制模型在某些下游任务中的适用性。本工作提出了WDM,这是一个基于小波的医学图像合成框架,在小波分解后的图像上应用扩散模型。所提出的方法是一种简单而有效的方式,可将3D扩散模型扩展至高分辨率,并且能够在单个40 GB的GPU上进行训练。在BraTS和LIDC-IDRI上以128×128×128分辨率进行无条件图像生成的实验结果表明,与最近的GAN、Diffusion Models和Latent Diffusion Models相比,该方法实现了最先进的图像保真度(FID)和样本多样性(MS-SSIM)分数。我们提出的方法是唯一能够以256×256×256分辨率生成高质量图像的方法,优于所有对比方法。

英文摘要:

Due to the three-dimensional nature of CT- or MR-scans, generative modeling of medical images is a particularly challenging task. Existing approaches mostly apply patch-wise, slice-wise, or cascaded generation techniques to fit the high-dimensional data into the limited GPU memory. However, these approaches may introduce artifacts and potentially restrict the model's applicability for certain downstream tasks. This work presents WDM, a wavelet-based medical image synthesis framework that applies a diffusion model on wavelet decomposed images. The presented approach is a simple yet effective way of scaling 3D diffusion models to high resolutions and can be trained on a single \SI{40}{\giga\byte} GPU. Experimental results on BraTS and LIDC-IDRI unconditional image generation at a resolution of $128 \times 128 \times 128$ demonstrate state-of-the-art image fidelity (FID) and sample diversity (MS-SSIM) scores compared to recent GANs, Diffusion Models, and Latent Diffusion Models. Our proposed method is the only one capable of generating high-quality images at a resolution of $256 \times 256 \times 256$, outperforming all comparing methods.

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

↑