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

基于潜在扩散模型的医学MRI图像时空超分辨率

Temporal and Spatial Super Resolution with Latent Diffusion Model in Medical MRI images

Vishal Dubey

AI总结:

针对医学成像中时空分辨率受限的问题,提出结合VQGAN编解码器的潜在扩散模型实现联合时空超分辨率,在心脏MRI数据集上多项指标优于基线,为医学影像超分提供新方案。

摘要:

超分辨率(SR)在计算机视觉中发挥着关键作用,尤其是在医学成像领域——硬件和采集时间的限制往往会导致空间和时间分辨率较低。尽管扩散模型已被应用于空间和时间超分辨率任务,但很少有研究探索其在联合时空超分辨率中的应用,在医学成像领域的相关研究更是匮乏。在本研究中,我们针对这一空白,提出将潜在扩散模型(Latent Diffusion Model, LDM)与基于向量量化生成对抗网络(Vector Quantised GAN, VQGAN)的编解码器架构相结合,用于联合超分辨率任务。我们将超分辨率建模为图像去噪问题,重点提升医学图像的空间和时间分辨率。我们使用数据科学杯心脏挑战赛(Data Science Bowl Cardiac Challenge)的心脏MRI数据集验证了所提方法的有效性,该数据集由2D电影图像组成,空间分辨率为256x256,每个时间步包含8-14个切片。我们的LDM模型取得了30.37的峰值信噪比(Peak Signal to Noise Ratio, PSNR)、0.7580的结构相似性指数(Structural Similarity Index, SSIM)以及0.2756的学习感知图像块相似度(Learned Perceptual Image Patch Similarity, LPIPS),在PSNR上比简单基线方法高出5%,SSIM高出6.5%,LPIPS高出39%。该LDM模型仅需15步扩散即可生成具有高保真度和感知质量的图像。这些结果表明,LDM在推动医学成像超分辨率技术发展方面颇具前景,有望提升诊断准确性和患者预后效果。本研究同时公开了代码链接。

英文摘要:

Super Resolution (SR) plays a critical role in computer vision, particularly in medical imaging, where hardware and acquisition time constraints often result in low spatial and temporal resolution. While diffusion models have been applied for both spatial and temporal SR, few studies have explored their use for joint spatial and temporal SR, particularly in medical imaging. In this work, we address this gap by proposing to use a Latent Diffusion Model (LDM) combined with a Vector Quantised GAN (VQGAN)-based encoder-decoder architecture for joint super resolution. We frame SR as an image denoising problem, focusing on improving both spatial and temporal resolution in medical images. Using the cardiac MRI dataset from the Data Science Bowl Cardiac Challenge, consisting of 2D cine images with a spatial resolution of 256x256 and 8-14 slices per time-step, we demonstrate the effectiveness of our approach. Our LDM model achieves Peak Signal to Noise Ratio (PSNR) of 30.37, Structural Similarity Index (SSIM) of 0.7580, and Learned Perceptual Image Patch Similarity (LPIPS) of 0.2756, outperforming simple baseline method by 5% in PSNR, 6.5% in SSIM, 39% in LPIPS. Our LDM model generates images with high fidelity and perceptual quality with 15 diffusion steps. These results suggest that LDMs hold promise for advancing super resolution in medical imaging, potentially enhancing diagnostic accuracy and patient outcomes. Code link is also shared.

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

↑