Generative restoration for cardiac PET attenuation correction: a two-stage 3D DDIM framework optimizing fidelity and clinical controllability
摘要
This study aims to develop a two-stage 3D denoising diffusion implicit model (DDIM) framework for CT-free attenuation correction in cardiac PET imaging, enabling direct generation of attenuation-corrected (AC) images from non-attenuation-corrected (NAC) PET scans. The method is comprehensively validated using both [18F]FDG PET and [13N]ammonia cardiac PET datasets to demonstrate its clinical applicability across different perfusion and metabolic imaging protocols.
MethodsThe framework employs a two-stage approach: (1) a noise-to-image DDIM was first pretrained on all available AC images (i.e., no need of paired NACs) to learn a diverse AC distributions, enabling the high-fidelity generation of AC images with varying appearances; (2) the pretrained model was fine-tuned with a limited set of paired NAC-AC images to form a conditional DDIM, ensuring anatomically aligned, controllable generation. The model architecture uses a 3D U-Net, trained on 224 paired NAC-AC and 396 unpaired AC images for [18F]FDG, and 608 paired NAC-AC images and 885 unpaired AC images for [13N]ammonia. Performance was evaluated through quantitative metrics (including NMAE, NRMSE, SSIM and PSNR) and visual assessment.
ResultsThe proposed two-stage DDIM framework achieved excellent agreement with clinical CT-based attenuation correction (CT-AC), demonstrating superior correlation (slope = 0.78,
The two-stage 3D DDIM framework achieves performance comparable to clinical CT-AC while effectively leveraging unpaired data, demonstrating significant potential for robust cardiac PET attenuation correction.