arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2503.20573math.PRmath.STstat.TH
收藏

扩散模型与采样中的随机传输映射

Stochastic Transport Maps in Diffusion Models and Sampling

Xicheng Zhang

AI总结:

本研究提出基于扩散过程的随机传输映射理论框架,证明相关SDE解的存在唯一性,并构建统一方法以推广扩散生成模型,分析粒子近似收敛性,为高维生成与采样提供新工具。

摘要:

在本工作中,我们提出了一个利用扩散过程在概率分布之间构建随机传输映射的理论与计算框架。首先,我们证明了两个独立扩散过程之和的时间边际分布满足福克-普朗克方程。基于此结果并应用Ambrosio-Figalli-Trevisan的叠加原理,我们建立了相关随机微分方程(SDE)解的存在性与唯一性。利用这些理论基础,我们开发了一种通过动态常微分方程(ODE)和SDE在任意概率分布之间构建(随机)传输映射的方法。此外,我们引入了一个统一框架,该框架推广并扩展了广泛的基于扩散的生成模型和采样技术。最后,我们分析了支撑我们框架的SDE的粒子近似的收敛性质,为其实际实现提供了理论保证。这项工作弥合了理论见解与实际应用之间的鸿沟,为高维空间中的生成建模和采样提供了新工具。

英文摘要:

In this work, we present a theoretical and computational framework for constructing stochastic transport maps between probability distributions using diffusion processes. We begin by proving that the time-marginal distribution of the sum of two independent diffusion processes satisfies a Fokker-Planck equation. Building on this result and applying Ambrosio-Figalli-Trevisan's superposition principle, we establish the existence and uniqueness of solutions to the associated stochastic differential equation (SDE). Leveraging these theoretical foundations, we develop a method to construct (stochastic) transport maps between arbitrary probability distributions using dynamical ordinary differential equations (ODEs) and SDEs. Furthermore, we introduce a unified framework that generalizes and extends a broad class of diffusion-based generative models and sampling techniques. Finally, we analyze the convergence properties of particle approximations for the SDEs underlying our framework, providing theoretical guarantees for their practical implementation. This work bridges theoretical insights with practical applications, offering new tools for generative modeling and sampling in high-dimensional spaces.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑