Deep learning for universal linear embeddings of nonlinear dynamics
Bethany LuschJ. KutzS. Brunton
Develops a deep autoencoder framework that discovers interpretable Koopman eigenfunctions from data, enabling linear prediction and control of strongly nonlinear dynamical systems with both discrete and continuous spectra.
Real-world engineering, physical, and biological systems are predominantly nonlinear and often lack known governing equations. While linear systems possess well-established methods for accurate prediction and automated control, representing complex nonlinear dynamics within a linear framework remains a major challenge. Koopman operator theory provides a mathematical foundation to globally transform nonlinear systems into linear ones via coordinate functions known as eigenfunctions. However, discovering these eigenfunctions has historically required difficult manual derivations, produced uninterpretable black-box models, or failed entirely when dealing with systems exhibiting continuous frequency spectra.
The article demonstrates a deep learning framework designed to discover interpretable, low-dimensional coordinate representations that globally linearize strongly nonlinear dynamics. The primary objective is to construct parsimonious linear models from trajectory data while handling both discrete and continuous eigenvalue spectra.
To achieve this, the authors designed a modified deep auto-encoder network trained on simulated trajectory datasets ranging from 5,000 to 20,000 samples. The architecture encodes high-dimensional state data into low-dimensional latent coordinates, advances those coordinates using a linear dynamics matrix, and decodes them back to the original state space. A specialized auxiliary network parameterizes continuously shifting frequencies and damping rates across the state space. The models were evaluated across three classic benchmark systems: a canonical two-state nonlinear system with discrete eigenvalues, a nonlinear pendulum exhibiting a continuous frequency spectrum, and a high-dimensional fluid flow past a cylinder at Reynolds number 100.
The key findings confirm that the deep learning architecture accurately linearizes complex dynamics while keeping model dimensionality minimal. First, the framework achieved very low test errors, on the order of 10⁻⁷ to 10⁻⁶ across all benchmarks, confirming precise reconstruction and long-term state prediction. Second, for the continuous-spectrum pendulum, the auxiliary network captured the energy-dependent frequency shift using only a single pair of conjugate eigenfunctions, avoiding the cumbersome infinite harmonic expansions required by conventional techniques. Third, the model successfully mapped high-dimensional fluid vortex shedding onto an interpretable two-to-three coordinate manifold, accurately predicting transient trajectories that started far off the natural limit cycle.
These results demonstrate that nonlinear dynamical modeling can be made both computationally tractable and physically interpretable without sacrificing accuracy. Enabling linear representations of complex dynamics significantly reduces the computational overhead and operational risk associated with nonlinear predictive modeling, sensor estimation, and feedback control design in applied engineering settings.
Organizations developing data-driven modeling and control architectures should consider adopting auto-encoder frameworks with auxiliary parameter networks when modeling oscillatory or continuous-spectrum dynamics. Future research should prioritize developing automated techniques to detect latent coordinate dimensions and extending the architecture to higher-dimensional systems such as turbulent fluid flow, neuroscience, and epidemiology.
Confidence in these findings is high for low-dimensional systems governed by smooth attractors. However, the reader should note key limitations: deep learning models act primarily as sophisticated interpolation tools with limited capacity to extrapolate beyond their training data distribution. Consequently, applying this approach requires sufficient volume and diversity in training data, particularly regarding transient states, as well as upfront manual tuning of the latent subspace dimension.
- Paper: Laplacian Eigenmaps and Spectral Techniques for Embedding and Clustering, Mikhail Belkin et al. (2001). It provides foundational spectral theory for nonlinear manifold learning and eigenmap discovery, establishing the core geometric intuition for finding intrinsic coordinate embeddings of complex data.
- Paper: NICE: Non-linear Independent Components Estimation, Laurent Dinh et al. (2014). It introduces deep invertible nonlinear transformations and coupling architectures that underpin the design of autoencoders for coordinate transformation and latent linearization.
- Paper: Tutorial on Variational Autoencoders, Carl Doersch (2016). It explains the principles of learning low-dimensional continuous latent coordinate spaces with neural network encoders and decoders.
- Paper: Learning nonlinear operators via DeepONet based on the universal approximation theorem of operators, Lu Lu et al. (2021). It generalizes neural operator learning from finite-dimensional state-space embeddings to infinite-dimensional function space operators.
- Paper: Fourier Neural Operator for Parametric Partial Differential Equations, Zongyi Li et al. (2020). It extends data-driven operator approximation for continuous dynamical systems by learning resolution-invariant operators directly in Fourier modal space.
- Paper: Who Said Neural Networks Aren't Linear?, Nimrod Berman et al. (2026). It builds directly on the concept of discovering coordinate transformations that globally linearize complex neural dynamics by sandwiching linear operators between invertible networks.
- Paper: Learning Latent Dynamics for Planning from Pixels, Danijar Hafner et al. (2018). It applies learned low-dimensional latent dynamical embeddings directly to continuous control and trajectory planning.
- Paper: Neural Ordinary Differential Equations, Ricky T. Q. Chen et al. (2018). It provides a continuous-time formulation for latent state evolution parameterized by neural networks, extending discrete-step latent dynamical models.
