通过近似匹配提升图像重建中不匹配去噪器的性能
Domain Adaptation of Mismatched Proximal Denoiser for Plug-and-Play Image Reconstruction

- 用近似匹配代替传统MSE,优化去噪器在域外使用时的更新步骤
- 理论证明收敛速度为1/K,误差项与域不匹配程度成正比
- 在少量样本下显著优于传统方法,适合真实场景中的少样本重建
插件式近端梯度下降(PnP-PGD)通过将去噪器作为隐式先验实现灵活的图像重建。实践中,这些去噪器常部署于其训练域之外。现有分析在去噪器满足结构假设(如为近端映射或压缩映射)下建立收敛性,但未量化域不匹配对PnP-PGD收敛的影响。本文定义该影响为‘近端失配’:部署去噪器$\\-hat{\mathsf D}$与目标域参考映射$\mathsf D_\star=\operatorname{prox}_{R_\star}$之间的差异,后者对应底层正则化项$R_\star$。在此失配下,每次去噪更新成为目标目标函数的近似近端步。我们进一步推导出一个以$\mathcal{O}(1/K)$速率衰减的平稳性界,其附加项与平均平方近端失配成正比。该结果启发我们采用近端匹配而非仅基于MSE的适应策略。我们在两类成熟去噪器族上研究此方法:学习型近端网络与梯度步去噪器。在存在显著域偏移的高斯模糊去模糊和超分辨率任务上,实验表明近端匹配适应显著提升重建质量,尤其在少样本情形下取得最大数值增益。
原文摘要 · Abstract (English)
Plug-and-play proximal gradient descent (PnP-PGD) enables flexible image reconstruction by using denoisers as implicit priors. In practice, these denoisers are often deployed outside their training domains. Existing analyses establish convergence under structural assumptions on the deployed denoiser, such as requiring it to be a proximal map or a contraction. However, they do not measure how domain mismatch affects convergence of PnP-PGD. We define this effect as \emph{proximal mismatch}: the discrepancy between a deployed denoiser $\widehat{\mathsf D}$ and a target-domain reference map $\mathsf D_\star=\operatorname{prox}_{R_\star}$ associated with the underlying regularizer $R_\star$. Under this mismatch, each denoising update becomes an inexact proximal step for the target objective. We further derive a stationarity bound that decays at a rate of $\mathcal{O}(1/K)$, with an additive term proportional to the average squared proximal mismatch. This result motivates adaptation via proximal matching rather than MSE-based adaptation alone. We study this approach with two established denoiser families: learned proximal networks and gradient-step denoisers. Experiments on Gaussian deblurring and super-resolution under substantial domain shift show that proximal matching adaptation improves reconstruction quality significantly over MSE-based adaptation, yielding the largest numerical gains in the few-shot regime.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。