论文标题

高斯措施之间的Schrödinger桥有一个封闭形式

The Schrödinger Bridge between Gaussian Measures has a Closed Form

论文作者

Bunne, Charlotte, Hsieh, Ya-Ping, Cuturi, Marco, Krause, Andreas

论文摘要

高斯人之间的静态最佳传输$(\ mathrm {ot})$问题试图恢复最佳地图,或更一般地是耦合的,以将高斯人变形为另一个。它已经进行了充分的研究,并应用于各种任务。在这里,我们着重于OT的动态表述,也称为Schrödinger桥(SB)问题,由于它与基于扩散的生成模型的联系,最近看到了对机器学习的兴趣激增。与静态设置相反,即使对于高斯分布,对动态设置的了解却少得多。在本文中,我们为高斯措施之间的SBS提供了封闭式表达式。与静态高斯ot问题相反,可以简单地简单地将其简单地研究为研究凸程序,我们解决SBS的框架需要更大的涉及工具,例如Riemannian几何形状和发电机理论。值得注意的是,我们确定高斯措施之间的SBS解决方案本身是具有明确的平均值和协方差内核的高斯过程,因此对于许多下游应用(例如生成建模或插值)而言,很容易适应。为了证明实用性,我们设计了一种建模单细胞基因组数据演变的新方法,并报告与现有基于SB的方法相比,数值稳定性显着提高。

The static optimal transport $(\mathrm{OT})$ problem between Gaussians seeks to recover an optimal map, or more generally a coupling, to morph a Gaussian into another. It has been well studied and applied to a wide variety of tasks. Here we focus on the dynamic formulation of OT, also known as the Schrödinger bridge (SB) problem, which has recently seen a surge of interest in machine learning due to its connections with diffusion-based generative models. In contrast to the static setting, much less is known about the dynamic setting, even for Gaussian distributions. In this paper, we provide closed-form expressions for SBs between Gaussian measures. In contrast to the static Gaussian OT problem, which can be simply reduced to studying convex programs, our framework for solving SBs requires significantly more involved tools such as Riemannian geometry and generator theory. Notably, we establish that the solutions of SBs between Gaussian measures are themselves Gaussian processes with explicit mean and covariance kernels, and thus are readily amenable for many downstream applications such as generative modeling or interpolation. To demonstrate the utility, we devise a new method for modeling the evolution of single-cell genomics data and report significantly improved numerical stability compared to existing SB-based approaches.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源