用分层特征引导的扩散模型,提升PET图像重建的清晰度和细节。
LegoPET: Hierarchical Feature Guided Conditional Diffusion for PET Image Reconstruction
- 分层特征引导的条件扩散模型,精准对齐不同域数据
- 视觉质量与PSNR/SSIM指标均优于现有方法
- 适合医学影像重建研究者及需要高保真图像的临床应用
正电子发射断层扫描(PET)因其能可视化体内功能与生物过程,广泛用于癌症检测。传统方法通过迭代算法(如OSEM、MLEM)从原始数据(投影图)重建图像。近年来,深度学习方法直接将投影图映射为图像,展现出潜力。但基于回归或GAN的方法常导致图像过度平滑或引入伪影。条件扩散概率模型(cDPMs)虽可生成高真实感、可控图像,但在跨域(如投影图与图像域)重建中仍面临保持对应关系与收敛慢的问题。为此,本文提出LegoPET,一种分层特征引导的条件扩散模型,用于从投影图高保真重建PET图像。实验表明,LegoPET不仅提升了cDPM性能,还在视觉质量及像素级PSNR/SSIM指标上超越现有深度学习重建方法。代码已开源:https://github.com/yransun/LegoPET。
原文摘要 · Abstract (English)
Positron emission tomography (PET) is widely utilized for cancer detection due to its ability to visualize functional and biological processes in vivo. PET images are usually reconstructed from histogrammed raw data (sinograms) using traditional iterative techniques (e.g., OSEM, MLEM). Recently, deep learning (DL) methods have shown promise by directly mapping raw sinogram data to PET images. However, DL approaches that are regression-based or GAN-based often produce overly smoothed images or introduce various artifacts respectively. Image-conditioned diffusion probabilistic models (cDPMs) are another class of likelihood-based DL techniques capable of generating highly realistic and controllable images. While cDPMs have notable strengths, they still face challenges such as maintain correspondence and consistency between input and output images when they are from different domains (e.g., sinogram vs. image domain) as well as slow convergence rates. To address these limitations, we introduce LegoPET, a hierarchical feature guided conditional diffusion model for high-perceptual quality PET image reconstruction from sinograms. We conducted several experiments demonstrating that LegoPET not only improves the performance of cDPMs but also surpasses recent DL-based PET image reconstruction techniques in terms of visual quality and pixel-level PSNR/SSIM metrics. Our code is available at https://github.com/yransun/LegoPET.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。