arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.03156cs.CVcs.LGphysics.med-ph

基于浅层扩散模型潜变量优化的迭代CT重建

Iterative CT Reconstruction via Latent Variable Optimization of Shallow Diffusion Models

Hirosaki University(弘前大学);Kyushu University(九州大学);University of Tokyo Hospital(东京大学医院);Osaka Medical and Pharmaceutical University(大阪医科药科大学)

Sho Ozaki, Shizuo Kaji, Toshikazu Imae, Kanabu Nawa, Hideomi Yamashita, Keiichi Nakagawa

AI总结:

本研究提出一种结合浅层扩散模型与迭代CT重建的新方法,通过优化扩散模型潜变量的保真度损失、固定逆过程噪声实现解剖结构保留,在稀疏投影CT重建任务上的定量指标优于现有方法,且可推广至多成像模态。

摘要:

图像生成式人工智能(AI)近年来受到广泛关注,其中作为生成式AI核心组件的扩散模型能够生成兼具高质量与丰富多样性的图像。本研究提出了一种将去噪扩散概率模型与迭代CT重建相结合的新型计算机断层扫描(CT)重建方法。与以往研究形成鲜明对比的是,我们针对扩散模型的潜变量而非图像和模型参数,优化CT重建的保真度损失。为抑制扩散模型导致的解剖结构改变,我们对扩散过程和逆过程进行了浅层化处理,并在逆过程中固定一组添加的噪声,使其在推理阶段具备确定性。我们通过使用1/10投影数据的稀疏投影CT重建实验,验证了所提方法的有效性。尽管实现方式简单,该方法有望在保留患者解剖结构的同时重建高质量图像,且在结构相似性指数、峰值信噪比等定量指标上优于现有方法,包括迭代重建、结合全变分的迭代重建以及单独使用扩散模型的方法。我们还使用同一训练好的扩散模型,探索了采用1/20投影数据的更稀疏投影CT重建场景。随着迭代次数增加,图像质量提升至与1/10稀疏投影CT重建相当的水平。原则上,该方法不仅可广泛应用于CT,还可推广至其他成像模态。

英文摘要:

Image-generative artificial intelligence (AI) has garnered significant attention in recent years. In particular, the diffusion model, a core component of generative AI, produces high-quality images with rich diversity. In this study, we proposed a novel computed tomography (CT) reconstruction method by combining the denoising diffusion probabilistic model with iterative CT reconstruction. In sharp contrast to previous studies, we optimized the fidelity loss of CT reconstruction with respect to the latent variable of the diffusion model, instead of the image and model parameters. To suppress the changes in anatomical structures produced by the diffusion model, we shallowed the diffusion and reverse processes and fixed a set of added noises in the reverse process to make it deterministic during the inference. We demonstrated the effectiveness of the proposed method through the sparse-projection CT reconstruction of 1/10 projection data. Despite the simplicity of the implementation, the proposed method has the potential to reconstruct high-quality images while preserving the patient's anatomical structures and was found to outperform existing methods, including iterative reconstruction, iterative reconstruction with total variation, and the diffusion model alone in terms of quantitative indices such as the structural similarity index and peak signal-to-noise ratio. We also explored further sparse-projection CT reconstruction using 1/20 projection data with the same trained diffusion model. As the number of iterations increased, the image quality improved comparable to that of 1/10 sparse-projection CT reconstruction. In principle, this method can be widely applied not only to CT but also to other imaging modalities.

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

↑