用扩散模型重建PET图像,兼顾清晰度与真实感。
End-to-End PET Image Reconstruction via a Posterior-Mean Diffusion Model
- 基于后验均值扩散模型,从 sinogram 直接生成图像
- 在 PSNR(dB)、SSIM、NRMSE 上优于5个最新SOTA方法
- 适合追求高保真医学影像的临床与研究场景
正电子发射断层成像(PET)是一种功能成像技术,可可视化多种组织中的生化与生理过程。近年来,基于深度学习(DL)的方法在直接从 sinogram 映射到 PET 图像方面取得显著进展。然而,回归类 DL 模型常产生过度平滑的重建结果(低失真,低感知质量),而 GAN 和似然基后验采样模型则容易引入不良伪影(高失真,高感知质量),限制了其临床应用。为实现感知-失真之间的稳健权衡,我们提出后验均值去噪扩散模型(PMDM-PET),该方法基于近期建立的数学理论,探索扩散模型空间中感知-失真函数的闭式表达,用于从 sinogram 重建 PET 图像。具体而言,PMDM-PET 首先在最小均方误差(MSE)下获得后验均值的 PET 预测,再最优地将这些预测分布传输至真实 PET 图像分布。实验结果表明,PMDM-PET 不仅生成具有可能最小失真和最优感知质量的真实感 PET 图像,且在定性视觉评估和定量像素级指标 PSNR(dB)、SSIM、NRMSE 上均超越五个最新的 SOTA 深度学习基线方法。
原文摘要 · Abstract (English)
Positron Emission Tomography (PET) is a functional imaging modality that enables the visualization of biochemical and physiological processes across various tissues. Recently, deep learning (DL)-based methods have demonstrated significant progress in directly mapping sinograms to PET images. However, regression-based DL models often yield overly smoothed reconstructions lacking of details (i.e., low distortion, low perceptual quality), whereas GAN-based and likelihood-based posterior sampling models tend to introduce undesirable artifacts in predictions (i.e., high distortion, high perceptual quality), limiting their clinical applicability. To achieve a robust perception-distortion tradeoff, we propose Posterior-Mean Denoising Diffusion Model (PMDM-PET), a novel approach that builds upon a recently established mathematical theory to explore the closed-form expression of perception-distortion function in diffusion model space for PET image reconstruction from sinograms. Specifically, PMDM-PET first obtained posterior-mean PET predictions under minimum mean square error (MSE), then optimally transports the distribution of them to the ground-truth PET images distribution. Experimental results demonstrate that PMDM-PET not only generates realistic PET images with possible minimum distortion and optimal perceptual quality but also outperforms five recent state-of-the-art (SOTA) DL baselines in both qualitative visual inspection and quantitative pixel-wise metrics PSNR (dB)/SSIM/NRMSE.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。