MAP-Diff: Multi-Anchor Guided Diffusion for Progressive 3D Whole-Body Low-Dose PET Denoising
Authors: Peiyuan Jing, Chun-Wun Cheng, Liutao Yang, Zhenxuan Zhang, Thiago V. Lima, Klaus Strobel, Antoine Leimgruber, Angelica Aviles-Rivero, Guang Yang, Javier A. Montoya-Zegarra
Organizations: School of Engineering, Zurich University of Applied Sciences, CH · Bioengineering Department and Imperial-X, Imperial College London, UK · DAMTP, University of Cambridge, UK · Lucerne University Teaching and Research Hospital, CH · Yau Mathematical Sciences Center, Tsinghua University, CN · National Heart and Lung Institute, Imperial College London, UK · Cardiovascular Research Centre, Royal Brompton Hospital, UK · School of Biomedical Engineering & Imaging Sciences, King’s College London, UK
Low-dose Positron Emission Tomography (PET) reduces radiation exposure but suffers from severe noise and quantitative degradation. Diffusion-based denoising models achieve strong final reconstructions, yet their reverse trajectories are typically unconstrained and not aligned with the progressive nature of PET dose formation. We propose MAP-Diff, a multi-anchor guided diffusion framework for progressive 3D whole-body PET denoising. MAP-Diff introduces clinically observed intermediate-dose scans as trajectory anchors and enforces timestep-dependent supervision to regularize the reverse process toward dose-aligned intermediate states. Anchor timesteps are calibrated via degradation matching between simulated diffusion corruption and real multi-dose PET pairs, and a timestep-weighted anchor loss stabilizes stage-wise learning. At inference, the model requires only ultra-low-dose input while enabling progressive, dose-consistent intermediate restoration. Experiments on internal (Siemens Biograph Vision Quadra) and cross-scanner (United Imaging uEXPLORER) datasets show consistent improvements over strong CNN-, Transformer-, GAN-, and diffusion-based baselines. On the internal dataset, MAP-Diff improves PSNR from 42.48 dB to 43.71 dB (+1.23 dB), increases SSIM to 0.986, and reduces NMAE from 0.115 to 0.103 (-0.012) compared to 3D DDPM. Performance gains generalize across scanners, achieving 34.42 dB PSNR and 0.141 NMAE on the external cohort, outperforming all competing methods.
Figures & tables
Figure 1: Overview of MAP-Diff. (a) Conceptual comparison between standard diffusion training with end-point supervision and the proposed multi-anchor trajectory-level supervision. (b) MAP-Diff training strategy with timestep zoning and anchor-weighted loss. (c) Progressive inference trajectory enabling dose-consistent intermediate outputs.
Method
Internal (University Hospital of Bern)
External (Shanghai Ruijin Hospital)
PSNR ↑
SSIM ↑
NMAE ↓
PSNR ↑
SSIM ↑
NMAE ↓
Low-dose Input
38.545 ± 1.798**
0.948 ± 0.018**
0.199 ± 0.036**
30.402 ± 3.475**
0.896 ± 0.033**
0.257 ± 0.059**
3D cGAN ( Wang et al., 2018 )
40.721 ± 1.293**
0.977 ± 0.006**
0.127 ± 0.015**
33.062 ± 3.624**
0.927 ± 0.053**
0.174 ± 0.031**
3D UNet ( Çiçek et al., 2016 )
41.810 ± 1.377**
0.972 ± 0.007**
0.120 ± 0.015**
33.654 ± 3.810**
0.904 ± 0.074**
0.164 ± 0.031**
RED-CNN ( Chen et al., 2017 )
42.185 ± 1.427**
0.981 ± 0.005**
0.119 ± 0.016**
33.989 ± 3.725**
0.945 ± 0.025**
0.161 ± 0.030**
EDCNN ( Liang et al., 2020 )
42.355 ± 1.579**
0.983 ± 0.006**
0.116 ± 0.017**
33.841 ± 3.615**
0.942 ± 0.023**
0.164 ± 0.024**
Table 1: Quantitative comparison results. Statistical significance is assessed using a two-sided Wilcoxon signed-rank test with Holm correction. * and ** denote statistically significant differences compared with MAP-Diff at p<0.05 and p<0.01 , respectively.
Figure 2: Visual comparison of PET denoising across methods.
Figure 3: Intermediate-dose reconstruction accuracy at anchor timesteps. (a) Quantitative comparison against paired real-dose references at the 1/10, 1/4, and 1/2 dose levels. (b) Coronal-view signed error maps against paired real-dose references.
Set
#A
Anchors
W
PSNR
SSIM
NMAE
Set
#A
Anchors
W
PSNR
SSIM
NMAE
Baseline 3D DDPM ( Yu et al., 2025 ) (No Anchors): PSNR: 42.483, SSIM: 0.983, NMAE: 0.115
S1
3
{1/10,1/4,1/2}
Poly
43.710
0.986
0.103
S2
2
{1/10,1/4}
Poly
43.672
0.984
0.104
S3
2
{1/10,1/2}
Poly
43.521
0.983
0.106
S4
2
{1/4,1/2}
Poly
43.414
0.982
0.107
S5
1
{1/10}
Poly
43.503
0.983
0.106
S6
1
{1/4}
Poly
43.267
0.982
0.109
S7
1
{1/2}
Poly
42.790
0.981
0.111
S8
3
{1/10,1/4,1/2}
Const
41.856
0.905
0.133
Table 2: Ablation study of anchor supervision design in MAP-Diff, evaluating the number of anchors and timestep-weighted loss. Best results in bold.
Positron emission tomography (PET) provides essential functional information for disease assessment, however reducing injected activity or acquisition time produces low-dose (LD) PET with stronger count dependent noise and less reliable uptake quantification. Diffusion models offer a promising solution for PET denoising by progressively recovering high-dose (HD) PET images from LD inputs. However, LD-to-HD PET denoising is still challenging due to insufficient anatomical guidance, unstable multi-scale feature propagation, and uncertain frequency domain uptake recovery. We propose AnF-DiffPET, an anatomy- and frequency-guided diffusion framework for computed tomography (CT) conditioned LD PET denoising. The framework integrates Anatomical-Frequency Guidance (AFG), Multi-Scale Cross-Transformer Reconstruction (MSCTR), and Frequency-Contrastive Hard Mining (FCHM) to enhance anatomy aware feature modulation and frequency domain consistency during denoising. Experimental results across four PET/CT datasets show that the proposed method improves image fidelity, anatomical consistency, and quantitative fidelity over representative CNN-based, GAN-based, transformer-based, and diffusion-based methods. The code and trained models will be publicly released upon acceptance.
Xuepeng Liu, Ruili Li, Zetong Liu +5
College of Medicine and Biological Information Engineering, Northeastern University, Shenyang 110169, China · Tohoku University School of Medicine, Sendai 980-8575, Japan · Department of Clinical Neurosciences, University of Cambridge, UK
Accurate quantification and uptake measurement in PET are critical for assessing disease progression and supporting clinical decision-making. While high-count PET provides reliable image quality, the associated radiation dose and prolonged acquisition remain significant clinical concerns, motivating the adoption of low-count protocols. Diffusion-model-based methods have demonstrated strong potential for restoring low-count PET to near high-count quality, but their iterative sampling procedure becomes prohibitively expensive when applied to high-resolution 3D PET volumes, introducing substantial inference latency that limits practical clinical deployment. To address these challenges, we propose a training-free Global-Local Skipping Strategy that accelerates diffusion model-based 3D PET denoising while simultaneously improving reconstruction quality. The proposed method is plug-and-play and directly applicable to pre-trained diffusion models without retraining or architectural modification. Specifically, we introduce: (i) a global denoising step skipping strategy that initializes the reverse diffusion process from an intermediate denoising step using a noise-consistent transformation of the low-count input, substantially reducing the number of required denoising steps; and (ii) a local feature reuse shortcut that reuses slowly-varying high-level U-Net features across neighboring denoising steps, further reducing per-step computation while preserving image fidelity. We evaluate the proposed approach on multiple PET tracers from in-house and public datasets, including 18F-FDG PET, 68Ga-DOTATATE PET, and 18F-PSMA PET, demonstrating consistent acceleration of over an order of magnitude alongside improved or comparable reconstruction performance relative to the full-step baseline. Blinded reader studies further confirm enhanced clinical confidence and perceived diagnostic quality.
Yuhan Liu, Scott M. Leonard, Marlee Crews +5
Department of Radiology, Northwestern University, Chicago, IL, USA
Reducing radiation exposure in Positron Emission Tomography (PET) is important for patient safety; however, ultra-low-dose imaging suffers from severe noise, which may affect diagnostic interpretation without appropriate image enhancement. While current 3D deep generative models, particularly diffusion models, have shown strong reconstruction fidelity, their practical use can be limited by long inference times. In contrast, faster 2D-based alternatives may have difficulty maintaining volumetric consistency, an important consideration for whole-body PET imaging analysis. To bridge this gap, we propose a one-pass conditional 3D rectified flow (3D Flow) framework for whole-body PET image denoising that incorporates a novel optimized non-uniform sampling strategy. The model is trained with a one-pass linear-interpolant velocity-matching objective. This approach reconstructs a full 3D volume in approximately 30 seconds in our implementation, compared with multi-hour inference for the evaluated 3D DDPM baseline. Evaluations including zero-shot transfer to an independent clinical dataset show that our model achieves favorable global image quality and lesion conspicuity compared with the evaluated 3D DDPM and DDIM baselines, including on challenging short-acquisition data. Furthermore, the proposed method shows promising zero-shot transfer performance across the evaluated datasets and unseen dose levels (down to 1/100 of the standard dose), with artifact-focused visual comparisons supporting the need for further lesion-level validation. By balancing reconstruction fidelity and computational efficiency, this work presents a candidate approach for ultra-low-dose whole-body PET image denoising.
Jiale Shen, Guolin Wang, Chenhao Wang +3
College of Biomedical Engineering & Instrument Science, Zhejiang University, Zhejiang, 310063, China