提出可解释的扩散模型后验采样方法,提升逆问题重建质量
Analyzing and Guiding Zero-Shot Posterior Sampling in Diffusion Models
- 基于先验高斯假设,推导出后验采样闭式解
- 在频域分析中发现参数需随扩散步长动态调整
- 方法通用且避免人工调参,兼顾图像感知与保真度
从退化观测中恢复信号是科学与工程中的长期挑战。近期基于零样本扩散的方法为这类逆问题提供了基于后验采样的解决方案,利用先验知识进行推理,但通常依赖人工调参和启发式策略。本文对这类近似后验采样器进行严格分析,假设先验服从高斯分布,在此条件下,我们证明理想后验采样器与基于扩散的重建算法均可表达为闭式形式,从而可在频域进行全面分析与比较。基于这些表示,我们提出一个原则性的参数设计框架,替代以往的启发式选择策略。该方法不依赖具体算法,能根据先验特性、退化信号及扩散动力学协同优化参数。实验表明,我们的频域建议在结构上不同于传统启发式方法,且随扩散步长变化,显著提升了感知质量与信号保真度的一致性。
原文摘要 · Abstract (English)
Recovering a signal from its degraded measurements is a long standing challenge in science and engineering. Recently, zero-shot diffusion based methods have been proposed for such inverse problems, offering a posterior sampling based solution that leverages prior knowledge. Such algorithms incorporate the observations through inference, often leaning on manual tuning and heuristics. In this work we propose a rigorous analysis of these approximate posterior samplers, relying on a Gaussianity assumption of the prior. Under this regime, we show that both the ideal posterior sampler and diffusion-based reconstruction algorithms can be expressed in closed-form, enabling their thorough analysis and comparisons in the spectral domain. Building on these representations, we introduce a principled framework for parameter design, replacing heuristic selection strategies used to date. The proposed approach is method-agnostic and yields tailored parameter choices that jointly account for the characteristics of the prior, the degraded signal, and the diffusion dynamics. We show that our spectral recommendations differ structurally from standard heuristics and vary with the diffusion step size, resulting in a consistent balance between perceptual quality and signal fidelity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。