Differential Equations for Machine Learning

Thumbnail Image

Files

isbn9789526406664.pdf (14.8 MB)

URL

Journal Title

Journal ISSN

Volume Title

School of Science | Doctoral thesis (article-based) | Defence date: 2022-02-18
Unless otherwise stated, all rights belong to the author. You may download, display and print this publication for Your own personal use. Commercial use is prohibited.

Authors

Date

2022

Department

Tietotekniikan laitos
Department of Computer Science

Major/Subject

Mcode

Degree programme

Language

en

Pages

106 + app. 82

Series

Aalto University publication series DOCTORAL THESES, 7/2022

Abstract

Mechanistic models express novel hypotheses for an observed phenomenon by constructing mathematical formulations of causal mechanisms. As opposed to this modeling paradigm, machine learning approaches learn input-output mappings by complicated and often non-interpretable models. While requiring large chunks of data for successful training and downstream performance,the resulting models can come with universal approximation guarantees. Historically, differential equations (DEs) developed in physics, economics, engineering, and numerous other fields have relied on the principles of mechanistic modeling. Despite providing causality and interpretability that machine learning approaches usually lack, mechanistic differential equation models tend tocarry oversimplified assumptions. In this dissertation, we aim to bring these two worlds together by demonstrating how machine learning problems can be tackled by means of differential equations, and how differential equation models can benefit from modern machine learning tools. First, we examine the problems in which mechanistic modeling becomes too difficult, which include the cases with partial knowledge about the observed system and with an excessive number of interactions. Such limitations complicate the process of constructing mathematical descriptions of the phenomenon of interest. To bypass this, we propose to place Gaussian process priors to the time differential and diffusion functions of unknown ordinary (ODEs) and stochastic differential equations (SDEs), and approximate the resulting intractable posterior distribution. We demonstrate that the model can estimate unknown dynamics from sparse and noisy observations. Motivated by the fact that our proposed approach is unable to learn sequences obtained by transforming the ODE states, we develop a new technique that can simultaneously embed the observations into a latent space, and learn an ODE system in the embedding space. Our new model infers the dynamics using Bayesian neural networks for uncertainty handling and more expressive power. We furthermore explicitly decompose the latent space into momentum and position components, which leads to increased predictive performance on a variety of physical tasks.Our next task concerns another problem involving DEs, namely, non-convex optimization. By carefully crafting the drift and diffusion functions of an SDE, we first obtain a stochastic gradient MCMC algorithm. Tuning a temperature variable in the proposed algorithm allows the chain to converge to the global minimum of a non-convex loss surface. We significantly speed up the convergence by using second-order Hessian information in an asynchronous parallel framework. Lastly, we explore how reinforcement learning problems can benefit from neural network based ODE models. In particular, we propose to learn dynamical systems controlled by external actions by a novel, uncertainty-aware neural ODE model. The inferred model, in turn, is utilized for learning optimal policy functions. We illustrate that our method is robust to both noisy and irregularly sampled data sequences, which poses major challenges to traditional methods.

Description

Defence is held on 18.2.2022 12:15 – 16:15 (Zoom), https://aalto.zoom.us/j/61873808631

Supervising professor

Lähdesmäki, Harri, Prof., Aalto University, Department of Computer Science, Finland

Thesis advisor

Lähdesmäki, Harri, Prof., Aalto University, Finland
Heinonen, Markus, Dr., Aalto University, Finland

Keywords

machine learning, differential equations, neural networks, Gaussian processes

Other note

Parts

  • [Publication 1]: Markus Heinonen, Çagatay Yıldız, Henrik Mannerström, Jukka Intosalmi, Harri Lähdesmäki. Learning Unknown ODE Models with Gaussian Processes. In International Conference on Machine Learning, Stockholm, pages 1959–1968, vol.80, July 2018.
    Full text in Acris/Aaltodoc: http://urn.fi/URN:NBN:fi:aalto-201907304456
  • DOI: 10.1109/MLSP.2018.8516991 View at publisher
  • [Publication 3]: Çagatay Yıldız, Markus Heinonen, Harri Lähdesmäki. ODE2VAE: Deep Generative Second Order ODEs with Bayesian Neural Networks. In Neural Information Processing Systems, Vancouver, pages 13412–13421, vol.32, December 2019
  • [Publication 4]: Umut Simsekli, Çagatay Yıldız, Than Huy Nguyen, Taylan Cemgil and Gael Richard. Asynchronous Stochastic Quasi-Newton MCMC for Non-Convex Optimization. In International Conference on Machine Learning, Stockholm, pages 4674–4683, vol.80, July 2018. Full ext in Acris/Aaltodoc: http://urn.fi/URN:NBN:fi:aalto-201907304448.
  • [Publication 5]: Çagatay Yıldız, Markus Heinonen, Harri Lähdesmäki. Continuous-Time Model-Based Reinforcement Learning. In International Conference on Machine Learning, Virtual, pages 12009–12018, vol.139, July 2021.
    Full text in Acris/Aaltodoc: http://urn.fi/URN:NBN:fi:aalto-202108258419

Citation

Permanent link to this item

https://urn.fi/URN:ISBN:978-952-64-0666-4

Collections

[diss] Perustieteiden korkeakoulu / SCI