Shaping the Koopman dictionary by learning on the Grassmannian
Roland Schurig, Pieter van Goor, Karl Worthmann, Rolf Findeisen
TL;DR
This work addresses shaping the Koopman dictionary for nonlinear dynamics by learning on the Grassmannian to produce approximately invariant subspaces. It combines EDMD with a subspace design that fixes a prior observable set and optimizes the remaining components via a trust-region method on the Grassmann manifold, ensuring efficient, basis-invariant state predictions. The approach yields improved prediction accuracy and generalization, demonstrated on a Duffing oscillator, and offers a flexible framework for incorporating system structure and alternative optimization criteria. By leveraging differential-geometric concepts, the paper contributes a principled, scalable route to data-driven Koopman surrogates for complex dynamics.
Abstract
Extended dynamic mode decomposition (EDMD) is a powerful tool to construct linear predictors of nonlinear dynamical systems by approximating the action of the Koopman operator on a subspace spanned by finitely many observable functions. However, its accuracy heavily depends on the choice of the observables, which remains a challenge. We propose a systematic framework to identify and shape observable dictionaries, reduce projection errors, and achieve approximately invariant subspaces. To this end, we leverage optimisation on the Grassmann manifold and exploit inherent geometric properties for computational efficiency. Numerical results demonstrate improved prediction accuracy and efficiency. In conclusion, we propose a novel approach to efficiently shape the Koopman dictionary using differential-geometric concepts for optimisation on manifolds resulting in enhanced data-driven Koopman surrogates for nonlinear dynamical systems.
