A Closed-Form Framework for Schrödinger Bridges Between Arbitrary Densities
By: Hanwen Huang
Potential Business Impact:
Makes computers create realistic images from noise.
Score-based generative models have recently attracted significant attention for their ability to generate high-fidelity data by learning maps from simple Gaussian priors to complex data distributions. A natural generalization of this idea to transformations between arbitrary probability distributions leads to the Schrödinger Bridge (SB) problem. However, SB solutions rarely admit closed-form expressios and are commonly obtained through iterative stochastic simulation procedures, which are computationally intensive and can be unstable. In this work, we introduce a unified closed-form framework for representing the stochastic dynamics of SB systems. Our formulation subsumes previously known analytical solutions including the Schrödinger Föllmer process and the Gaussian SB as specific instances. Notably, the classical Gaussian SB solution, previously derived using substantially more sophisticated tools such as Riemannian geometry and generator theory, follows directly from our formulation as an immediate corollary. Leveraging this framework, we develop a simulation-free algorithm that infers SB dynamics directly from samples of the source and target distributions. We demonstrate the versatility of our approach in two settings: (i) modeling developmental trajectories in single-cell genomics and (ii) solving image restoration tasks such as inpainting and deblurring. This work opens a new direction for efficient and scalable nonlinear diffusion modeling across scientific and machine learning applications.
Similar Papers
A Closed-Form Diffusion Model for Learnring Dynamics from Marginal Observations
Computation
Creates better AI for images and biology.
Robust time series generation via Schrödinger Bridge: a comprehensive evaluation
Machine Learning (CS)
Creates realistic future data from past patterns.
Schrödinger bridge for generative AI: Soft-constrained formulation and convergence analysis
Machine Learning (CS)
Makes AI create better pictures and music.