Soft-constrained Schrodinger Bridge: a Stochastic Control Approach
CoRR(2024)
摘要
Schrödinger bridge can be viewed as a continuous-time stochastic control
problem where the goal is to find an optimally controlled diffusion process
whose terminal distribution coincides with a pre-specified target distribution.
We propose to generalize this problem by allowing the terminal distribution to
differ from the target but penalizing the Kullback-Leibler divergence between
the two distributions. We call this new control problem soft-constrained
Schrödinger bridge (SSB). The main contribution of this work is a
theoretical derivation of the solution to SSB, which shows that the terminal
distribution of the optimally controlled process is a geometric mixture of the
target and some other distribution. This result is further extended to a time
series setting. One application is the development of robust generative
diffusion models. We propose a score matching-based algorithm for sampling from
geometric mixtures and showcase its use via a numerical example for the MNIST
data set.
更多查看译文
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要