When Attention Beats Fourier: Multi-Scale Transformers for PDE Solving on Irregular Domains
On the map: State-space sequence models (S4, Mamba) → Time series forecasting with SSMs
Abstract (arXiv)
We study the problem of architecture selection for deep learning models trained to solve partial differential equations (PDEs), asking when transformer-based architectures with learned attention outperform Fourier-domain neural operators. We introduce the Multi-Scale Attention Transformer (msat), a deep learning architecture that encodes spatiotemporal solution histories as token sequences and trains end-to-end via a composite supervised objective with optional physics-informed regularization terms. We conduct a comprehensive empirical evaluation against nine baselines -- including physics-informed neural networks (PINNs), neural operators (FNO, DeepONet, GNOT), and state-space models (Mamba-NO) -- across five benchmark problems from the PINNacle suite, using identical train/test splits and reference data for all methods. msat achieves state-of-the-art generalization on complex geometry problems (L2rel = 0.0101 on Heat2D-CG, a 3.7× improvement over FNO) at 34,s total inference vs. 120,812,s for Mamba-NO. Ablation studies over the physics regularization component reveal a precise inductive bias tradeoff: physics priors reduce test error on diffusion-dominated problems but degrade generalization on chaotic and recirculating-flow regimes, directly characterizing the prior misspecification boundary. Approximation error bounds as a function of domain boundary complexity kappa provide a theoretical basis for these empirical findings and a principled rule for architecture selection.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Time series forecasting with SSMs, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 11 of its 16 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
11 of 16 references.
References · 11
- 2017Attention Is All You Needcited by 59,932
- 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
- 2021Characterizing possible failure modes in physics-informed neural networkscited by 632
- 2019DeepXDE: A deep learning library for solving differential equationscited by 617
- 2021Physics-Informed Neural Operator for Learning Partial Differential Equationscited by 516
- 2021Choose a Transformer: Fourier or Galerkincited by 333
- 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 321
- 2021Finite Basis Physics-Informed Neural Networks (FBPINNs): a scalable domain decomposition approach for solving differential equationscited by 142
- 2023PINNacle: A Comprehensive Benchmark of Physics-Informed Neural Networks for Solving PDEscited by 84
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2025PDE-Transformer: Efficient and Versatile Transformers for Physics Simulations
- 2025FreqMoE: Dynamic Frequency Enhancement for Neural PDE Solvers
- 2025Merging Memory and Space: A State Space Neural Operator
- 2024Text2PDE: Latent Diffusion Models for Accessible Physics Simulation
- 2025Wave-PDE Nets: Trainable Wave-Equation Layers as an Alternative to Attention
- 2026PDE-SSM: A Spectral State Space Approach to Spatial Mixing in Diffusion Transformers
- 2026UniFluids: Unified Neural Operator Learning with Conditional Flow-matching
- 2023Real-time Inference and Extrapolation via a Diffusion-inspired Temporal Transformer Operator (DiTTO)