Understanding the Expressivity and Trainability of Fourier Neural Operator: A Mean-Field Perspective
Not on the map. Why?
Abstract (arXiv)
In this paper, we explores the expressivity and trainability of the Fourier Neural Operator (FNO). We establish a mean-field theory for the FNO, analyzing the behavior of the random FNO from an edge of chaos perspective. Our investigation into the expressivity of a random FNO involves examining the ordered-chaos phase transition of the network based on the weight distribution. This phase transition demonstrates characteristics unique to the FNO, induced by mode truncation, while also showcasing similarities to those of densely connected networks. Furthermore, we identify a connection between expressivity and trainability: the ordered and chaotic phases correspond to regions of vanishing and exploding gradients, respectively. This finding provides a practical prerequisite for the stable training of the FNO. Our experimental results corroborate our theoretical findings.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 29 of its 37 references that are in the map. First posted on arXiv 10 Oct 2023.
Builds on
29 of 37 references.
References · 29
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,806
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2022FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operatorscited by 748
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 722
- 2021Physics-Informed Neural Operator for Learning Partial Differential Equationscited by 516
- 2022Fourier Neural Operator with Learned Deformations for PDEs on General Geometriescited by 416
- 2022PDEBENCH: An Extensive Benchmark for Scientific Machine Learningcited by 396
- 2020Multipole Graph Neural Operator for Parametric Partial Differential Equationscited by 343
- 2021U-FNO -- An enhanced Fourier neural operator-based deep-learning model for multiphase flowcited by 292
- 2022Transformer for Partial Differential Equations' Operator Learningcited by 270
- 2016Exponential expressivity in deep neural networks through transient chaoscited by 260
17 more
- 2021Adaptive Fourier Neural Operators: Efficient Token Mixers for Transformerscited by 237
- 2021Factorized Fourier Neural Operatorscited by 237
- 2022U-NO: U-shaped Neural Operatorscited by 219
- 2016Deep Information Propagationcited by 194
- 2018Dynamical Isometry and a Mean Field Theory of CNNs: How to Train 10,000-Layer Vanilla Convolutional Neural Networkscited by 176
- 2017Resurrecting the sigmoid in deep learning through dynamical isometry: theory and practicecited by 132
- 2022Clifford Neural Layers for PDE Modelingcited by 91
- 2017Mean Field Residual Networks: On the Edge of Chaoscited by 83
- 2022Wavelet neural operator: a neural operator for parametric partial differential equationscited by 77
- 2022Lie Point Symmetry Data Augmentation for Neural PDE Solverscited by 76
- 2018The Emergence of Spectral Universality in Deep Networkscited by 74
- 2021Seismic wave propagation and inversion with Neural Operatorscited by 51
- 2023Group Equivariant Fourier Neural Operators for Partial Differential Equationscited by 42
- 2021Solving PDE-constrained Control Problems Using Operator Learningcited by 41
- 2022Transform Once: Efficient Operator Learning in Frequency Domaincited by 20
- 2018On the Selection of Initialization and Activation Function for Deep Neural Networkscited by 11
- 2019Multiphase flow prediction with deep neural networkscited by 1
Built on by
Citing papers per year: 2024: 2; 2025: 4; 2026: 4.
10 papers.
- 2025KANO: Kolmogorov-Arnold Neural Operatorcited by 6
- 2026Diagnosing Failure Modes of Neural Operators Across Diverse PDE Familiescited by 5
- 2024Spectral Convolutional Conditional Neural Processescited by 4
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamicscited by 2
- 2026PDEInvBench: A Comprehensive Dataset and Design Space Exploration of Neural Networks for PDE Inverse Problemscited by 2
- 2025Maximal Update Parametrization and Zero-Shot Hyperparameter Transfer for Fourier Neural Operatorscited by 1
- 2026Multi-Scale Wavelet Transformers for Operator Learning of Dynamical Systemscited by 1
- 2025Analysis of Fourier Neural Operators via Effective Field Theorycited by 0
- 2025PGOT: A Physics-Geometry Operator Transformer for Complex PDEscited by 0
- 2026A neural operator view on U-Nets for inverse imaging problemscited by 0
- 2026A neural operator view on U-Nets for inverse imaging problems6 Aug 2026
- 2026PDEInvBench: A Comprehensive Dataset and Design Space Exploration of Neural Networks for PDE Inverse Problems25 May 2026
- 2026Multi-Scale Wavelet Transformers for Operator Learning of Dynamical Systems1 Feb 2026
- 2026Diagnosing Failure Modes of Neural Operators Across Diverse PDE Families16 Jan 2026
- 2025PGOT: A Physics-Geometry Operator Transformer for Complex PDEs29 Dec 2025
- 2025KANO: Kolmogorov-Arnold Neural Operator20 Sep 2025
- 2025Analysis of Fourier Neural Operators via Effective Field Theory29 Jul 2025
- 2025Maximal Update Parametrization and Zero-Shot Hyperparameter Transfer for Fourier Neural Operators24 Jun 2025
- 2024Graph Fourier Neural ODEs: Modeling Spatial-temporal Multi-scales in Molecular Dynamics3 Nov 2024
- 2024Spectral Convolutional Conditional Neural Processes19 Apr 2024