arXiv:2512.10524cs.LGcs.CV2025-12被引 1

用新方法让扩散模型直接解逆问题,更快更准。

Inverse problems with diffusion models: MAP estimation via mode-seeking loss

  • 提出变分模式搜索损失,引导生成样本逼近最优解
  • 线性逆问题下可解析推导,无需近似假设
  • 实测在图像修复任务中速度快、效果好,适合工程应用

预训练的无条件扩散模型结合后验采样或最大后验(MAP)估计,可在不进行特定任务训练或微调的情况下解决任意逆问题。然而,现有后验采样和MAP估计方法常依赖建模近似,且计算成本较高。本文提出一种新的MAP估计策略,引入变分模式搜索损失(VML),证明其在每个反向扩散步骤中最小化能引导生成样本趋向于MAP估计(实际中为模式)。VML源于最小化扩散后验 $p(\oldsymbol{x}_0|\oldsymbol{x}_t)$ 与观测后验 $p(\oldsymbol{x}_0|\oldsymbol{y})$ 之间的KL散度的新视角,其中 $\oldsymbol{y}$ 表示观测值。重要的是,对于线性逆问题,VML可无近似地解析推导。基于进一步的理论洞察,提出VML-MAP算法,通过大量实验验证其在多种图像恢复任务和数据集上兼具性能与效率优势。

原文摘要 · Abstract (English)

A pre-trained unconditional diffusion model, combined with posterior sampling or maximum a posteriori (MAP) estimation techniques, can solve arbitrary inverse problems without task-specific training or fine-tuning. However, existing posterior sampling and MAP estimation methods often rely on modeling approximations and can also be computationally demanding. In this work, we propose a new MAP estimation strategy for solving inverse problems with a pre-trained unconditional diffusion model. Specifically, we introduce the variational mode-seeking loss (VML) and show that its minimization at each reverse diffusion step guides the generated sample towards the MAP estimate (modes in practice). VML arises from a novel perspective of minimizing the Kullback-Leibler (KL) divergence between the diffusion posterior $p(\mathbf{x}_0|\mathbf{x}_t)$ and the measurement posterior $p(\mathbf{x}_0|\mathbf{y})$, where $\mathbf{y}$ denotes the measurement. Importantly, for linear inverse problems, VML can be analytically derived without any modeling approximations. Based on further theoretical insights, we propose VML-MAP, an empirically effective algorithm for solving inverse problems via VML minimization, and validate its efficacy in both performance and computational time through extensive experiments on diverse image-restoration tasks across multiple datasets.

扩散模型逆问题图像修复优化

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