# Contrastive Diffusion Model with Auxiliary Guidance for Coarse-to-Fine PET Reconstruction

arXiv: https://arxiv.org/abs/2308.10157 · HTML: https://mlsanity.com/paper/2308.10157/

## Key facts

- Cited by 18 papers in the map (data through 14 Oct 2026).
- Placed in Medical imaging and segmentation, one of 14 directions in Diffusion models.
- Builds on 15 of its 33 references that are in the map.
- First posted on arXiv 20 Aug 2023.

## Placed in

- Diffusion models / [Medical imaging and segmentation](https://mlsanity.com/diffusion/medical-imaging-and-segmentation/)

## Builds on (most cited first, 15 of 15)

- [Denoising Diffusion Probabilistic Models](https://mlsanity.com/paper/2006.11239/) (2020)
- [High-Resolution Image Synthesis with Latent Diffusion Models](https://mlsanity.com/paper/2112.10752/) (2021)
- [Diffusion Models Beat GANs on Image Synthesis](https://mlsanity.com/paper/2105.05233/) (2021)
- [Deep Unsupervised Learning using Nonequilibrium Thermodynamics](https://mlsanity.com/paper/1503.03585/) (2015)
- [Generative Adversarial Networks](https://mlsanity.com/paper/2203.00667/) (2022)
- [Improved Techniques for Training GANs](https://mlsanity.com/paper/1606.03498/) (2016)
- [Image Super-Resolution via Iterative Refinement](https://mlsanity.com/paper/2104.07636/) (2021)
- [WaveGrad: Estimating Gradients for Waveform Generation](https://mlsanity.com/paper/2009.00713/) (2020)
- [Solving Inverse Problems in Medical Imaging with Score-Based Generative Models](https://mlsanity.com/paper/2111.08005/) (2021)
- [Come-Closer-Diffuse-Faster: Accelerating Conditional Diffusion Models for Inverse Problems through Stochastic Contraction](https://mlsanity.com/paper/2112.05146/) (2021)
- [Deblurring via Stochastic Refinement](https://mlsanity.com/paper/2112.02475/) (2021)
- [Unrolled Generative Adversarial Networks](https://mlsanity.com/paper/1611.02163/) (2016)
- [Efficient Diffusion Models for Vision: A Survey](https://mlsanity.com/paper/2210.09292/) (2022)
- [DeepPET: A deep encoder-decoder network for directly solving the PET reconstruction inverse problem](https://mlsanity.com/paper/1804.07851/) (2018)
- [200x Low-dose PET Reconstruction using Deep Learning](https://mlsanity.com/paper/1712.04119/) (2017)

## Built on by (most cited first, 18 of 18)

- [Parameter-Efficient Fine-Tuning for Large Models: A Comprehensive Survey](https://mlsanity.com/paper/2403.14608/) (2024)
- [Language-driven Grasp Detection](https://mlsanity.com/paper/2406.09489/) (2024)
- [Likelihood-Scheduled Score-Based Generative Modeling for Fully 3D PET Image Reconstruction](https://mlsanity.com/paper/2412.04339/) (2024)
- [PET Image Reconstruction Using Deep Diffusion Image Prior](https://mlsanity.com/paper/2507.15078/) (2025)
- [Diffusion Models for Computational Neuroimaging: A Survey](https://mlsanity.com/paper/2502.06552/) (2025)
- [MCAD: Multi-modal Conditioned Adversarial Diffusion Model for High-Quality PET Image Reconstruction](https://mlsanity.com/paper/2406.13150/) (2024)
- [LegoPET: Hierarchical Feature Guided Conditional Diffusion for PET Image Reconstruction](https://mlsanity.com/paper/2411.16629/) (2024)
- [S3PET: Semi-supervised Standard-dose PET Image Reconstruction via Dose-aware Token Swap](https://mlsanity.com/paper/2407.20878/) (2024)
- [ConnectomeDiffuser: Generative AI Enables Brain Network Construction from Diffusion Tensor Imaging](https://mlsanity.com/paper/2505.22683/) (2025)
- [UniFuse: A Unified All-in-One Framework for Multi-Modal Medical Image Fusion Under Diverse Degradations and Misalignments](https://mlsanity.com/paper/2506.22736/) (2025)
- [FourierPET: Deep Fourier-based Unrolled Network for Low-count PET Reconstruction](https://mlsanity.com/paper/2601.11680/) (2026)
- [M2Diff: Multi-Modality Multi-Task Enhanced Diffusion Model for MRI-Guided Low-Dose PET Enhancement](https://mlsanity.com/paper/2603.09075/) (2026)
- [IgCONDA-PET: Weakly-Supervised PET Anomaly Detection using Implicitly-Guided Attention-Conditional Counterfactual Diffusion Modeling -- a Multi-Center, Multi-Cancer, and Multi-Tracer Study](https://mlsanity.com/paper/2405.00239/) (2024)
- [Contrastive-Adversarial and Diffusion: Exploring pre-training and fine-tuning strategies for sulcal identification](https://mlsanity.com/paper/2405.19204/) (2024)
- [Two-stage deep learning framework for the restoration of incomplete-ring PET images](https://mlsanity.com/paper/2504.00816/) (2025)
- [Harnessing EHRs for Diffusion-based Anomaly Detection on Chest X-rays](https://mlsanity.com/paper/2505.17311/) (2025)
- [Double-Constraint Diffusion Model with Nuclear Regularization for Ultra-low-dose PET Reconstruction](https://mlsanity.com/paper/2509.00395/) (2025)
- [Meta-information Guided Cross-domain Synergistic Diffusion Model for Low-dose PET Reconstruction](https://mlsanity.com/paper/2512.22237/) (2025)

## Abstract (arXiv)

To obtain high-quality positron emission tomography (PET) scans while reducing radiation exposure to the human body, various approaches have been proposed to reconstruct standard-dose PET (SPET) images from low-dose PET (LPET) images. One widely adopted technique is the generative adversarial networks (GANs), yet recently, diffusion probabilistic models (DPMs) have emerged as a compelling alternative due to their improved sample quality and higher log-likelihood scores compared to GANs. Despite this, DPMs suffer from two major drawbacks in real clinical settings, i.e., the computationally expensive sampling process and the insufficient preservation of correspondence between the conditioning LPET image and the reconstructed PET (RPET) image. To address the above limitations, this paper presents a coarse-to-fine PET reconstruction framework that consists of a coarse prediction module (CPM) and an iterative refinement module (IRM). The CPM generates a coarse PET image via a deterministic process, and the IRM samples the residual iteratively. By delegating most of the computational overhead to the CPM, the overall sampling speed of our method can be significantly improved. Furthermore, two additional strategies, i.e., an auxiliary guidance strategy and a contrastive diffusion strategy, are proposed and integrated into the reconstruction process, which can enhance the correspondence between the LPET image and the RPET image, further improving clinical reliability. Extensive experiments on two human brain PET datasets demonstrate that our method outperforms the state-of-the-art PET reconstruction methods. The source code is available at urlhttps://github.com/Show-han/PET-Reconstruction.
