Diffusion Tempering Improves Parameter Estimation with Probabilistic Integrators for Ordinary Differential Equations

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Beck, Jonas, Bosch, Nathanael, Deistler, Michael, Kadhim, Kyra L., Macke, Jakob H., Hennig, Philipp, Berens, Philipp
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916329888415744
author Beck, Jonas
Bosch, Nathanael
Deistler, Michael
Kadhim, Kyra L.
Macke, Jakob H.
Hennig, Philipp
Berens, Philipp
author_facet Beck, Jonas
Bosch, Nathanael
Deistler, Michael
Kadhim, Kyra L.
Macke, Jakob H.
Hennig, Philipp
Berens, Philipp
contents Ordinary differential equations (ODEs) are widely used to describe dynamical systems in science, but identifying parameters that explain experimental measurements is challenging. In particular, although ODEs are differentiable and would allow for gradient-based parameter optimization, the nonlinear dynamics of ODEs often lead to many local minima and extreme sensitivity to initial conditions. We therefore propose diffusion tempering, a novel regularization technique for probabilistic numerical methods which improves convergence of gradient-based parameter optimization in ODEs. By iteratively reducing a noise parameter of the probabilistic integrator, the proposed method converges more reliably to the true parameters. We demonstrate that our method is effective for dynamical systems of different complexity and show that it obtains reliable parameter estimates for a Hodgkin-Huxley model with a practically relevant number of parameters.
format Preprint
id arxiv_https___arxiv_org_abs_2402_12231
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Diffusion Tempering Improves Parameter Estimation with Probabilistic Integrators for Ordinary Differential Equations
Beck, Jonas
Bosch, Nathanael
Deistler, Michael
Kadhim, Kyra L.
Macke, Jakob H.
Hennig, Philipp
Berens, Philipp
Machine Learning
Ordinary differential equations (ODEs) are widely used to describe dynamical systems in science, but identifying parameters that explain experimental measurements is challenging. In particular, although ODEs are differentiable and would allow for gradient-based parameter optimization, the nonlinear dynamics of ODEs often lead to many local minima and extreme sensitivity to initial conditions. We therefore propose diffusion tempering, a novel regularization technique for probabilistic numerical methods which improves convergence of gradient-based parameter optimization in ODEs. By iteratively reducing a noise parameter of the probabilistic integrator, the proposed method converges more reliably to the true parameters. We demonstrate that our method is effective for dynamical systems of different complexity and show that it obtains reliable parameter estimates for a Hodgkin-Huxley model with a practically relevant number of parameters.
title Diffusion Tempering Improves Parameter Estimation with Probabilistic Integrators for Ordinary Differential Equations
topic Machine Learning
url https://arxiv.org/abs/2402.12231