Provably Convergent Schr"odinger Bridge with Applications to Probabilistic Time Series Imputation
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
The Schr"odinger bridge problem (SBP) is gaining increasing attention in generative modeling and showing promising potential even in comparison with the score-based generative models (SGMs). SBP can be interpreted as an entropy-regularized optimal transport problem, which conducts projections onto every other marginal alternatingly. However, in practice, only approximated projections are accessible and their convergence is not well understood. To fill this gap, we present a first convergence analysis of the Schr"odinger bridge algorithm based on approximated projections. As for its practical applications, we apply SBP to probabilistic time series imputation by generating missing values conditioned on observed data. We show that optimizing the transport cost improves the performance and the proposed algorithm achieves the state-of-the-art result in healthcare and environmental data while exhibiting the advantage of exploring both temporal and feature patterns in probabilistic time series imputation.
Key facts. Cited by 27 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 42 of its 66 references that are in the map. First posted on arXiv 12 May 2023.
Builds on
42 of 66 references.
Closest ancestors · 8
Same direction, newest first
- 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptions22 Sep
- 2022Convergence for score-based generative modeling with polynomial complexity13 Jun
- 2022Conditional Simulation Using Diffusion Schr"odinger Bridges27 Feb
- 2022Understanding DDPM Latent Codes Through Optimal Transport14 Feb
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modeling1 Jun
- 2021Maximum Likelihood Training of Score-Based Diffusion Models22 Jan
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 34
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2017Modeling Long- and Short-Term Temporal Patterns with Deep Neural Networkscited by 879
- 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 594
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputationcited by 582
- 2016Recurrent Neural Networks for Multivariate Time Series with Missing Valuescited by 554
- 2020Neural Controlled Differential Equations for Irregular Time Seriescited by 506
- 2021Autoregressive Denoising Diffusion Models for Multivariate Probabilistic Time Series Forecastingcited by 337
- 2013A survey of the Schr"odinger problem and some of its connections with optimal transportcited by 292
- 2018BRITS: Bidirectional Recurrent Imputation for Time Seriescited by 290
- 2020Scalable Gradients for Stochastic Differential Equationscited by 271
22 more
- 2017Non-convex learning via Stochastic Gradient Langevin Dynamics: a nonasymptotic analysiscited by 243
- 2021Multi-Time Attention Networks for Irregularly Sampled Time Seriescited by 238
- 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 219
- 2019Latent ODEs for Irregularly-Sampled Time Seriescited by 183
- 2019GP-VAE: Deep Probabilistic Time Series Imputationcited by 172
- 2019High-Dimensional Multivariate Forecasting with Low-Rank Gaussian Copula Processescited by 148
- 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 141
- 2021Deep Generative Learning via Schr"odinger Bridgecited by 104
- 2022Neural Conservation Laws: A Divergence-Free Perspectivecited by 70
- 2022Statistical Efficiency of Score Matching: The View from Isoperimetrycited by 64
- 2022The Schr"odinger Bridge between Gaussian Measures has a Closed Formcited by 63
- 2019Wasserstein Proximal Algorithms for the Schr"odinger Bridge Problem: Density Control with Nonlinear Driftcited by 50
- 2020Uncertainty-Aware Variational-Recurrent Imputation Network for Clinical Time Seriescited by 33
- 2018The data-driven Schroedinger bridgecited by 26
- 2020A Contour Stochastic Gradient Langevin Dynamics Algorithm for Simulations of Multi-modal Distributionscited by 26
- 2021Stability of Entropic Optimal Transport and Schr"odinger Bridgescited by 26
- 2022On the Convergence Rate of Sinkhorn's Algorithmcited by 16
- 2014Stochastic bridges of linear systemscited by 14
- 2022Stability of Schr"odinger Potentials and Convergence of Sinkhorn's Algorithmcited by 13
- 2023Quantitative contraction rates for Sinkhorn's algorithm: beyond bounded costs and compact marginalscited by 13
- 2021Continuous Latent Process Flowscited by 10
- 2023Non-asymptotic convergence bounds for Sinkhorn iterates and their gradients: a coupling approachcited by 7
Built on by
Citing papers per year: 2023: 4; 2024: 13; 2025: 3; 2026: 7.
27 papers.
Sort:
By area:
- 2023The probability flow ODE is provably fastcited by 146
- 2024UniTS: A Unified Multi-Task Time Series Modelcited by 94
- 2024Deep Learning for Multivariate Time Series Imputation: A Surveycited by 57
- 2024TSI-Bench: Benchmarking Time Series Imputationcited by 33
- 2025Non-stationary Diffusion For Probabilistic Time Series Forecastingcited by 27
- 2024Sports-Traj: A Unified Trajectory Generation Model for Multi-Agent Movement in Sportscited by 19
- 2023Light Schr"odinger Bridgecited by 18
- 2024SSD-TS: Exploring the Potential of Linear State Space Models for Diffusion Models in Time Series Imputationcited by 18
- 2024MTSCI: A Conditional Diffusion Model for Multivariate Time Series Consistent Imputationcited by 13
- 2024Reflected Schr"odinger Bridge for Constrained Generative Modelingcited by 12
- 2023BayOTIDE: Bayesian Online Multivariate Time series Imputation with functional decompositioncited by 10
- 2024Variational Schr"odinger Diffusion Modelscited by 9
15 more
- 2024How Deep is your Guess? A Fresh Perspective on Deep Learning for Medical Time-Series Imputationcited by 5
- 2024Conditional Lagrangian Wasserstein Flow for Time Series Imputationcited by 4
- 2024Rethinking the Diffusion Models for Numerical Tabular Data Imputation from the Perspective of Wasserstein Gradient Flowcited by 3
- 2024Recurrent Interpolants for Probabilistic Time Series Predictioncited by 2
- 2025Optimal Stochastic Trace Estimation in Generative Modelingcited by 2
- 2023Deep sub-ensembles meets quantile regression: uncertainty-aware imputation for time seriescited by 1
- 2025Causal View of Time Series Imputation: Some Identification Results on Missing Mechanismcited by 1
- 2026Spatial-Temporal Feedback Diffusion Guidance for Controlled Traffic Imputationcited by 1
- 2026Pathwise Learning of Stochastic Dynamical Systems with Partial Observationscited by 1
- 2026ABC: Any-Subset Autoregression via Non-Markovian Diffusion Bridges in Continuous Time and Spacecited by 1
- 2024Modular Deep Learning for Multivariate Time-Series: Decoupling Imputation and Downstream Taskscited by 0
- 2026Analyzing and Improving Diffusion Models for Time-Series Data Imputation: A Proximal Recursion Perspectivecited by 0
- 2026PAMF: Prior-Aware Multimodal Fusion for Incomplete Time Series Datacited by 0
- 2026TimeES: Probabilistic and Deterministic Time Series Forecasting via Evolutionary Spectracited by 0
- 2026MissPath-FM: Flow Matching with Structured Priors for Partially Observed Time Seriescited by 0
- 2026MissPath-FM: Flow Matching with Structured Priors for Partially Observed Time Series13 Oct 2026
- 2026TimeES: Probabilistic and Deterministic Time Series Forecasting via Evolutionary Spectra26 Sep 2026
- 2026PAMF: Prior-Aware Multimodal Fusion for Incomplete Time Series Data4 Jun 2026
- 2026ABC: Any-Subset Autoregression via Non-Markovian Diffusion Bridges in Continuous Time and Space30 Apr 2026
- 2026Analyzing and Improving Diffusion Models for Time-Series Data Imputation: A Proximal Recursion Perspective1 Feb 2026
- 2026Pathwise Learning of Stochastic Dynamical Systems with Partial Observations29 Jan 2026
- 2026Spatial-Temporal Feedback Diffusion Guidance for Controlled Traffic Imputation8 Jan 2026
- 2025Causal View of Time Series Imputation: Some Identification Results on Missing Mechanism12 May 2025
- 2025Non-stationary Diffusion For Probabilistic Time Series Forecasting7 May 2025
- 2025Optimal Stochastic Trace Estimation in Generative Modeling26 Feb 2025
- 2024Modular Deep Learning for Multivariate Time-Series: Decoupling Imputation and Downstream Tasks6 Nov 2024
- 2024SSD-TS: Exploring the Potential of Linear State Space Models for Diffusion Models in Time Series Imputation17 Oct 2024
15 more
- 2024Conditional Lagrangian Wasserstein Flow for Time Series Imputation10 Oct 2024
- 2024Recurrent Interpolants for Probabilistic Time Series Prediction18 Sep 2024
- 2024MTSCI: A Conditional Diffusion Model for Multivariate Time Series Consistent Imputation11 Aug 2024
- 2024How Deep is your Guess? A Fresh Perspective on Deep Learning for Medical Time-Series Imputation11 Jul 2024
- 2024Rethinking the Diffusion Models for Numerical Tabular Data Imputation from the Perspective of Wasserstein Gradient Flow22 Jun 2024
- 2024TSI-Bench: Benchmarking Time Series Imputation18 Jun 2024
- 2024Sports-Traj: A Unified Trajectory Generation Model for Multi-Agent Movement in Sports27 May 2024
- 2024Variational Schr"odinger Diffusion Models8 May 2024
- 2024UniTS: A Unified Multi-Task Time Series Model29 Feb 2024
- 2024Deep Learning for Multivariate Time Series Imputation: A Survey6 Feb 2024
- 2024Reflected Schr"odinger Bridge for Constrained Generative Modeling6 Jan 2024
- 2023Deep sub-ensembles meets quantile regression: uncertainty-aware imputation for time series3 Dec 2023
- 2023Light Schr"odinger Bridge2 Oct 2023
- 2023BayOTIDE: Bayesian Online Multivariate Time series Imputation with functional decomposition28 Aug 2023
- 2023The probability flow ODE is provably fast19 May 2023
- 2023The probability flow ODE is provably fastcited by 146
- 2025Non-stationary Diffusion For Probabilistic Time Series Forecastingcited by 27
- 2024SSD-TS: Exploring the Potential of Linear State Space Models for Diffusion Models in Time Series Imputationcited by 18
- 2024MTSCI: A Conditional Diffusion Model for Multivariate Time Series Consistent Imputationcited by 13
- 2024Reflected Schr"odinger Bridge for Constrained Generative Modelingcited by 12
- 2024Variational Schr"odinger Diffusion Modelscited by 9
- 2024Rethinking the Diffusion Models for Numerical Tabular Data Imputation from the Perspective of Wasserstein Gradient Flowcited by 3
- 2025Optimal Stochastic Trace Estimation in Generative Modelingcited by 2
- 2026Spatial-Temporal Feedback Diffusion Guidance for Controlled Traffic Imputationcited by 1
- 2026Analyzing and Improving Diffusion Models for Time-Series Data Imputation: A Proximal Recursion Perspectivecited by 0
- 2026PAMF: Prior-Aware Multimodal Fusion for Incomplete Time Series Datacited by 0
- 2026MissPath-FM: Flow Matching with Structured Priors for Partially Observed Time Seriescited by 0
- 2026MissPath-FM: Flow Matching with Structured Priors for Partially Observed Time Series13 Oct 2026
- 2026PAMF: Prior-Aware Multimodal Fusion for Incomplete Time Series Data4 Jun 2026
- 2026Analyzing and Improving Diffusion Models for Time-Series Data Imputation: A Proximal Recursion Perspective1 Feb 2026
- 2026Spatial-Temporal Feedback Diffusion Guidance for Controlled Traffic Imputation8 Jan 2026
- 2025Non-stationary Diffusion For Probabilistic Time Series Forecasting7 May 2025
- 2025Optimal Stochastic Trace Estimation in Generative Modeling26 Feb 2025
- 2024SSD-TS: Exploring the Potential of Linear State Space Models for Diffusion Models in Time Series Imputation17 Oct 2024
- 2024MTSCI: A Conditional Diffusion Model for Multivariate Time Series Consistent Imputation11 Aug 2024
- 2024Rethinking the Diffusion Models for Numerical Tabular Data Imputation from the Perspective of Wasserstein Gradient Flow22 Jun 2024
- 2024Variational Schr"odinger Diffusion Models8 May 2024
- 2024Reflected Schr"odinger Bridge for Constrained Generative Modeling6 Jan 2024
- 2023The probability flow ODE is provably fast19 May 2023
- 2024UniTS: A Unified Multi-Task Time Series Modelcited by 94
- 2024Deep Learning for Multivariate Time Series Imputation: A Surveycited by 57
- 2024TSI-Bench: Benchmarking Time Series Imputationcited by 33
- 2023Light Schr"odinger Bridgecited by 18
- 2023BayOTIDE: Bayesian Online Multivariate Time series Imputation with functional decompositioncited by 10
- 2024How Deep is your Guess? A Fresh Perspective on Deep Learning for Medical Time-Series Imputationcited by 5
- 2024Conditional Lagrangian Wasserstein Flow for Time Series Imputationcited by 4
- 2024Recurrent Interpolants for Probabilistic Time Series Predictioncited by 2
- 2023Deep sub-ensembles meets quantile regression: uncertainty-aware imputation for time seriescited by 1
- 2025Causal View of Time Series Imputation: Some Identification Results on Missing Mechanismcited by 1
- 2026Pathwise Learning of Stochastic Dynamical Systems with Partial Observationscited by 1
- 2026ABC: Any-Subset Autoregression via Non-Markovian Diffusion Bridges in Continuous Time and Spacecited by 1
12 most cited of 14 All 14 (JSON)
- 2026TimeES: Probabilistic and Deterministic Time Series Forecasting via Evolutionary Spectra26 Sep 2026
- 2026ABC: Any-Subset Autoregression via Non-Markovian Diffusion Bridges in Continuous Time and Space30 Apr 2026
- 2026Pathwise Learning of Stochastic Dynamical Systems with Partial Observations29 Jan 2026
- 2025Causal View of Time Series Imputation: Some Identification Results on Missing Mechanism12 May 2025
- 2024Modular Deep Learning for Multivariate Time-Series: Decoupling Imputation and Downstream Tasks6 Nov 2024
- 2024Conditional Lagrangian Wasserstein Flow for Time Series Imputation10 Oct 2024
- 2024Recurrent Interpolants for Probabilistic Time Series Prediction18 Sep 2024
- 2024How Deep is your Guess? A Fresh Perspective on Deep Learning for Medical Time-Series Imputation11 Jul 2024
- 2024TSI-Bench: Benchmarking Time Series Imputation18 Jun 2024
- 2024UniTS: A Unified Multi-Task Time Series Model29 Feb 2024
- 2024Deep Learning for Multivariate Time Series Imputation: A Survey6 Feb 2024
- 2023Deep sub-ensembles meets quantile regression: uncertainty-aware imputation for time series3 Dec 2023
12 newest of 14 All 14 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026Foundations of Schr"odinger Bridges for Generative Modeling
- 2025A Closed-Form Framework for Schr"odinger Bridges Between Arbitrary Densities
- 2024Soft-constrained Schrodinger Bridge: a Stochastic Control Approach
- 2023Simulation-free Schr"odinger bridges via score and flow matching
- 2024Schr"odinger Bridge Flow for Unpaired Data Translation
- 2025Data-to-Energy Stochastic Dynamics
- 2023Generalized Schr"odinger Bridge Matching
- 2025Incorporating Pre-trained Diffusion Models in Solving the Schr"odinger Bridge Problem