arXiv:2504.07437cs.LGstat.CO2025-04被引 12

用偏微分方程统一推导扩散模型,解决物理逆问题。

Unifying and extending Diffusion Models through PDEs for solving Inverse Problems

  • 基于线性PDE思想构造正向与反向过程,实现可构造性推导。
  • 统一多种模型形式与采样策略,发现一类新的方差保持模型。
  • 单模型适配多测量算子,适用于复杂逆问题求解。

扩散模型已成为计算机视觉与科学机器学习中强大的生成工具,用于求解大规模概率逆问题。传统方法基于变分推断、去噪、统计信号处理和随机微分方程,而本文改用线性偏微分方程思想推导扩散模型,带来多项优势:可构造性地建立正向与反向过程;统一多种模型形式与采样策略;发现一类新的方差保持模型。进一步将条件版本应用于经典条件密度估计与挑战性逆问题,建立系统性评估基准以量化不同方法的性能。最后,提出并实现单一模型适配多个测量算子的机制。本研究为扩散模型在物理基逆问题中的应用提供了新视角与新方向。

原文摘要 · Abstract (English)

Diffusion models have emerged as powerful generative tools with applications in computer vision and scientific machine learning (SciML), where they have been used to solve large-scale probabilistic inverse problems. Traditionally, these models have been derived using principles of variational inference, denoising, statistical signal processing, and stochastic differential equations. In contrast to the conventional presentation, in this study we derive diffusion models using ideas from linear partial differential equations and demonstrate that this approach has several benefits that include a constructive derivation of the forward and reverse processes, a unified derivation of multiple formulations and sampling strategies, and the discovery of a new class of variance preserving models. We also apply the conditional version of these models to solve canonical conditional density estimation problems and challenging inverse problems. These problems help establish benchmarks for systematically quantifying the performance of different formulations and sampling strategies in this study and for future studies. Finally, we identify and implement a mechanism through which a single diffusion model can be applied to measurements obtained from multiple measurement operators. Taken together, the contents of this manuscript provide a new understanding of and several new directions in the application of diffusion models to solving physics-based inverse problems.

扩散模型逆问题PDE生成建模

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