Interpretable Functional Koopman Learning with Non-Markovian Closure for Spatiotemporal Systems
Wanfeng Lu, He Ma, Wei Lin, Qunxi Zhu
Abstract
Precise prediction of spatiotemporal dynamics over predictive horizons is constrained by the computational cost of high-fidelity solvers and the sparsity, noise, and irregularity of data. We introduce MERLIN, a Koopman-based framework that lifts dynamics to the evolution of learned observation functionals with near-linear progression, enabling full-field reconstruction at arbitrary resolutions. Theoretically, we develop a functional Koopman theory for PDEs and compensate for the loss of finite-dimensional linear invariance via the Mori-Zwanzig formalism, which augments the linear backbone with non-Markovian memory terms to improve predictive accuracy. Practically, MERLIN employs discretization-invariant function encoders that map partial, irregular observations to observables, and resolution-free function decoders that reconstruct states at arbitrary query points. Training under linear constraints yields an interpretable, low-dimensional model that captures principal modes and supports reduced-order modeling, while memory correction further enables stable long-horizon rollouts even in ultra-low-dimensional latent spaces. Our code is available at: https://github.com/ RobinLufdu/MERLIN.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f460b66c-54d6-42c3-8eca-0f233c4fe6efBuilds on14
- Implicit Neural Representations with Periodic Activation FunctionsVincent Sitzmann, Julien N. P. Martel, Alexander W. Bergman, David B. Lindell et al.NeurIPS 2020 · 4,008 citations
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Forecasting Sequential Data Using Consistent Koopman AutoencodersOmri Azencot, N. Benjamin Erichson, Vanessa Lin, Michael W. MahoneyICML 2020 · 203 citations
- Scalable Transformer for PDE Surrogate ModelingZijie Li, Dule Shu, Amir Barati FarimaniNeurIPS 2023 · 188 citations
- Multiplicative Filter NetworksRizal Fathony, Anit Kumar Sahu, Devin Willmott, J. Zico KolterICLR 2021 · 185 citations
Related papers
- Efficient Dynamics Modeling in Interactive Environments with Koopman TheoryArnab Kumar Mondal, Siba Smarak Panigrahi, Sai Rajeswar, Kaleem Siddiqi et al.ICLR 2024 · 12 citations
- On the Benefits of Memory for Modeling Time-Dependent PDEsRicardo Buitrago Ruiz, Tanya Marwah, Albert Gu, Andrej RisteskiICLR 2025
- Course Correcting Koopman RepresentationsMahan Fathi, Clement Gehring, Jonathan Pilault, David Kanaa et al.ICLR 2024 · 1 citation
- Physics-informed Reduced Order Modeling of Time-dependent PDEs via Differentiable SolversNima Hosseini Dashtbayaz, Hesam Salehipour, Adrian Butscher, Nigel MorrisNeurIPS 2025 · 3 citations
- NOMAD: Nonlinear Manifold Decoders for Operator LearningJacob H. Seidman, Georgios Kissas, Paris Perdikaris, George J. PappasNeurIPS 2022 · 125 citations
