arXiv:2601.07093cs.CVcs.AI2026-01

用小波结构先验控制扩散模型,提升全身低剂量PET图像去噪质量。

3D Wavelet-Based Structural Priors for Controlled Diffusion in Whole-Body Low-Dose PET Denoising

  • 通过小波变换注入3D频域结构先验,引导扩散模型去噪
  • 在1/20剂量下,PSNR提升1.21dB,结构失真降低
  • 适用于低剂量、全身体积PET成像,适合医学影像重建

低剂量正电子发射断层扫描(PET)虽降低患者辐射暴露,但噪声增加导致图像质量下降和诊断可靠性降低。尽管扩散模型具备强大去噪能力,其随机性难以保证解剖结构一致性,尤其在信号-噪声比低和全身体积成像场景下。本文提出基于小波的条件控制网络(WCC-Net),一种全3D扩散框架,通过小波表示引入显式频域结构先验,指导体积PET去噪。通过轻量级控制分支将小波结构引导注入预训练扩散主干,实现解剖结构与噪声的解耦,同时保持生成表达力和3D结构连续性。大量实验表明,WCC-Net持续优于CNN、GAN及扩散基线方法。在内部1/20剂量测试集上,相比强扩散基线,PSNR提升+1.21 dB,SSIM提升+0.008,同时降低结构失真(GMSD)和强度误差(NMAE)。此外,WCC-Net对未见过的剂量水平(1/50和1/4)具有鲁棒泛化能力,实现更优定量性能和更高的体积解剖一致性。

原文摘要 · Abstract (English)

Low-dose Positron Emission Tomography (PET) imaging reduces patient radiation exposure but suffers from increased noise that degrades image quality and diagnostic reliability. Although diffusion models have demonstrated strong denoising capability, their stochastic nature makes it challenging to enforce anatomically consistent structures, particularly in low signal-to-noise regimes and volumetric whole-body imaging. We propose Wavelet-Conditioned ControlNet (WCC-Net), a fully 3D diffusion-based framework that introduces explicit frequency-domain structural priors via wavelet representations to guide volumetric PET denoising. By injecting wavelet-based structural guidance into a frozen pretrained diffusion backbone through a lightweight control branch, WCC-Net decouples anatomical structure from noise while preserving generative expressiveness and 3D structural continuity. Extensive experiments demonstrate that WCC-Net consistently outperforms CNN-, GAN-, and diffusion-based baselines. On the internal 1/20-dose test set, WCC-Net improves PSNR by +1.21 dB and SSIM by +0.008 over a strong diffusion baseline, while reducing structural distortion (GMSD) and intensity error (NMAE). Moreover, WCC-Net generalizes robustly to unseen dose levels (1/50 and 1/4), achieving superior quantitative performance and improved volumetric anatomical consistency.

PET去噪扩散模型小波变换3D图像

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