mlsanityData to 7 Oct 2026

When Attention Beats Fourier: Multi-Scale Transformers for PDE Solving on Irregular Domains

arXiv 2605.08318 ↗First posted 8 May 2026cs.LG cs.AI cs.NAcited by 0

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

  1. 2017Attention Is All You Needcited by 59,932
  2. 2023Mamba: Linear-Time Sequence Modeling with Selective State Spacescited by 4,616
  3. 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
  4. 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
  5. 2021Characterizing possible failure modes in physics-informed neural networkscited by 632
  6. 2019DeepXDE: A deep learning library for solving differential equationscited by 617
  7. 2021Physics-Informed Neural Operator for Learning Partial Differential Equationscited by 516
  8. 2021Choose a Transformer: Fourier or Galerkincited by 333
  9. 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 321
  10. 2021Finite Basis Physics-Informed Neural Networks (FBPINNs): a scalable domain decomposition approach for solving differential equationscited by 142
  11. 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

  1. 2025PDE-Transformer: Efficient and Versatile Transformers for Physics Simulations
  2. 2025FreqMoE: Dynamic Frequency Enhancement for Neural PDE Solvers
  3. 2025Merging Memory and Space: A State Space Neural Operator
  4. 2024Text2PDE: Latent Diffusion Models for Accessible Physics Simulation
  5. 2025Wave-PDE Nets: Trainable Wave-Equation Layers as an Alternative to Attention
  6. 2026PDE-SSM: A Spectral State Space Approach to Spatial Mixing in Diffusion Transformers
  7. 2026UniFluids: Unified Neural Operator Learning with Conditional Flow-matching
  8. 2023Real-time Inference and Extrapolation via a Diffusion-inspired Temporal Transformer Operator (DiTTO)