20
23

Schrödinger Bridge Samplers

Abstract

Consider a reference Markov process with initial distribution π0\pi_{0} and transition kernels {Mt}t[1:T]\{M_{t}\}_{t\in[1:T]}, for some TNT\in\mathbb{N}. Assume that you are given distribution πT\pi_{T}, which is not equal to the marginal distribution of the reference process at time TT. In this scenario, Schr\"odinger addressed the problem of identifying the Markov process with initial distribution π0\pi_{0} and terminal distribution equal to πT\pi_{T} which is the closest to the reference process in terms of Kullback--Leibler divergence. This special case of the so-called Schr\"odinger bridge problem can be solved using iterative proportional fitting, also known as the Sinkhorn algorithm. We leverage these ideas to develop novel Monte Carlo schemes, termed Schr\"odinger bridge samplers, to approximate a target distribution π\pi on Rd\mathbb{R}^{d} and to estimate its normalizing constant. This is achieved by iteratively modifying the transition kernels of the reference Markov chain to obtain a process whose marginal distribution at time TT becomes closer to πT=π\pi_T = \pi, via regression-based approximations of the corresponding iterative proportional fitting recursion. We report preliminary experiments and make connections with other problems arising in the optimal transport, optimal control and physics literatures.

View on arXiv
Comments on this paper