Bilevel optimization for data-driven learning of Koopman embeddings using kernel-based autoencoders

Abstract

Koopman operator theory provides a linear framework for analyzing nonlinear dynamical systems and has become a major tool for data-driven modeling. A central challenge, however, is that finite-dimensional approximations computed by methods such as extended dynamic mode decomposition (EDMD) require the dictionary to be specified a priori. Recent machine-learning approaches address this limitation by learning the dictionary from data, predominantly using artificial neural network (ANN) autoencoder architectures. Although kernel methods offer an alternative with greater interpretability and tractability for theoretical analysis, they have received little attention in this setting. We introduce extended dynamic mode decomposition with kernel-based dictionary learning (EDMD-kDL), a kernel-based method for learning finite-dimensional Koopman embeddings directly from data. The method combines ideas from collocation methods and bilevel optimization to simultaneously learn a kernel dictionary and the corresponding Koopman approximation. We evaluate EDMD-kDL against state-of-the-art ANN-based approaches on a range of numerical experiments, including global sea-surface-temperature forecasting and learning directly from video data. Across all tested settings, EDMD-kDL achieves performance comparable to or better than the ANN-based methods. Moreover, in contrast to standard kernel methods, the proposed approach is scalable to large datasets by design since the size of the required kernel matrices depends on the number of collocation points rather than the size of the training dataset.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

More in Machine Learning

• arXiv preprint · 8 October 2026

  Understanding Latent-Dimension Scaling in Dynamical-System Learning through Spectral Reliability

  Itsushi Sakata, Yuta Miyauchi and Yoshinobu Kawahara
• arXiv preprint · 9 October 2026

  Asymptotic Analysis of Discrete-Time Self-Attention Dynamics in Transformers

  Claudio Altafini
• arXiv preprint · 13 October 2026

  Contractor Filters

  Simon Rohou
• arXiv preprint · 7 October 2026

  Koopman Observers for Diffusion Acceleration: Correcting Feature Forecasts with Shallow Measurements

  Hanru Bai, Yuanchao Xu and Fengyi Li
• arXiv preprint · 15 October 2026

  SALTO: Solver-Anchored Lifted Transfer Operators for Nonlinear Event Inference

  Qi Wang and Rylie Foster

All new papers in Machine Learning