Mamba neural operator: Who wins? transformers vs. state-space models for PDEs

• We introduce the Mamba Neural Operator (MNO) and establish the first theoretical link between neural operators and structured state-space models, revealing their shared mathematical foundations. • MNO provides global receptive fields with linear complexity, outperforming attention-based Transformers while remaining computationally efficient and scalable. • Extensive experiments across four 2D PDEs (Darcy Flow, Shallow Water, Diffusion–Reaction, CFD2D) show that MNO achieves state-of-the-art accuracy, with up to around 90% error reduction over leading Transformer baselines. • MNO greatly improves long-time stability and robustness, reducing error accumulation in temporal rollouts and enabling more reliable PDE forecasting. Partial differential equations (PDEs) are widely used to model complex physical systems, but solving them efficiently remains a significant challenge. Recently, Transformers have emerged as the preferred architecture for PDEs due to their ability to capture intricate dependencies. However, they struggle with representing continuous dynamics and long-range interactions. To overcome these limitations, we introduce the Mamba Neural Operator (MNO), a novel framework that enhances neural operator-based techniques for solving PDEs. MNO establishes a formal theoretical connection between structured state-space models (SSMs) and neural operators, offering a unified structure that can adapt to diverse architectures, including Transformer-based models. By leveraging the structured design of SSMs, MNO captures long-range dependencies and continuous dynamics more effectively than traditional Transformers. Through extensive analysis, we show that MNO significantly boosts the expressive power and accuracy of neural operators, making it not just a complement but a superior framework for PDE-related tasks, bridging the gap between efficient representation and accurate solution approximation.

Authors

Institutions

Publication Details

Journal
Journal of Computational Physics
Published
2025-12-06
DOI
https://doi.org/10.1016/j.jcp.2025.114567
Primary Topic
Model Reduction and Neural Networks
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Mamba neural operator: Who wins? transformers vs. state-space models for PDEs

Carola‐Bibiane Schönlieb, Chun-Wun Cheng, Angelica I. Aviles-Rivero, Yi Zhang et al.
Journal of Computational Physics
Model Reduction and Neural Networks
article

Mamba neural operator: Who wins? transformers vs. state-space models for PDEs

Carola‐Bibiane Schönlieb, Chun-Wun Cheng, Angelica I. Aviles-Rivero, Yi Zhang, Guang Yang, Jiahao Huang
article en

Abstract

• We introduce the Mamba Neural Operator (MNO) and establish the first theoretical link between neural operators and structured state-space models, revealing their shared mathematical foundations. • MNO provides global receptive fields with linear complexity, outperforming attention-based Transformers while remaining computationally efficient and scalable. • Extensive experiments across four 2D PDEs (Darcy Flow, Shallow Water, Diffusion–Reaction, CFD2D) show that MNO achieves state-of-the-art accuracy, with up to around 90% error reduction over leading Transformer baselines. • MNO greatly improves long-time stability and robustness, reducing error accumulation in temporal rollouts and enabling more reliable PDE forecasting. Partial differential equations (PDEs) are widely used to model complex physical systems, but solving them efficiently remains a significant challenge. Recently, Transformers have emerged as the preferred architecture for PDEs due to their ability to capture intricate dependencies. However, they struggle with representing continuous dynamics and long-range interactions. To overcome these limitations, we introduce the Mamba Neural Operator (MNO), a novel framework that enhances neural operator-based techniques for solving PDEs. MNO establishes a formal theoretical connection between structured state-space models (SSMs) and neural operators, offering a unified structure that can adapt to diverse architectures, including Transformer-based models. By leveraging the structured design of SSMs, MNO captures long-range dependencies and continuous dynamics more effectively than traditional Transformers. Through extensive analysis, we show that MNO significantly boosts the expressive power and accuracy of neural operators, making it not just a complement but a superior framework for PDE-related tasks, bridging the gap between efficient representation and accurate solution approximation.

Journal of Computational PhysicsVol. 548
King's College London (GB), University of Cambridge (GB), Southern University of Science and Technology (CN), Lung Institute (US), Imperial College London (GB), Tsinghua University (CN)
Openalex Percentile: Top 69%
Model Reduction and Neural Networks
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.