提出噪声自适应扩散采样法,无需调参即可高效解决各类逆问题。
Noise-Adaptive Diffusion Sampling for Inverse Problems Without Task-Specific Tuning

- 在初始噪声空间进行哈密顿采样,避免局部最优
- 在4个线性、3个非线性任务中均超越现有方法
- 适合无噪声先验、需鲁棒重建的逆问题场景
扩散模型(DMs)在逆问题(IPs)上表现优异。基于优化的方法虽可利用DM作为强先验快速求解,但易陷入局部极小值并过拟合噪声。尽管DM能为贝叶斯方法提供强大先验,但在去噪过程中强制满足测量一致性会导致流形不可行问题。本文提出噪声空间哈密顿蒙特卡洛(N-HMC),将逆扩散视为从初始噪声到干净图像的确定性映射,实现解空间的充分探索,避免局部最优。通过将推理完全置于初始噪声空间,确保提案位于学习到的数据流形上。我们提供了全面的理论分析,并扩展出噪声自适应变体(NA-NHMC),有效处理未知噪声类型与水平的逆问题。在4个线性与3个非线性逆问题上的大量实验表明,NA-NHMC在不同超参数和初始化下均表现稳健,重建质量显著优于最新方法。
原文摘要 · Abstract (English)
Diffusion models (DMs) have recently shown remarkable performance on inverse problems (IPs). Optimization-based methods can fast solve IPs using DMs as powerful regularizers, but they are susceptible to local minima and noise overfitting. Although DMs can provide strong priors for Bayesian approaches, enforcing measurement consistency during the denoising process leads to manifold infeasibility issues. We propose Noise-space Hamiltonian Monte Carlo (N-HMC), a posterior sampling method that treats reverse diffusion as a deterministic mapping from initial noise to clean images. N-HMC enables comprehensive exploration of the solution space, avoiding local optima. By moving inference entirely into the initial-noise space, N-HMC keeps proposals on the learned data manifold. We provide a comprehensive theoretical analysis of our approach and extend the framework to a noise-adaptive variant (NA-NHMC) that effectively handles IPs with unknown noise type and level. Extensive experiments across four linear and three nonlinear inverse problems demonstrate that NA-NHMC achieves superior reconstruction quality with robust performance across different hyperparameters and initializations, significantly outperforming recent state-of-the-art methods. The code is available at https://github.com/NA-HMC/NA-HMC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。