Approximate Bayesian Neural Operators: Uncertainty Quantification for Parametric PDEs
Not on the map. Why?
Abstract (arXiv)
Neural operators are a type of deep architecture that learns to solve (i.e. learns the nonlinear solution operator of) partial differential equations (PDEs). The current state of the art for these models does not provide explicit uncertainty quantification. This is arguably even more of a problem for this kind of tasks than elsewhere in machine learning, because the dynamical systems typically described by PDEs often exhibit subtle, multiscale structure that makes errors hard to spot by humans. In this work, we first provide a mathematically detailed Bayesian formulation of the ''shallow'' (linear) version of neural operators in the formalism of Gaussian processes. We then extend this analytic treatment to general deep neural operators using approximate methods from Bayesian deep learning. We extend previous results on neural operators by providing them with uncertainty quantification. As a result, our approach is able to identify cases, and provide structured uncertainty estimates, where the neural operator fails to predict well.
Key facts. Cited by 25 papers in the map (data through 14 Oct 2026). Builds on 37 of its 73 references that are in the map. First posted on arXiv 2 Aug 2022.
Builds on
37 of 73 references.
References · 37
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2017Neural Message Passing for Quantum Chemistrycited by 3,487
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,385
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 2,018
- 2022FourCastNet: A Global Data-driven High-resolution Weather Model using Adaptive Fourier Neural Operatorscited by 776
- 2020Neural Operator: Graph Kernel Network for Partial Differential Equationscited by 734
- 2017DGM: A deep learning algorithm for solving partial differential equationscited by 678
- 2021Characterizing possible failure modes in physics-informed neural networkscited by 643
- 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 560
25 more
- 2021Physics-Informed Neural Operator for Learning Partial Differential Equationscited by 527
- 2021Neural Operator: Learning Maps Between Function Spacescited by 414
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 365
- 2020Multipole Graph Neural Operator for Parametric Partial Differential Equationscited by 349
- 2018Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problemcited by 323
- 2015Scalable Bayesian Optimization Using Deep Neural Networkscited by 284
- 2019Physics-Constrained Deep Learning for High-dimensional Surrogate Modeling and Uncertainty Quantification without Labeled Datacited by 262
- 2020Model Reduction and Neural Networks for Parametric PDEscited by 258
- 2021Adaptive Fourier Neural Operators: Efficient Token Mixers for Transformerscited by 248
- 2021Multiwavelet-based Operator Learning for Differential Equationscited by 234
- 2020Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU Networkscited by 231
- 2022U-NO: U-shaped Neural Operatorscited by 230
- 2019Cyclical Stochastic Gradient MCMC for Bayesian Deep Learningcited by 203
- 2018Fast and Scalable Bayesian Deep Learning by Weight-Perturbation in Adamcited by 131
- 2021Solving and Learning Nonlinear PDEs with Gaussian Processescited by 129
- 2021Scalable Marginal Likelihood Estimation for Model Selection in Deep Learningcited by 106
- 2017Noisy Natural Gradient as Variational Inferencecited by 94
- 2020The Random Feature Model for Input-Output Maps between Banach Spacescited by 90
- 2019Approximate Inference Turns Deep Networks into Gaussian Processescited by 89
- 2020A physics-informed operator regression framework for extracting data-driven continuum modelscited by 60
- 2022Generative Adversarial Neural Operatorscited by 52
- 2021Learning elliptic partial differential equations with randomized linear algebracited by 29
- 2022Variational Bayes Deep Operator Network: A data-driven Bayesian solver for parametric differential equationscited by 11
- 2020Improving predictions of Bayesian neural nets via local linearizationcited by 7
- 2022Laplace HypoPINN: Physics-Informed Neural Network for hypocenter localization and its predictive uncertaintycited by 3
Built on by
Citing papers per year: 2024: 5; 2025: 9; 2026: 11.
25 papers.
Sort:
By area:
- 2024Calibrated Uncertainty Quantification for Operator Learning via Conformal Predictioncited by 26
- 2025Operator Learning: A Statistical Perspectivecited by 17
- 2024Using Uncertainty Quantification to Characterize and Improve Out-of-Domain Learning for PDEscited by 15
- 2024Linearization Turns Neural Operators into Function-Valued Gaussian Processescited by 12
- 2025Epistemic Artificial Intelligence is Essential for Machine Learning Models to Truly 'Know When They Do Not Know'cited by 8
- 2025Light-Weight Diffusion Multiplier and Uncertainty Quantification for Fourier Neural Operatorscited by 6
- 2024Learning Semilinear Neural Operators : A Unified Recursive Framework For Prediction And Data Assimilationcited by 4
- 2024Neural Operator induced Gaussian Process framework for probabilistic solution of parametric partial differential equationscited by 4
- 2025Locally Adaptive Conformal Inference for Operator Modelscited by 2
- 2026Structure-Aware Epistemic Uncertainty Quantification for Neural Operator PDE Surrogatescited by 2
- 2025ON-Traffic: An Operator Learning Framework for Online Traffic Flow Estimation and Uncertainty Quantification from Lagrangian Sensorscited by 1
- 2025Neural Measures for learning distributions of Random PDEscited by 1
13 more
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospectscited by 1
- 2026Operator learning for the 2D incompressible Navier-Stokes equations: a conformal prediction approach in the data-scarce regimecited by 1
- 2026Geometry-Aware Post-Hoc Uncertainty Quantification in Operator Learningcited by 1
- 2026Neural Operator Processes for Probabilistic Operator Learning under Partial Observationscited by 1
- 2026GeoQ: Geometry-Aware Conditional Quantile Error Estimation for Scientific Surrogate Modelscited by 1
- 2025Are Statistical Methods Obsolete in the Era of Deep Learning? A Study of ODE Inverse Problemscited by 0
- 2025From Local Interactions to Global Operators: Scalable Gaussian Process Operator for Physical Systemscited by 0
- 2025Infinite Neural Operators: Gaussian processes on functionscited by 0
- 2026Generative Neural Operators through Diffusion Last Layercited by 0
- 2026Posterior-First Neural PDE Simulation: Inferring Hidden Problem State from a Single Fieldcited by 0
- 2026Martingale Neural Operators: Learning Stochastic Marginals via Doob-Meyer Factorizationcited by 0
- 2026Conformal Uncertainty Quantification Guarantees for Neural Operatorscited by 0
- 2026Threshold-Aware Conformal Routingcited by 0
- 2026Threshold-Aware Conformal Routing1 Oct 2026
- 2026Conformal Uncertainty Quantification Guarantees for Neural Operators28 Aug 2026
- 2026GeoQ: Geometry-Aware Conditional Quantile Error Estimation for Scientific Surrogate Models21 Aug 2026
- 2026Neural Operator Processes for Probabilistic Operator Learning under Partial Observations22 Jun 2026
- 2026Geometry-Aware Post-Hoc Uncertainty Quantification in Operator Learning16 Jun 2026
- 2026Operator learning for the 2D incompressible Navier-Stokes equations: a conformal prediction approach in the data-scarce regime7 Jun 2026
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospects16 May 2026
- 2026Martingale Neural Operators: Learning Stochastic Marginals via Doob-Meyer Factorization15 May 2026
- 2026Posterior-First Neural PDE Simulation: Inferring Hidden Problem State from a Single Field5 May 2026
- 2026Structure-Aware Epistemic Uncertainty Quantification for Neural Operator PDE Surrogates24 Feb 2026
- 2026Generative Neural Operators through Diffusion Last Layer4 Feb 2026
- 2025Infinite Neural Operators: Gaussian processes on functions19 Oct 2025
13 more
- 2025Light-Weight Diffusion Multiplier and Uncertainty Quantification for Fourier Neural Operators1 Aug 2025
- 2025Locally Adaptive Conformal Inference for Operator Models28 Jul 2025
- 2025Neural Measures for learning distributions of Random PDEs2 Jul 2025
- 2025From Local Interactions to Global Operators: Scalable Gaussian Process Operator for Physical Systems18 Jun 2025
- 2025Are Statistical Methods Obsolete in the Era of Deep Learning? A Study of ODE Inverse Problems27 May 2025
- 2025Epistemic Artificial Intelligence is Essential for Machine Learning Models to Truly 'Know When They Do Not Know'8 May 2025
- 2025Operator Learning: A Statistical Perspective4 Apr 2025
- 2025ON-Traffic: An Operator Learning Framework for Online Traffic Flow Estimation and Uncertainty Quantification from Lagrangian Sensors18 Mar 2025
- 2024Linearization Turns Neural Operators into Function-Valued Gaussian Processes7 Jun 2024
- 2024Neural Operator induced Gaussian Process framework for probabilistic solution of parametric partial differential equations24 Apr 2024
- 2024Using Uncertainty Quantification to Characterize and Improve Out-of-Domain Learning for PDEs15 Mar 2024
- 2024Learning Semilinear Neural Operators : A Unified Recursive Framework For Prediction And Data Assimilation24 Feb 2024
- 2024Calibrated Uncertainty Quantification for Operator Learning via Conformal Prediction2 Feb 2024
- 2026Generative Neural Operators through Diffusion Last Layercited by 0
- 2026Generative Neural Operators through Diffusion Last Layer4 Feb 2026
- 2024Calibrated Uncertainty Quantification for Operator Learning via Conformal Predictioncited by 26
- 2025Operator Learning: A Statistical Perspectivecited by 17
- 2024Using Uncertainty Quantification to Characterize and Improve Out-of-Domain Learning for PDEscited by 15
- 2024Linearization Turns Neural Operators into Function-Valued Gaussian Processescited by 12
- 2025Epistemic Artificial Intelligence is Essential for Machine Learning Models to Truly 'Know When They Do Not Know'cited by 8
- 2025Light-Weight Diffusion Multiplier and Uncertainty Quantification for Fourier Neural Operatorscited by 6
- 2024Learning Semilinear Neural Operators : A Unified Recursive Framework For Prediction And Data Assimilationcited by 4
- 2024Neural Operator induced Gaussian Process framework for probabilistic solution of parametric partial differential equationscited by 4
- 2025Locally Adaptive Conformal Inference for Operator Modelscited by 2
- 2026Structure-Aware Epistemic Uncertainty Quantification for Neural Operator PDE Surrogatescited by 2
- 2025ON-Traffic: An Operator Learning Framework for Online Traffic Flow Estimation and Uncertainty Quantification from Lagrangian Sensorscited by 1
- 2025Neural Measures for learning distributions of Random PDEscited by 1
12 most cited of 24 All 24 (JSON)
- 2026Threshold-Aware Conformal Routing1 Oct 2026
- 2026Conformal Uncertainty Quantification Guarantees for Neural Operators28 Aug 2026
- 2026GeoQ: Geometry-Aware Conditional Quantile Error Estimation for Scientific Surrogate Models21 Aug 2026
- 2026Neural Operator Processes for Probabilistic Operator Learning under Partial Observations22 Jun 2026
- 2026Geometry-Aware Post-Hoc Uncertainty Quantification in Operator Learning16 Jun 2026
- 2026Operator learning for the 2D incompressible Navier-Stokes equations: a conformal prediction approach in the data-scarce regime7 Jun 2026
- 2026Harnessing AI for Inverse Partial Differential Equation Problems: Past, Present, and Prospects16 May 2026
- 2026Martingale Neural Operators: Learning Stochastic Marginals via Doob-Meyer Factorization15 May 2026
- 2026Posterior-First Neural PDE Simulation: Inferring Hidden Problem State from a Single Field5 May 2026
- 2026Structure-Aware Epistemic Uncertainty Quantification for Neural Operator PDE Surrogates24 Feb 2026
- 2025Infinite Neural Operators: Gaussian processes on functions19 Oct 2025
- 2025Light-Weight Diffusion Multiplier and Uncertainty Quantification for Fourier Neural Operators1 Aug 2025
12 newest of 24 All 24 (JSON)