Deep Embedded Multiplicative DMD for Algebra-Preserving Koopman Learning

2026-06-03Machine Learning

Machine Learning
AI summary

The authors propose DeepMDMD, a method that learns simpler hidden coordinates to turn complex nonlinear dynamics into easier linear problems. Their approach enforces exact mathematical rules (a product rule) on these learned coordinates, improving how the system captures the dynamics. By alternately updating operators and clustering the latent space, the method creates a compact and accurate representation of the dynamics, even for complicated systems like chaotic flows. Compared to previous methods, their approach reduces errors and provides more stable predictions under noise. Overall, the authors show that learning the right coordinates while enforcing key algebraic constraints leads to better understanding and forecasting of complex systems.

Koopman theorynonlinear dynamicsdynamic mode decompositionlatent spacespectral problemHamiltonian systemschaotic systemsfluid dynamicsoperator theoryspectral pollution
Authors
Kelan Gray, Finlay Brown, Nicolas Boullé, Matthew J. Colbrook
Abstract
Koopman theory turns nonlinear dynamics into a linear spectral problem. In computation, however, everything depends on a hard finite-dimensional choice: the observables must be expressive, nearly invariant under the dynamics, and, ideally, compatible with composition. Deep Koopman methods learn flexible coordinates, whereas structure-preserving methods enforce operator identities on fixed dictionaries. We combine these ideas by introducing Deep Embedded Multiplicative Dynamic Mode Decomposition (DeepMDMD), a method that learns a latent space and a partition of it, while enforcing the Koopman product rule as an exact algebraic constraint. Training alternates between an exact multiplicative operator update and a differentiable latent-clustering step that promotes Koopman closure. The result is a finite transition map on learned latent cells. Its nonzero spectrum lies on the unit circle, its dictionary is shaped by the dynamics rather than by ambient geometry, and forecasts are made in latent coordinates before being decoded to physical space. Across Hamiltonian, chaotic, and fluid examples, DeepMDMD learns dictionaries that are far more compact and dynamically coherent than those produced by geometric MDMD partitions. It reduces spectral pollution, reveals richer continuous-spectrum structure, and gives stable forecasts under severe noise. In high-dimensional flows, including a 158,624-dimensional cylinder wake and a noisy $Re=20,000$ lid-driven cavity, it preserves coherent structures and long-time spectral statistics where state-space MDMD fails. These results suggest a practical rule for Koopman learning: learn the coordinates, constrain the algebra.