On the training and generalization of deep operator networks
Not on the map. Why?
Abstract (arXiv)
We present a novel training method for deep operator networks (DeepONets), one of the most popular neural network models for operators. DeepONets are constructed by two sub-networks, namely the branch and trunk networks. Typically, the two sub-networks are trained simultaneously, which amounts to solving a complex optimization problem in a high dimensional space. In addition, the nonconvex and nonlinear nature makes training very challenging. To tackle such a challenge, we propose a two-step training method that trains the trunk network first and then sequentially trains the branch network. The core mechanism is motivated by the divide-and-conquer paradigm and is the decomposition of the entire complex training task into two subtasks with reduced complexity. Therein the Gram-Schmidt orthonormalization process is introduced which significantly improves stability and generalization ability. On the theoretical side, we establish a generalization error estimate in terms of the number of training data, the width of DeepONets, and the number of input and output sensors. Numerical examples are presented to demonstrate the effectiveness of the two-step training method, including Darcy flow in heterogeneous porous media.
Key facts. Cited by 31 papers in the map (data through 7 Oct 2026). Builds on 22 of its 46 references that are in the map. First posted on arXiv 2 Sep 2023.
Builds on
22 of 46 references.
References · 22
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 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
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,965
- 2016An overview of gradient descent optimization algorithmscited by 1,021
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 784
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 722
- 2018Gradient Descent Provably Optimizes Over-parameterized Neural Networkscited by 655
- 2018Gradient Descent Finds Global Minima of Deep Neural Networkscited by 572
- 2021Factorized Fourier Neural Operatorscited by 237
- 2022FourCastNet: Accelerating Global High-Resolution Weather Forecasting using Adaptive Fourier Neural Operatorscited by 188
- 2022MIONet: Learning multiple-input operators via tensor productcited by 180
10 more
- 2019Towards moderate overparameterization: global convergence guarantees for training shallow neural networkscited by 176
- 2022Learning Operators with Coupled Attentioncited by 88
- 2021Improved architectures and training algorithms for deep operator networkscited by 85
- 2022Deep transfer operator learning for partial differential equations under conditional shiftcited by 83
- 2022Learning two-phase microstructure evolution using neural operators and autoencoder architecturescited by 72
- 2021Exponential Convergence of Deep Operator Networks for Elliptic Partial Differential Equationscited by 44
- 2021On the Optimal Memorization Power of ReLU Neural Networkscited by 40
- 2022SVD Perspectives for Augmenting DeepONet Flexibility and Interpretabilitycited by 35
- 2020Provable Memorization via Deep Neural Networks using Sub-linear Parameterscited by 34
- 2020Plateau Phenomenon in Gradient Descent Training of ReLU networks: Explanation, Quantification and Avoidancecited by 11
Built on by
Citing papers per year: 2023: 1; 2024: 10; 2025: 12; 2026: 8.
31 papers.
Sort:
By area:
- 2024Transformers as Neural Operators for Solutions of Differential Equations with Finite Regularitycited by 33
- 2024RiemannONets: Interpretable Neural Operators for Riemann Problemscited by 28
- 2023Rethinking materials simulations: Blending direct numerical simulations with neural operatorscited by 25
- 2024Basis-to-Basis Operator Learning Using Function Encoderscited by 17
- 2024Neural Scaling Laws of Deep ReLU and Deep Operator Network: A Theoretical Studycited by 13
- 2024Predicting Crack Nucleation and Propagation in Brittle Materials Using Deep Operator Networks with Diverse Trunk Architecturescited by 12
- 2024Ensemble and Mixture-of-Experts DeepONets For Operator Learningcited by 8
- 2025Time Marching Neural Operator FE Coupling: AI Accelerated Physics Modelingcited by 8
- 2025ATHENA: Agentic Team for Hierarchical Evolutionary Numerical Algorithmscited by 7
- 2024Efficient Training of Deep Neural Operator Networks via Randomized Samplingcited by 6
- 2025A Variational Framework for Residual-Based Adaptivity in Neural PDE Solvers and Operator Learningcited by 6
- 2024Hamilton-Jacobi Based Policy-Iteration via Deep Operator Learningcited by 5
19 more
- 2025BubbleOKAN: A Physics-Informed Interpretable Neural Operator for High-Frequency Bubble Dynamicscited by 5
- 2026Diagnosing Failure Modes of Neural Operators Across Diverse PDE Familiescited by 5
- 2026FEKAN: Feature-Enriched Kolmogorov-Arnold Networkscited by 5
- 2025Stochastic Operator Network: A Stochastic Maximum Principle Based Approach to Operator Learningcited by 3
- 2025Single vs. Multiple Branches in DeepONet and S-DeepONet: Network Architecture Follows Coupling in Multiphysics Systemscited by 2
- 2025AMORE: Adaptive Multi-Output Operator Network for Stiff Chemical Kineticscited by 2
- 2025Probabilistic Predictions of Process-Induced Deformation in Carbon/Epoxy Composites Using a Deep Operator Networkcited by 2
- 2025Invertible Koopman neural operator for data-driven modeling of partial differential equationscited by 1
- 2025Leveraging Deep Operator Networks (DeepONet) for Acoustic Full Waveform Inversion (FWI)cited by 1
- 2025Deep operator network for surrogate modeling of poroelasticity with random permeability fieldscited by 1
- 2026Physics-Informed Laplace Neural Operator for Solving Partial Differential Equationscited by 1
- 2026Spectrally Safe Neural Operator Warm-Starts for Large-Scale Newton Solverscited by 1
- 2024A DeepONet for inverting the Neumann-to-Dirichlet Operator in Electrical Impedance Tomography: An approximation theoretic perspective and numerical resultscited by 0
- 2024Neural Hamilton: Can A.I. Understand Hamiltonian Mechanics?cited by 0
- 2025Crack Path Prediction with Operator Learning using Discrete Particle System data Generationcited by 0
- 2026The Error of Deep Operator Networks Is the Sum of Its Parts: Branch-Trunk and Mode Error Decompositionscited by 0
- 2026Fourier Neural Operators with Least-Squares Readout Refit for Learning Random Obstacle-to-Solution Mapscited by 0
- 2026Two-Step MV-DeepONet: Probabilistic Operator Learning for Uncertainty Propagation Driven by Random Input Fieldscited by 0
- 2026Sequential operator learning under dependent datacited by 0
- 2026Sequential operator learning under dependent data25 Aug 2026
- 2026Two-Step MV-DeepONet: Probabilistic Operator Learning for Uncertainty Propagation Driven by Random Input Fields10 Aug 2026
- 2026Fourier Neural Operators with Least-Squares Readout Refit for Learning Random Obstacle-to-Solution Maps28 Jun 2026
- 2026Spectrally Safe Neural Operator Warm-Starts for Large-Scale Newton Solvers20 Jun 2026
- 2026The Error of Deep Operator Networks Is the Sum of Its Parts: Branch-Trunk and Mode Error Decompositions25 Feb 2026
- 2026FEKAN: Feature-Enriched Kolmogorov-Arnold Networks18 Feb 2026
- 2026Physics-Informed Laplace Neural Operator for Solving Partial Differential Equations13 Feb 2026
- 2026Diagnosing Failure Modes of Neural Operators Across Diverse PDE Families16 Jan 2026
- 2025Probabilistic Predictions of Process-Induced Deformation in Carbon/Epoxy Composites Using a Deep Operator Network15 Dec 2025
- 2025ATHENA: Agentic Team for Hierarchical Evolutionary Numerical Algorithms3 Dec 2025
- 2025AMORE: Adaptive Multi-Output Operator Network for Stiff Chemical Kinetics15 Oct 2025
- 2025A Variational Framework for Residual-Based Adaptivity in Neural PDE Solvers and Operator Learning17 Sep 2025
19 more
- 2025Deep operator network for surrogate modeling of poroelasticity with random permeability fields15 Sep 2025
- 2025BubbleOKAN: A Physics-Informed Interpretable Neural Operator for High-Frequency Bubble Dynamics5 Aug 2025
- 2025Stochastic Operator Network: A Stochastic Maximum Principle Based Approach to Operator Learning10 Jul 2025
- 2025Single vs. Multiple Branches in DeepONet and S-DeepONet: Network Architecture Follows Coupling in Multiphysics Systems4 Jul 2025
- 2025Crack Path Prediction with Operator Learning using Discrete Particle System data Generation15 May 2025
- 2025Time Marching Neural Operator FE Coupling: AI Accelerated Physics Modeling15 Apr 2025
- 2025Leveraging Deep Operator Networks (DeepONet) for Acoustic Full Waveform Inversion (FWI)14 Apr 2025
- 2025Invertible Koopman neural operator for data-driven modeling of partial differential equations25 Mar 2025
- 2024Predicting Crack Nucleation and Propagation in Brittle Materials Using Deep Operator Networks with Diverse Trunk Architectures15 Dec 2024
- 2024Neural Hamilton: Can A.I. Understand Hamiltonian Mechanics?28 Oct 2024
- 2024Neural Scaling Laws of Deep ReLU and Deep Operator Network: A Theoretical Study1 Oct 2024
- 2024Basis-to-Basis Operator Learning Using Function Encoders30 Sep 2024
- 2024Efficient Training of Deep Neural Operator Networks via Randomized Sampling20 Sep 2024
- 2024A DeepONet for inverting the Neumann-to-Dirichlet Operator in Electrical Impedance Tomography: An approximation theoretic perspective and numerical results24 Jul 2024
- 2024Hamilton-Jacobi Based Policy-Iteration via Deep Operator Learning16 Jun 2024
- 2024Transformers as Neural Operators for Solutions of Differential Equations with Finite Regularity29 May 2024
- 2024Ensemble and Mixture-of-Experts DeepONets For Operator Learning20 May 2024
- 2024RiemannONets: Interpretable Neural Operators for Riemann Problems16 Jan 2024
- 2023Rethinking materials simulations: Blending direct numerical simulations with neural operators8 Dec 2023
- 2024Transformers as Neural Operators for Solutions of Differential Equations with Finite Regularitycited by 33
- 2024RiemannONets: Interpretable Neural Operators for Riemann Problemscited by 28
- 2023Rethinking materials simulations: Blending direct numerical simulations with neural operatorscited by 25
- 2024Basis-to-Basis Operator Learning Using Function Encoderscited by 17
- 2024Neural Scaling Laws of Deep ReLU and Deep Operator Network: A Theoretical Studycited by 13
- 2024Predicting Crack Nucleation and Propagation in Brittle Materials Using Deep Operator Networks with Diverse Trunk Architecturescited by 12
- 2025Time Marching Neural Operator FE Coupling: AI Accelerated Physics Modelingcited by 8
- 2024Efficient Training of Deep Neural Operator Networks via Randomized Samplingcited by 6
- 2025A Variational Framework for Residual-Based Adaptivity in Neural PDE Solvers and Operator Learningcited by 6
- 2024Hamilton-Jacobi Based Policy-Iteration via Deep Operator Learningcited by 5
- 2025BubbleOKAN: A Physics-Informed Interpretable Neural Operator for High-Frequency Bubble Dynamicscited by 5
- 2026Diagnosing Failure Modes of Neural Operators Across Diverse PDE Familiescited by 5
12 most cited of 29 All 29 (JSON)
- 2026Sequential operator learning under dependent data25 Aug 2026
- 2026Two-Step MV-DeepONet: Probabilistic Operator Learning for Uncertainty Propagation Driven by Random Input Fields10 Aug 2026
- 2026Fourier Neural Operators with Least-Squares Readout Refit for Learning Random Obstacle-to-Solution Maps28 Jun 2026
- 2026Spectrally Safe Neural Operator Warm-Starts for Large-Scale Newton Solvers20 Jun 2026
- 2026The Error of Deep Operator Networks Is the Sum of Its Parts: Branch-Trunk and Mode Error Decompositions25 Feb 2026
- 2026FEKAN: Feature-Enriched Kolmogorov-Arnold Networks18 Feb 2026
- 2026Physics-Informed Laplace Neural Operator for Solving Partial Differential Equations13 Feb 2026
- 2026Diagnosing Failure Modes of Neural Operators Across Diverse PDE Families16 Jan 2026
- 2025Probabilistic Predictions of Process-Induced Deformation in Carbon/Epoxy Composites Using a Deep Operator Network15 Dec 2025
- 2025AMORE: Adaptive Multi-Output Operator Network for Stiff Chemical Kinetics15 Oct 2025
- 2025A Variational Framework for Residual-Based Adaptivity in Neural PDE Solvers and Operator Learning17 Sep 2025
- 2025Deep operator network for surrogate modeling of poroelasticity with random permeability fields15 Sep 2025
12 newest of 29 All 29 (JSON)