提出新方法,让扩散模型更准地恢复被噪声破坏的图像。
Coupled Data and Measurement Space Dynamics for Enhanced Diffusion Posterior Sampling
- 在数据空间和测量空间同步构建扩散过程,实现联合建模。
- 无需近似似然或调参,直接推导出可递归采样的后验分布。
- 在医学成像等逆问题上效果优于现有方法,尤其在高噪声下稳定。
逆问题旨在从含噪或不完整观测中恢复未知信号,广泛应用于医学成像、遥感和计算生物学。扩散模型近年作为强大先验被用于解决此类问题。然而,现有方法要么依赖基于投影的技术,通过启发式更新强制测量一致性;要么近似似然 $p(oldsymbol{y} ackslashmid oldsymbol{x})$,在复杂或高噪声条件下易产生伪影且不稳定。为此,本文提出一种新框架——耦合数据与测量空间扩散后验采样(C-DPS),无需约束调参或似然近似。C-DPS在测量空间 $\\{oldsymbol{y}_t\ }$ 引入前向随机过程,与数据空间扩散 $\\_x_t\ }$ 并行演化,从而推导出闭式后验 $p(oldsymbol{x}_{t-1} ackslashmid oldsymbol{x}_t, oldsymbol{y}_{t-1})$。该耦合机制支持基于明确定义后验的精确递归采样。实验表明,C-DPS在多个逆问题基准测试中均持续优于现有基线,定性和定量表现俱佳。
原文摘要 · Abstract (English)
Inverse problems, where the goal is to recover an unknown signal from noisy or incomplete measurements, are central to applications in medical imaging, remote sensing, and computational biology. Diffusion models have recently emerged as powerful priors for solving such problems. However, existing methods either rely on projection-based techniques that enforce measurement consistency through heuristic updates, or they approximate the likelihood $p(\boldsymbol{y} \mid \boldsymbol{x})$, often resulting in artifacts and instability under complex or high-noise conditions. To address these limitations, we propose a novel framework called \emph{coupled data and measurement space diffusion posterior sampling} (C-DPS), which eliminates the need for constraint tuning or likelihood approximation. C-DPS introduces a forward stochastic process in the measurement space $\{\boldsymbol{y}_t\}$, evolving in parallel with the data-space diffusion $\{\boldsymbol{x}_t\}$, which enables the derivation of a closed-form posterior $p(\boldsymbol{x}_{t-1} \mid \boldsymbol{x}_t, \boldsymbol{y}_{t-1})$. This coupling allows for accurate and recursive sampling based on a well-defined posterior distribution. Empirical results demonstrate that C-DPS consistently outperforms existing baselines, both qualitatively and quantitatively, across multiple inverse problem benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。