RED: Residual Estimation Diffusion for Low-Dose PET Sinogram Reconstruction
About
Recent advances in diffusion models have demonstrated exceptional performance in generative tasks across vari-ous fields. In positron emission tomography (PET), the reduction in tracer dose leads to information loss in sino-grams. Using diffusion models to reconstruct missing in-formation can improve imaging quality. Traditional diffu-sion models effectively use Gaussian noise for image re-constructions. However, in low-dose PET reconstruction, Gaussian noise can worsen the already sparse data by introducing artifacts and inconsistencies. To address this issue, we propose a diffusion model named residual esti-mation diffusion (RED). From the perspective of diffusion mechanism, RED uses the residual between sinograms to replace Gaussian noise in diffusion process, respectively sets the low-dose and full-dose sinograms as the starting point and endpoint of reconstruction. This mechanism helps preserve the original information in the low-dose sinogram, thereby enhancing reconstruction reliability. From the perspective of data consistency, RED introduces a drift correction strategy to reduce accumulated prediction errors during the reverse process. Calibrating the inter-mediate results of reverse iterations helps maintain the data consistency and enhances the stability of reconstruc-tion process. Experimental results show that RED effec-tively improves the quality of low-dose sinograms as well as the reconstruction results. The code is available at: https://github.com/yqx7150/RED.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| PET Reconstruction | In-House (1% Count) | SSIM0.9472 | 10 | |
| PET Reconstruction | UDPET DRF-100 (1% Count) | SSIM0.889 | 10 | |
| PET Reconstruction | BrainWeb 20% Count | SSIM96.64 | 10 | |
| PET Reconstruction | BrainWeb 40% Count simulated (test) | SSIM0.9599 | 10 | |
| PET Reconstruction | In-House 10% Count real (test) | SSIM0.9501 | 10 |