Pub-AI: AI in Science

Latent Dynamics · 2026-09-30

SMORE: Stability-Promoting Mesh-Agnostic Model Reduction for Time-Dependent PDEs

Yangyuan Li, Weichao Li, Shaowu Pan

arXiv:2609.33205PDF

Auto-summarized: this summary was generated by a language model from the paper’s text and has not been reviewed by an editor. Check the paper before relying on it. Benchmark numbers appear only after a human has verified them.

TL;DR

SMORE is a mesh-agnostic reduced-order model for time-dependent PDEs. It uses an INR autodecoder to map sparse spatial measurements to a latent state, then evolves the latent state with structured latent dynamics (linear or linear-quadratic). A Lyapunov-guided stability regularization is added to promote stable long-horizon rollouts, and the authors provide theoretical guarantees under structural assumptions.

Why it matters

The authors target a common failure mode of neural PDE surrogates: poor temporal stability leading to unstable multistep rollouts and exploding gradients. SMORE addresses this by combining mesh-agnostic implicit decoding with stability-promoting latent dynamics, aiming to improve long-horizon rollout generalization and robustness while retaining the ability to predict continuous PDE fields from sparse initial measurements.

Method

  • Mesh-agnostic model reduction: an INR autodecoder reconstructs PDE solution fields at queried spatial coordinates from inferred latent codes (FiLM-modulated decoder).
  • Structured latent dynamics: SMORE-Koopman uses linear latent dynamics; SMORE-LRLQ uses low-rank linear-quadratic latent dynamics.
  • Stability and training: Lyapunov-guided stability regularization is added to the latent dynamics loss; training uses scheduled sampling and pointwise minibatching for memory-efficient decoding.
Abstract (from arXiv)

High-fidelity simulations of time-dependent partial differential equations (PDEs) are computationally expensive, motivating data-driven reduced-order surrogates for many-query tasks such as uncertainty quantification, design optimization, data assimilation, and optimal control. However, existing surrogate models often exhibit poor temporal stability, which can lead to unstable rollouts and exploding gradients during backpropagation, especially in multistep long-horizon forecasting. To address this, we propose SMORE, a mesh-agnostic framework for model order reduction of time-dependent PDEs. Its latent dynamics are trained with Lyapunov-guided stability regularization, which promotes stable long-horizon rollouts. We provide theoretical guarantees under the stated structural assumptions. Beyond forecasting PDE evolution, the learned latent dynamics, which are interpretable and linear or linear-quadratic, could bring benefits for downstream tasks such as data assimilation and optimal control. Moreover, our framework is capable of predicting continuous PDE solution fields from sparse measurements of the initial condition. We evaluate SMORE on a range of problems, including wave propagation, the Navier-Stokes equations, and the shallow water equations. Our results show that it improves long-horizon rollout generalization and empirical robustness, and achieves competitive accuracy at comparable parameter budgets relative to competitive baselines including DINo, FNO, CNO, and Transolver.

Related papers