用滞后时间修正提升图像修复的扩散后验采样效果
Improving Diffusion Posterior Samplers with Lagged Temporal Corrections for Image Restoration
- 提出二阶离散化,引入前后估计差值作为滞后校正
- 在不增加采样次数下,显著优于DiffPIR、DDRM等基线
- 可作为模块化插件适配多种后验采样框架
基于扩散的后验采样(PS)是成像逆问题的主流框架,结合了学习到的先验与测量约束。然而其标准形式依赖瞬时数据一致估计,导致反向动态存在时间波动。本文从动力学视角重新审视PS,发现标准更新相当于一阶离散化加残差校正,用于弥补去噪预测与数据一致估计之间的偏差。而二阶离散化天然引入基于连续估计变化的滞后校正。据此,我们提出LAMP方法,结合二阶更新与残差校正,实现带有滞后时间修正的后验采样。LAMP保持后验采样结构,并通过单步风险分析揭示其在偏差-方差权衡下的改进机制。在多个成像任务中,实验表明其性能持续优于DiffPIR、DDRM等强基线,且无需增加去噪评估次数。
原文摘要 · Abstract (English)
Diffusion-based posterior sampling (PS) is a leading framework for imaging inverse problems, combining learned priors with measurement constraints. Yet, its standard formulations rely on instantaneous data-consistent estimates, which induce temporal variability in the reverse dynamics. We reinterpret PS from a dynamical perspective, showing that the standard PS update corresponds to a first-order discretization of the diffusion dynamics plus a residual correction capturing the mismatch between the denoised prediction and the data-consistent estimate. A second-order discretization, however, naturally introduces a temporal correction based on the variation of consecutive estimates. Building on this, we propose LAMP, combining the second-order update with the residual correction characterizing a PS technique. LAMP thus inherits a lagged temporal correction, and it can be implemented as a modular plug-in over the PS backbone. We show that LAMP preserves the structure of a posterior sampler, and we perform a one-step risk analysis to characterize when LAMP improves the reverse transition via a bias-variance trade-off. Experiments across multiple imaging tasks demonstrate consistent improvements over strong baselines such as DiffPIR and DDRM, without increasing the number of denoising evaluations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。