arXiv:2602.23214cs.CVcs.LG2026-02中稿 · ICML

用双变量耦合与频域调制,解决医学影像重建中的伪影与偏差难题

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

  • 引入双变量反馈机制,使重建过程具有历史记忆,避免长期偏差
  • 通过频域同质化处理,将结构化伪影转为符合扩散模型假设的噪声
  • 在CT和MRI重建中实现更准更快,适合高精度医学影像应用

基于预训练生成模型的插件式扩散先验(PnPDP)框架在图像逆问题中表现强劲,但现有求解器(如基于HQS或近端梯度法)作为无记忆算子,仅依赖瞬时梯度更新估计,导致无法消除稳态偏差,尤其在严重污染下难以满足物理测量约束。为此,本文提出双耦合PnP扩散(DC-PnPDP),恢复经典对偶变量以提供积分反馈,逐步强制数据一致性和先验间的一致性。然而,这种严格的几何耦合引入了谱色化、结构化的累积对偶残差,违反了扩散先验的加性白高斯噪声(AWGN)假设,引发严重幻觉。为此,我们提出频谱同质化(SH)机制,在频域中将这些结构化残差转化为统计上合规的伪AWGN输入,有效对齐求解器优化轨迹与去噪器的有效统计流形。在CT与MRI重建上的大量实验表明,该方法解决了偏差-幻觉权衡,实现了最先进的保真度与显著加速的收敛速度。代码已开源。

原文摘要 · Abstract (English)

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

医学影像扩散模型图像重建双变量优化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。