Evolve Smoothly, Fit Consistently: Learning Smooth Latent Dynamics For Advection-Dominated Systems
Zhong Yi Wan, Leonardo Zepeda-Núñez, Anudhyan Boral, Fei Sha
Abstract
We present a data-driven, space-time continuous framework to learn surrogate models for complex physical systems described by advection-dominated partial differential equations. Those systems have slow-decaying Kolmogorov n-width that hinders standard methods, including reduced order modeling, from producing high-fidelity simulations at low cost. In this work, we construct hypernetwork-based latent dynamical models directly on the parameter space of a compact representation network. We leverage the expressive power of the network and a specially designed consistency-inducing regularization to obtain latent trajectories that are both low-dimensional and smooth. These properties render our surrogate models highly efficient at inference time. We show the efficacy of our framework by learning models that generate accurate multi-step rollout predictions at much faster inference speed compared to competitors, for several challenging examples.
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 9c1bb86c-ca0e-4700-be14-fac07be797a4Cited by top-tier papers6
- DYffusion: A Dynamics-informed Diffusion Model for Spatiotemporal ForecastingSalva Rühling Cachay, Bo Zhao, Hailey Joren, Rose YuNeurIPS 2023 · 164 citations
- DySLIM: Dynamics Stable Learning by Invariant Measure for Chaotic SystemsYair Schiff, Zhong Yi Wan, Jeffrey B. Parker, Stephan Hoyer et al.ICML 2024 · 30 citations
- Unifying Predictions of Deterministic and Stochastic Physics in Mesh-reduced Space with Sequential Flow Generative ModelLuning Sun, Xu Han, Han Gao, Jian-Xun Wang et al.NeurIPS 2023 · 22 citations
- CoLoRA: Continuous low-rank adaptation for reduced implicit neural modeling of parameterized partial differential equationsJules Berman, Benjamin PeherstorferICML 2024 · 17 citations
- Learning Chaos In A Linear WayXiaoyuan Cheng, Yi He, Yiming Yang, Xiao Xue et al.ICLR 2025
Builds on8
- Fourier Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola Borislavov Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.ICLR 2021 · 3,911 citations
- Learning to Simulate Complex Physics with Graph NetworksAlvaro Sanchez-Gonzalez, Jonathan Godwin, Tobias Pfaff, Rex Ying et al.ICML 2020 · 1,439 citations
- Learning Mesh-Based Simulation with Graph NetworksTobias Pfaff, Meire Fortunato, Alvaro Sanchez-Gonzalez, Peter W. BattagliaICLR 2021 · 1,175 citations
- Multipole Graph Neural Operator for Parametric Partial Differential EquationsZongyi Li, Nikola B. Kovachki, Kamyar Azizzadenesheli, Burigede Liu et al.NeurIPS 2020 · 569 citations
- Message Passing Neural PDE SolversJohannes Brandstetter, Daniel E. Worrall, Max WellingICLR 2022 · 410 citations
Related papers
- Low-Rank Registration Based Manifolds for Convection-Dominated PDEsRambod Mojgani, Maciej BalajewiczAAAI 2021 · 25 citations
- Learning to Accelerate Partial Differential Equations via Latent Global EvolutionTailin Wu, Takashi Maruyama, Jure LeskovecNeurIPS 2022 · 49 citations
- CALM-PDE: Continuous and Adaptive Convolutions for Latent Space Modeling of Time-dependent PDEsJan Hagnberger, Daniel Musekamp, Mathias NiepertNeurIPS 2025 · 6 citations
- Model-Agnostic Knowledge Guided Correction for Improved Neural Surrogate RolloutBharat Srikishan, Daniel O'Malley, Mohamed Mehana, Nicholas Lubbers et al.ICLR 2025
- Physics-informed Reduced Order Modeling of Time-dependent PDEs via Differentiable SolversNima Hosseini Dashtbayaz, Hesam Salehipour, Adrian Butscher, Nigel MorrisNeurIPS 2025 · 3 citations
