arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2301.13636cs.LG
收藏

带有支持的传输:数据条件扩散桥

Transport with Support: Data-Conditional Diffusion Bridges

Aalto University(阿尔托大学)

Ella Tamir, Martin Trapp, Arno Solin

AI总结:

本文提出迭代平滑桥(ISB),通过将贝叶斯滤波和最优控制集成到扩散过程学习中,扩展了仅依赖初始和终端约束的Schrödinger桥,以解决具有中间阶段稀疏观测约束的时间序列数据生成问题。

摘要:

动态Schrödinger桥问题为解决作为最优传输问题提出的约束时间序列数据生成任务提供了一个极具吸引力的框架。它包括使用高效的迭代求解器来学习非线性扩散过程。最近的工作展示了最先进的结果(例如,在建模单细胞胚胎RNA序列或从复杂后验中采样时),但仅限于学习仅具有初始和终端约束的桥。我们的工作通过提出迭代平滑桥(ISB)扩展了这一范式。我们将贝叶斯滤波和最优控制集成到扩散过程的学习中,从而能够生成受中间阶段稀疏观测和终端约束控制的约束随机过程。我们在合成和真实世界数据生成任务上评估了我们方法的有效性,并表明ISB能很好地泛化到高维数据,在计算上高效,并且能够准确估计中间和终端时刻的边缘分布。

英文摘要:

The dynamic Schrödinger bridge problem provides an appealing setting for solving constrained time-series data generation tasks posed as optimal transport problems. It consists of learning non-linear diffusion processes using efficient iterative solvers. Recent works have demonstrated state-of-the-art results (eg. in modelling single-cell embryo RNA sequences or sampling from complex posteriors) but are limited to learning bridges with only initial and terminal constraints. Our work extends this paradigm by proposing the Iterative Smoothing Bridge (ISB). We integrate Bayesian filtering and optimal control into learning the diffusion process, enabling the generation of constrained stochastic processes governed by sparse observations at intermediate stages and terminal constraints. We assess the effectiveness of our method on synthetic and real-world data generation tasks and we show that the ISB generalises well to high-dimensional data, is computationally efficient, and provides accurate estimates of the marginals at intermediate and terminal times.

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

↑