Parallel-in-Time Probabilistic Numerical ODE Solvers

Probabilistic numerical solvers for ordinary differential equations (ODEs) treat the numerical simulation of dynamical systems as problems of Bayesian state estimation. Aside from producing posterior distributions over ODE solutions and thereby quantifying the numerical approximation error of the method itself, one less-often noted advantage of this formalism is the algorithmic flexibility gained by formulating numerical simulation in the framework of Bayesian filtering and smoothing. In this paper, we leverage this flexibility and build on the time-parallel formulation of iterated extended Kalman smoothers to formulate a parallel-in-time probabilistic numerical ODE solver. Instead of simulating the dynamical system sequentially in time, as done by current probabilistic solvers, the proposed method processes all time steps in parallel and thereby reduces the computational complexity from linear to logarithmic in the number of time steps. We demonstrate the effectiveness of our approach on a variety of ODEs and compare it to a range of both classic and probabilistic numerical ODE solvers.

Keywords

Bayesian filtering and smoothing, Numerical analysis, Ordinary differential equations, Parallel-in-time methods, Probabilistic numerics

Citation

Bosch, N, Corenflos, A, Yaghoobi, F, Tronarp, F, Hennig, P & Särkkä, S 2024, ' Parallel-in-Time Probabilistic Numerical ODE Solvers ', Journal of Machine Learning Research, vol. 25, 206, pp. 1-27 . < https://www.jmlr.org/papers/v25/23-1261.html >

Permanent link to this item

https://urn.fi/URN:NBN:fi:aalto-202409116309

Collections

[article-cris] Sähkötekniikan korkeakoulu / ELEC

Show all metadata

Parallel-in-Time Probabilistic Numerical ODE Solvers

Access rights

URL

Journal Title

Journal ISSN

Volume Title

Authors

Date

Department

Major/Subject

Mcode

Degree programme

Language

Pages

Series

Abstract

Description

Keywords

Other note

Citation

Permanent link to this item

Collections