A Generalized Sinkhorn Algorithm for Mean-Field Schrödinger Bridge
- URL: http://arxiv.org/abs/2604.06531v2
- Date: Thu, 09 Apr 2026 16:35:00 GMT
- Title: A Generalized Sinkhorn Algorithm for Mean-Field Schrödinger Bridge
- Abstract summary: The mean-field Schrdinger bridge (MFSB) problem concerns designing a minimum-effort controller that guides a diffusion process with nonlocal interaction to reach a given distribution by a fixed deadline.<n>Unlike Schrdinger bridge, the dynamical constraint for MFSB is a population of interacting agents with repulsive controls.<n>We propose a generalization the Hopf-Cole algorithm for MFSB and, on building it, design a Sinkhorn-type algorithm to solve the associated system integro-PDEs.
- Score: 19.922542189921447
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: The mean-field Schrödinger bridge (MFSB) problem concerns designing a minimum-effort controller that guides a diffusion process with nonlocal interaction to reach a given distribution from another by a fixed deadline. Unlike the standard Schrödinger bridge, the dynamical constraint for MFSB is the mean-field limit of a population of interacting agents with controls. It serves as a natural model for large-scale multi-agent systems. The MFSB is computationally challenging because the nonlocal interaction makes the problem nonconvex. We propose a generalization of the Hopf-Cole transform for MFSB and, building on it, design a Sinkhorn-type recursive algorithm to solve the associated system of integro-PDEs. Under mild assumptions on the interaction potential, we discuss convergence guarantees for the proposed algorithm. We present numerical examples with repulsive and attractive interactions to illustrate the theoretical contributions.
Related papers
- Schrödinger Bridges over Kinetic Swarming Models [16.998652690384596]
We consider finite-horizon minimum-energy collective steering for swarms subject to disturbances.<n>Our objective is to steer the swarm between prescribed endpoint distributions using a state-feedback control.<n>We derive necessary optimality systems with a time-symmetric structure reminiscent of classical Schrdinger bridges.
arXiv Detail & Related papers (2026-08-26T01:34:36Z) - Optimal and Scalable MAPF via Multi-Marginal Optimal Transport and Schrödinger Bridges [1.6042394978941512]
We show that MAPF can be cast as a special class of multi-marginal optimal (MMOT) problems with an underlying Markovian structure.<n>To adapt the approach to large-scale problems, we cast the MAPF-MMOT in a probabilistic framework via Schrdinger bridges.
arXiv Detail & Related papers (2026-05-11T17:52:15Z) - Parallel Complex Diffusion for Scalable Time Series Generation [50.01609741902786]
PaCoDi is a spectral-native architecture that decouples generative modeling in the frequency domain.<n>We show that PaCoDi outperforms existing baselines in both generation quality and inference speed.
arXiv Detail & Related papers (2026-02-10T14:31:53Z) - Schrödinger bridge for generative AI: Soft-constrained formulation and convergence analysis [6.584866740785309]
We study the so-called soft-constrained Schr"odinger bridge problem (SCSBP)<n>We prove that as the penalty grows, both the controls and value functions converge to those of the classical SBP at a linear rate.<n>These results provide the first quantitative convergence guarantees for soft-constrained bridges.
arXiv Detail & Related papers (2025-10-13T18:29:15Z) - Latent Schrodinger Bridge: Prompting Latent Diffusion for Fast Unpaired Image-to-Image Translation [58.19676004192321]
Diffusion models (DMs), which enable both image generation from noise and inversion from data, have inspired powerful unpaired image-to-image (I2I) translation algorithms.
We tackle this problem with Schrodinger Bridges (SBs), which are differential equations (SDEs) between distributions with minimal transport cost.
Inspired by this observation, we propose Latent Schrodinger Bridges (LSBs) that approximate the SB ODE via pre-trained Stable Diffusion.
We demonstrate that our algorithm successfully conduct competitive I2I translation in unsupervised setting with only a fraction of cost required by previous DM-
arXiv Detail & Related papers (2024-11-22T11:24:14Z) - Generalized Schrödinger Bridge Matching [54.171931505066]
Generalized Schr"odinger Bridge (GSB) problem setup is prevalent in many scientific areas both within and without machine learning.
We propose Generalized Schr"odinger Bridge Matching (GSBM), a new matching algorithm inspired by recent advances.
We show that such a generalization can be cast as solving conditional optimal control, for which variational approximations can be used.
arXiv Detail & Related papers (2023-10-03T17:42:11Z) - Light Schrödinger Bridge [72.88707358656869]
We develop a lightweight, simulation-free and theoretically justified Schr"odinger Bridges solver.
Our light solver resembles the Gaussian mixture model which is widely used for density estimation.
Inspired by this similarity, we also prove an important theoretical result showing that our light solver is a universal approximator of SBs.
arXiv Detail & Related papers (2023-10-02T13:06:45Z) - Deep Generalized Schr\"odinger Bridge [26.540105544872958]
Mean-Field Game serves as a crucial mathematical framework in modeling the collective behavior of individual agents.
We show that Schr"odinger Bridge - as an entropy-regularized optimal transport model - can be generalized to accept mean-field structures.
Our method, named Deep Generalized Schr"odinger Bridge (DeepGSB), outperforms prior methods in solving classical population navigation MFGs.
arXiv Detail & Related papers (2022-09-20T17:47:15Z) - The Schr\"odinger Bridge between Gaussian Measures has a Closed Form [101.79851806388699]
We focus on the dynamic formulation of OT, also known as the Schr"odinger bridge (SB) problem.
In this paper, we provide closed-form expressions for SBs between Gaussian measures.
arXiv Detail & Related papers (2022-02-11T15:59:01Z) - Lifting the Convex Conjugate in Lagrangian Relaxations: A Tractable
Approach for Continuous Markov Random Fields [53.31927549039624]
We show that a piecewise discretization preserves better contrast from existing discretization problems.
We apply this theory to the problem of matching two images.
arXiv Detail & Related papers (2021-07-13T12:31:06Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.