Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Plug-and-Play Diffusion Meets ADMM: Dual-Variable Coupling for Robust Medical Image Reconstruction

About

Plug-and-Play diffusion prior (PnPDP) frameworks have emerged as a powerful paradigm for solving imaging inverse problems by treating pretrained generative models as modular priors. However, we identify a critical flaw in prevailing PnP solvers (e.g., based on HQS or Proximal Gradient): they function as memoryless operators, updating estimates solely based on instantaneous gradients. This lack of historical tracking inevitably leads to non-vanishing steady-state bias, where the reconstruction fails to strictly satisfy physical measurements under heavy corruption. To resolve this, we propose Dual-Coupled PnP Diffusion (DC-PnPDP), which restores the classical dual variable to provide integral feedback, progressively enforce agreement between the data-consistency and prior. However, this rigorous geometric coupling introduces a secondary challenge: the accumulated dual residuals exhibit spectrally colored, structured artifacts that violate the Additive White Gaussian Noise (AWGN) assumption of diffusion priors, causing severe hallucinations. To bridge this gap, we introduce Spectral Homogenization (SH), a frequency-domain adaptation mechanism that modulates these structured residuals into statistically compliant pseudo-AWGN inputs. This effectively aligns the solver's rigorous optimization trajectory with the denoiser's valid statistical manifold. Extensive experiments on CT and MRI reconstruction demonstrate that our approach resolves the bias-hallucination trade-off, achieving state-of-the-art fidelity with significantly accelerated convergence. The code is available at https://github.com/duchenhe/DC-PnPDP

Chenhe Du, Xuanyu Tian, Qing Wu, Muyu Liu, Jingyi Yu, Hongjiang Wei, Yuyao Zhang• 2026

Related benchmarks

TaskDatasetResultRank
CT Image Reconstructionclinical GE LACT dataset (90◦ angular coverage)
SSIM0.955
14
CT ReconstructionSVCT-20
PSNR40.55
8
MRI ReconstructionfastMRI brain AF=6
PSNR36.43
8
MRI ReconstructionfastMRI brain (AF=10)
PSNR30.91
8
4x super-resolutionNatural Images
PSNR30.807
3
Gaussian DeblurringNatural Images
PSNR30.741
3
Limited-Angle CT ReconstructionAAPM
PSNR (90° range)33.8669
3
Limited-Angle CT ReconstructionDeepLesion
PSNR (90° range)33.0598
3
Motion DeblurringNatural Images
PSNR33.305
3
Sparse-View CT ReconstructionAAPM
PSNR (18 views)36.4784
3
Showing 10 of 11 rows

Other info

Follow for update