Neural Scaling Laws of Transformer Operator Network
Not on the map. Why?
Abstract (arXiv)
Transformers have emerged as powerful architectures for learning solution operators of physical systems. Empirically the prediction error has been observed to decrease when the data size and model size increase, suggesting neural scaling behavior. Yet a theoretical understanding of such scaling laws for transformer-based operator learning remains limited. In this work, we develop a theoretical framework for characterizing the approximation and generalization errors of transformer-based operator learning. Our analysis builds on a local-to-global approximation principle that is naturally aligned with the softmax attention mechanism and yields discretization-invariant output functions. On approximation theory, we derive a universal approximation error of transformer-based operator learning for H"older-regular operators. On generalization theory, we establish a power scaling law between the prediction error and the training data size. The rate of convergence represented by the scaling exponent explicitly reflects the dimensions of the input and output domains, the regularity of the underlying functions and operators, and crucially, the intrinsic dimension of the input function class. By exploiting this intrinsic low-dimensional structure, our analysis yields a power-law generalization rate for operator learning, in contrast to the logarithmic-type power-law rates appearing in existing analyses of operator learning with feedforward neural networks. Numerical experiments validate the predicted power-law scaling and confirm that the convergence rate varies systematically with the intrinsic dimension of the input function class.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 28 of its 35 references that are in the map. First posted on arXiv 27 Sep 2026.
Builds on
28 of 35 references.
References · 28
- 2017Attention Is All You Needcited by 59,932
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 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
- 2022FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operatorscited by 748
- 2021Choose a Transformer: Fourier or Galerkincited by 333
- 2023GNOT: A General Neural Operator Transformer for Operator Learningcited by 321
- 2024Transolver: A Fast Transformer Solver for PDEs on General Geometriescited by 299
- 2023Geometry-Informed Neural Operator for Large-Scale 3D PDEscited by 293
- 2022Transformer for Partial Differential Equations' Operator Learningcited by 270
- 2021Inductive Biases and Variable Creation in Self-Attention Mechanismscited by 144
- 2024Operator Learning: Algorithms and Analysiscited by 84
16 more
- 2022Deep Nonparametric Estimation of Operators between Infinite Dimensional Spacescited by 45
- 2024Continuum Attention for Neural Operatorscited by 39
- 2024Transformers as Neural Operators for Solutions of Differential Equations with Finite Regularitycited by 33
- 2024Understanding Scaling Laws with Statistical and Approximation Theory for Transformer Neural Networks on Intrinsically Low-dimensional Datacited by 30
- 2023The Parametric Complexity of Operator Learningcited by 27
- 2024HAMLET: Graph Transformer Neural Operator for Partial Differential Equationscited by 24
- 2024Positional Knowledge is All You Need: Position-induced Transformer (PiT) for Operator Learningcited by 17
- 2024Neural Scaling Laws of Deep ReLU and Deep Operator Network: A Theoretical Studycited by 13
- 2024Generalization Error Guaranteed Auto-Encoder-Based Nonlinear Model Reduction for Operator Learningcited by 12
- 2024Hierarchical Neural Operator Transformer with Learnable Frequency-aware Loss Prior for Arbitrary-scale Super-resolutioncited by 11
- 2024Nonlinear functional regression by functional deep neural network with kernel embeddingcited by 7
- 2024Universal Approximation of Operators with Transformers and Neural Integral Operatorscited by 2
- 2026Function graph transformers universally approximate operators between function spacescited by 2
- 2026Learning Theory of Transformers: Local-to-Global Approximation via Softmax Partition of Unitycited by 1
- 2026Generalization Guarantees for Multi-Input Neural Operator Learning in Sobolev Spacescited by 1
- 2026Transformers as Cross-Task Learners: Shared Structure Drives Sample Efficiency in In-Context Learningcited by 1
Built on by
Not cited yet.