Latent Matters: Learning Deep State-Space Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Klushyn, Alexej, Kurle, Richard, Soelch, Maximilian, Cseke, Botond, van der Smagt, Patrick
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917296841162752
author Klushyn, Alexej
Kurle, Richard
Soelch, Maximilian
Cseke, Botond
van der Smagt, Patrick
author_facet Klushyn, Alexej
Kurle, Richard
Soelch, Maximilian
Cseke, Botond
van der Smagt, Patrick
contents Deep state-space models (DSSMs) enable temporal predictions by learning the underlying dynamics of observed sequence data. They are often trained by maximising the evidence lower bound. However, as we show, this does not ensure the model actually learns the underlying dynamics. We therefore propose a constrained optimisation framework as a general approach for training DSSMs. Building upon this, we introduce the extended Kalman VAE (EKVAE), which combines amortised variational inference with classic Bayesian filtering/smoothing to model dynamics more accurately than RNN-based DSSMs. Our results show that the constrained optimisation framework significantly improves system identification and prediction accuracy on the example of established state-of-the-art DSSMs. The EKVAE outperforms previous models w.r.t. prediction accuracy, achieves remarkable results in identifying dynamical systems, and can furthermore successfully learn state-space representations where static and dynamic features are disentangled.
format Preprint
id arxiv_https___arxiv_org_abs_2602_23050
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Latent Matters: Learning Deep State-Space Models
Klushyn, Alexej
Kurle, Richard
Soelch, Maximilian
Cseke, Botond
van der Smagt, Patrick
Machine Learning
Deep state-space models (DSSMs) enable temporal predictions by learning the underlying dynamics of observed sequence data. They are often trained by maximising the evidence lower bound. However, as we show, this does not ensure the model actually learns the underlying dynamics. We therefore propose a constrained optimisation framework as a general approach for training DSSMs. Building upon this, we introduce the extended Kalman VAE (EKVAE), which combines amortised variational inference with classic Bayesian filtering/smoothing to model dynamics more accurately than RNN-based DSSMs. Our results show that the constrained optimisation framework significantly improves system identification and prediction accuracy on the example of established state-of-the-art DSSMs. The EKVAE outperforms previous models w.r.t. prediction accuracy, achieves remarkable results in identifying dynamical systems, and can furthermore successfully learn state-space representations where static and dynamic features are disentangled.
title Latent Matters: Learning Deep State-Space Models
topic Machine Learning
url https://arxiv.org/abs/2602.23050