vix.ing · top · new · best · stats

Multi-fidelity Bayesian Neural Networks: Algorithms and Applications

2020/12/19 by Xuhui Meng, Hessam Babaee, George Em Karniadakis · 188 citations
Computer Science · Engineering · Mathematics · Physics and Astronomy · #Algorithm #Artificial intelligence #Artificial neural network #Bayesian probability #Computer science #Fidelity #Gaussian Processes and Bayesian Inference #Hyperparameter #Inverse problem #Machine learning #Mathematics #Model Reduction and Neural Networks #Nonlinear system #Physics #Reservoir Engineering and Simulation Methods #cs.LG #physics.comp-ph

paper · pdf · doi:10.1016/j.jcp.2021.110361

published in Journal of Computational Physics 438, 110361 (Elsevier BV) · 31 pages, 11 figures

arxiv created 2020/12/19 · openalex publication_date 2021/04/16 · arxiv updated 2021/06/02 · openalex created_date 2025/10/10 · openalex updated_date 2026/08/05

Abstract

We propose a new class of Bayesian neural networks (BNNs) that can be trained using noisy data of variable fidelity, and we apply them to learn function approximations as well as to solve inverse problems based on partial differential equations (PDEs). These multi-fidelity BNNs consist of three neural networks: The first is a fully connected neural network, which is trained following the maximum a posteriori probability (MAP) method to fit the low-fidelity data; the second is a Bayesian neural network employed to capture the cross-correlation with uncertainty quantification between the low- and high-fidelity data; and the last one is the physics-informed neural network, which encodes the physical laws described by PDEs. For the training of the last two neural networks, we use the Hamiltonian Monte Carlo method to estimate accurately the posterior distributions for the corresponding hyperparameters. We demonstrate the accuracy of the present method using synthetic data as well as real measurements. Specifically, we first approximate a one- and four-dimensional function, and then infer the reaction rates in one- and two-dimensional diffusion-reaction systems. Moreover, we infer the sea surface temperature (SST) in the Massachusetts and Cape Cod Bays using satellite images and in-situ measurements. Taken together, our results demonstrate that the present method can capture both linear and nonlinear correlation between the low- and high-fideilty data adaptively, identify unknown parameters in PDEs, and quantify uncertainties in predictions, given a few scattered noisy high-fidelity data. Finally, we demonstrate that we can effectively and efficiently reduce the uncertainties and hence enhance the prediction accuracy with an active learning approach, using as examples a specific one-dimensional function approximation and an inverse PDE problem.

Citations

Cited by