Foundation Inference Models for Ordinary Differential Equations
Johannes Hübers, Maximilian Mauel, David Berghaus, Patrick Seifner, Ramses J Sanchez
Abstract
Ordinary differential equations (ODEs) are central to scientific modelling, but inferring their vector fields from noisy trajectories remains challenging. Current approaches such as symbolic regression, Gaussian process (GP) regression, and Neural ODEs often require complex training pipelines and substantial machine learning expertise, or they depend strongly on system-specific prior knowledge. We propose FIM-ODE, a pretrained Foundation Inference Model that amortises lowdimensional ODE inference by predicting the vector field directly from noisy trajectory data in a single forward pass. We pretrain FIM-ODE on a prior distribution over ODEs with low-degree polynomial vector fields and represent the target field with neural operators. FIM-ODE achieves strong zero-shot performance, matching and often improving upon ODEFormer, a recent pretrained symbolic baseline, across a range of regimes despite using a simpler pretraining prior distribution. Pretraining also provides a strong initialisation for finetuning, enabling fast and stable adaptation that outperforms modern neural and GP baselines without requiring machine learning expertise. Our pretrained model, code repository, and tutorials are available online 1 .
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 abd83969-ba1c-4357-89e3-9db8ea6d172eCited by top-tier papers1
Ask how each one uses itBuilds on21
- Transformers are RNNs: Fast Autoregressive Transformers with Linear AttentionAngelos Katharopoulos, Apoorv Vyas, Nikolaos Pappas, François FleuretICML 2020 · 2,665 citations
- Transformers Can Do Bayesian InferenceSamuel Müller, Noah Hollmann, Sebastian Pineda-Arango, Josif Grabocka et al.ICLR 2022 · 287 citations
- ForecastPFN: Synthetically-Trained Zero-Shot ForecastingSamuel Dooley, Gurnoor Singh Khurana, Chirag Mohapatra, Siddartha V. Naidu et al.NeurIPS 2023 · 142 citations
- Amortized Inference for Causal Structure LearningLars Lorch, Scott Sussex, Jonas Rothfuss, Andreas Krause et al.NeurIPS 2022 · 118 citations
- How to Train Your Neural ODE: the World of Jacobian and Kinetic RegularizationChris Finlay, Jörn-Henrik Jacobsen, Levon Nurbekyan, Adam M. ObermanICML 2020 · 76 citations
Related papers
- 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
- Predicting Ordinary Differential Equations with TransformersSören Becker, Michal Klein, Alexander Neitz, Giambattista Parascandolo et al.ICML 2023 · 28 citations
- ODEFormer: Symbolic Regression of Dynamical Systems with TransformersStéphane d'Ascoli, Sören Becker, Philippe Schwaller, Alexander Mathis et al.ICLR 2024 · 56 citations
- Zero-shot Imputation with Foundation Inference Models for Dynamical SystemsPatrick Seifner, Kostadin Cvejoski, Antonia Körner, Ramsés J. SánchezICLR 2025
- From Fourier to Neural ODEs: Flow Matching for Modeling Complex SystemsXin Li, Jingdong Zhang, Qunxi Zhu, Chengli Zhao et al.ICML 2024 · 8 citations
