🤖 AI Summary
This work addresses the problem of efficiently and accurately bridging arbitrary probability density functions within a bounded time horizon. Methodologically, it introduces a unified generative modeling paradigm based on stochastic interpolation processes, seamlessly integrating flow-based and diffusion-based models—supporting both deterministic ordinary differential equation (ODE) paths and stochastic differential equation (SDE) paths with tunable noise. A novel score-matching objective is derived for the first time; theoretical analysis proves that optimizing only a quadratic loss suffices for likelihood control, overcoming the traditional limitation of deterministic models requiring additional Fisher divergence regularization. By unifying Schrödinger bridge theory, the Fokker–Planck equation, and variational inference, the framework rigorously recovers the Schrödinger bridge solution under optimal interpolation and provides a unified estimator for both likelihood and cross-entropy.
📝 Abstract
A class of generative models that unifies flow-based and diffusion-based methods is introduced. These models extend the framework proposed in Albergo&Vanden-Eijnden (2023), enabling the use of a broad class of continuous-time stochastic processes called `stochastic interpolants' to bridge any two arbitrary probability density functions exactly in finite time. These interpolants are built by combining data from the two prescribed densities with an additional latent variable that shapes the bridge in a flexible way. The time-dependent probability density function of the stochastic interpolant is shown to satisfy a first-order transport equation as well as a family of forward and backward Fokker-Planck equations with tunable diffusion. Upon consideration of the time evolution of an individual sample, this viewpoint immediately leads to both deterministic and stochastic generative models based on probability flow equations or stochastic differential equations with an adjustable level of noise. The drift coefficients entering these models are time-dependent velocity fields characterized as the unique minimizers of simple quadratic objective functions, one of which is a new objective for the score of the interpolant density. Remarkably, we show that minimization of these quadratic objectives leads to control of the likelihood for any of our generative models built upon stochastic dynamics. By contrast, we establish that generative models based upon a deterministic dynamics must, in addition, control the Fisher divergence between the target and the model. We also construct estimators for the likelihood and the cross-entropy of interpolant-based generative models, discuss connections with other stochastic bridges, and demonstrate that such models recover the Schr""odinger bridge between the two target densities when explicitly optimizing over the interpolant.
