Home > Research > Publications & Outputs > Recurrent Kalman networks

Electronic data

  • 19icml_kalman

    Accepted author manuscript, 522 KB, PDF document

    Available under license: Unspecified

  • 19icml_kalman_supp

    Rights statement: Please note: this is the Appendix to the article.

    Other version, 416 KB, PDF document

Links

View graph of relations

Recurrent Kalman networks: factorized inference in high-dimensional deep feature spaces

Research output: Contribution in Book/Report/Proceedings - With ISBN/ISSNConference contribution/Paperpeer-review

Published
Close
Publication date13/06/2019
Host publicationProceedings of Machine Learning Research (PMLR)
Pages544-552
Number of pages9
Volume97
<mark>Original language</mark>English

Publication series

NameProceedings of Machine Learning Research (PMLR)
Volume97
ISSN (electronic)2640-3498

Abstract

In order to integrate uncertainty estimates into deep time-series modelling, Kalman Filters (KFs) have been integrated with deep learning models; however, such approaches typically rely on approximate inference techniques such as variational inference which makes learning more complex and often less scalable due to approximation errors. We propose a new deep approach to Kalman filtering which can be learned directly in an end-to-end manner using backpropagation without additional approximations. Our approach uses a high-dimensional factorized latent state representation for which the Kalman updates simplify to scalar operations and thus avoids hard to backpropagate, computationally heavy and potentially unstable matrix inversions. Moreover, we use locally linear dynamic models to efficiently propagate the latent state to the next time step. The resulting network architecture, which we call Recurrent Kalman Network (RKN), can be used for any time-series data, similar to a LSTM (Hochreiter & Schmidhuber, 1997) but uses an explicit representation of uncertainty. As shown by our experiments, the RKN obtains much more accurate uncertainty estimates than an LSTM or Gated Recurrent Units (GRUs) (Cho et al., 2014) while also showing a slightly improved prediction performance and outperforms various recent generative models on an image imputation task.