arXiv:2508.01441eess.IV2025-08

用动态调和方法让图像重建更稳定,避免后期质量下降。

Viscosity Stabilized Plug-and-Play Reconstruction

  • 用收缩算子与不稳定的去噪器自适应加权,实现动态稳定
  • 在多种算法和模型上验证,有效防止后期质量退化
  • 无需修改预训练去噪器,适合通用图像重建任务

插件式(PnP)方法将深度去噪器嵌入到基于模型的图像重建(IR)的近端算法中。与端到端重建不同,PnP 可复用同一预训练去噪器于多种成像任务,无需重新训练。然而,黑盒网络可能导致迭代过程不稳定:视觉质量与 PSNR 通常先提升后下降。现有稳定方法常对去噪器施加严格约束,但标准去噪器仅针对单步去噪训练,未必满足这些条件。本文提出一种简单、数据驱动的稳定机制:将可能不稳定的 PnP 算子与一个收缩的 IR 算子自适应平均,形成类似粘性正则化的策略。收缩成分随迭代逐步抑制更新,有效抑制振荡并防止发散。我们在不同近端算法、去噪架构及成像任务上验证了该机制的有效性。

原文摘要 · Abstract (English)

The plug-and-play (PnP) method uses a deep denoiser within a proximal algorithm for model-based image reconstruction (IR). Unlike end-to-end IR, PnP allows the same pretrained denoiser to be used across different imaging tasks, without the need for retraining. However, black-box networks can make the iterative process in PnP unstable. A common issue observed across architectures like CNNs, diffusion models, and transformers is that the visual quality and PSNR often improve initially but then degrade in later iterations. Previous attempts to ensure stability usually impose restrictive constraints on the denoiser. However, standard denoisers, which are freely trained for single-step noise removal, need not satisfy such constraints. We propose a simple data-driven stabilization mechanism that adaptively averages the potentially unstable PnP operator with a contractive IR operator. This acts as a form of viscosity regularization, where the contractive component progressively dampens updates in later iterations, helping to suppress oscillations and prevent divergence. We validate the effectiveness of our stabilization mechanism across different proximal algorithms, denoising architectures, and imaging tasks.

图像重建稳定性PnP去噪器

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