Multi-marginal temporal Schrödinger Bridge Matching from unpaired data
Thomas Gravier, Thomas Boyer, Auguste Genovesio
Abstract
Many natural dynamic processes -such as in vivo cellular differentiation or disease progression-can only be observed through the lens of static sample snapshots. While challenging, reconstructing their temporal evolution to decipher underlying dynamic properties is of major interest to scientific research. Existing approaches enable data transport along a temporal axis but are poorly scalable in high dimension and require restrictive assumptions to be met. To address these issues, we propose Multi-marginal temporal Schrödinger Bridge Matching (MMtSBM) from unpaired data, extending the theoretical guarantees and empirical efficiency of Diffusion Schrödinger Bridge Matching (Shi et al., 2023) by deriving the Iterative Markovian Fitting algorithm to multiple marginals in a novel factorized fashion. Experiments show that MMtSBM retains theoretical properties on toy examples, achieves state-of-theart performance on real-world datasets such as transcriptomic trajectory inference in 100 dimensions, and for the first time recovers couplings and dynamics in very high dimensional image settings, effectively generating temporally coherent videos from purely unpaired data. Our work establishes multi-marginal Schrödinger Bridges as a practical and theoretically principled approach for recovering hidden dynamics from static data.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on18
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Diffusion Schrödinger Bridge with Applications to Score-Based Generative ModelingValentin De Bortoli, James Thornton, Jeremy Heng, Arnaud DoucetNeurIPS 2021 · 811 citations
- Likelihood Training of Schrödinger Bridge using Forward-Backward SDEs TheoryTianrong Chen, Guan-Horng Liu, Evangelos A. TheodorouICLR 2022 · 249 citations
- Diffusion Schrödinger Bridge MatchingYuyang Shi, Valentin De Bortoli, Andrew Campbell, Arnaud DoucetNeurIPS 2023 · 178 citations
- Deep Generative Learning via Schrödinger BridgeGefei Wang, Yuling Jiao, Qian Xu, Yang Wang et al.ICML 2021 · 136 citations
Related papers
- Momentum Multi-Marginal Schrödinger Bridge MatchingPanagiotis Theodoropoulos, Augustinos D. Saravanos, Evangelos A. Theodorou, Guan-Horng LiuNeurIPS 2025 · 15 citations
- Deep Momentum Multi-Marginal Schrödinger BridgeTianrong Chen, Guan-Horng Liu, Molei Tao, Evangelos A. TheodorouNeurIPS 2023 · 47 citations
- Modeling Complex System Dynamics with Flow Matching Across Time and ConditionsMartin Rohbeck, Edward De Brouwer, Charlotte Bunne, Jan-Christian Huetter et al.ICLR 2025
- Multi-Marginal Stochastic Flow Matching for High-Dimensional Snapshot Data at Irregular Time PointsJustin Lee, Behnaz Moradijamei, Heman ShakeriICML 2025
- Efficient Generative Modeling beyond Memoryless Diffusion via Adjoint Schrödinger Bridge MatchingJeongwoo Shin, Jinhwan Sul, Joonseok Lee, Jaewoong Choi et al.ICML 2026
