PGMNO: A physics-Guided mamba neural operator framework for partial differential equations
Yanan Guo1, Junqiang Song2, Xiaoqun Cao3
1College of Meteorology and Oceanography, National University of Defense Technology, No.109 Deya Road, Changsha, 410073, Hunan, China; College of Computer Science, National University of Defense Technology, No.109 Deya Road, Changsha, 410073, Hunan, China; Department of Atmospheric and Oceanic Sciences, School of Physics, Peking University, No.5 Yiheyuan Road, Beijing, 100871, Beijing, China.
Abstract:
Accurately modeling the long-term evolution of complex physical systems governed by partial differential equations (PDEs) remains a central challenge in operator learning. In this work, we introduce the Physics-Guided Mamba Neural Operator (PGMNO), a framework specifically designed to capture long-range spatiotemporal dependencies. PGMNO overcomes the limitations of conventional single-step and Transformer-based operators by unifying linear multistep numerical methods with structured state space models (SSMs). To ensure temporal stability, the framework employs multistep temporal modeling in the forward pass, complemented by an implicit backward differentiation formula (BDF)-based scheme during training. Simultaneously, PGMNO leverages the efficient kernel integration properties of SSMs to achieve scalable and stable operator learning. We validate PGMNO on diverse PDE benchmarks. Experimental results demonstrate that PGMNO consistently outperforms state-of-the-art models in prediction accuracy, computational efficiency, and long-term stability. Additionally, the framework exhibits strong resolution-invariant extrapolation and generalization across varying spatial discretizations. By prioritizing clear methodological articulation, this work underscores the potential of unifying numerical integration with state space modeling to build robust surrogate solvers for PDE-governed dynamical systems.
Related Concept Videos
Mechanistic Models: Compartment Models in Algorithms for Numerical Problem Solving
In individual population analyses, different algorithms are employed, such as Cauchy's method, which uses a...
Modeling with Differential Equations
Separable Differential Equations
Linear Approximation in Time Domain
For a simple pendulum with a mass evenly distributed along its length and the center of mass located at half the pendulum's length,...
Relation between Mathematical Equations and Block Diagrams
Differential Equations: Problem Solving
