SDE Matching: Scalable and Simulation-Free Training of Latent Stochastic Differential Equations
Grigory Bartosh, Dmitry P. Vetrov, Christian A. Naesseth
Abstract
The Latent Stochastic Differential Equation (SDE) is a powerful tool for time series and sequence modeling. However, training Latent SDEs typically relies on adjoint sensitivity methods, which depend on simulation and backpropagation through approximate SDE solutions, which limit scalability. In this work, we propose SDE Matching, a new simulation-free method for training Latent SDEs. Inspired by modern Score-and Flow Matching algorithms for learning generative dynamics, we extend these ideas to the domain of stochastic dynamics for time series and sequence modeling, eliminating the need for costly numerical simulations. Our results demonstrate that SDE Matching achieves performance comparable to adjoint sensitivity methods while drastically reducing computational complexity.
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.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 64af12d5-ca09-442c-b21d-5f7071dbb20dCited by top-tier papers7
- Curly Flow Matching for Learning Non-gradient Field DynamicsKatarina Petrovic, Lazar Atanackovic, Viggo Moro, Kacper Kapusniak et al.NeurIPS 2025 · 23 citations
- In-Context Learning of Stochastic Differential Equations with Foundation Inference ModelsPatrick Seifner, Kostadin Cvejoski, David Berghaus, César Ali Ojeda Marin et al.NeurIPS 2025 · 8 citations
- SING: SDE Inference via Natural GradientsAmber Hu, Henry Smith, Scott W. LindermanNeurIPS 2025 · 6 citations
- Neural Stochastic Flows: Solver-Free Modelling and Inference for SDE SolutionsNaoki Kiyohara, Edward Johns, Yingzhen LiNeurIPS 2025 · 5 citations
- Diffusion Differentiable ResamplingJennifer R. Andersson, Zheng ZhaoICML 2026 · 2 citations
Builds on21
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Maximum Likelihood Training of Score-Based Diffusion ModelsYang Song, Conor Durkan, Iain Murray, Stefano ErmonNeurIPS 2021 · 958 citations
- Practical and Asymptotically Exact Conditional Sampling in Diffusion ModelsLuhuan Wu, Brian L. Trippe, Christian A. Naesseth, David M. Blei et al.NeurIPS 2023 · 276 citations
- Stochastic Normalizing FlowsHao Wu, Jonas Köhler, Frank NoéNeurIPS 2020 · 230 citations
Related papers
- Trajectory Flow Matching with Applications to Clinical Time Series ModellingXi Zhang, Yuan Pu, Yuki Kawamura, Andrew Loza et al.NeurIPS 2024 · 44 citations
- Amortized Reparametrization: Efficient and Scalable Variational Inference for Latent SDEsKevin Course, Prasanth B. NairNeurIPS 2023 · 17 citations
- Simulation-Free Training of Neural ODEs on Paired DataSemin Kim, Jaehoon Yoo, Jinwoo Kim, Yeonwoo Cha et al.NeurIPS 2024 · 10 citations
- Efficient Training of Neural Stochastic Differential Equations by Matching Finite Dimensional DistributionsJianxin Zhang, Josh Viktorov, Doosan Jung, Emily PitlerICLR 2025
- AdjointDEIS: Efficient Gradients for Diffusion ModelsZander W. Blasingame, Chen LiuNeurIPS 2024 · 8 citations
