Approximate Bayesian Model Inversion for PDEs with Heterogeneous and State-Dependent Coefficients

Abstract

We present two approximate Bayesian inference methods for parameter estimation in partial differential equation (PDE) models with space-dependent and state-dependent parameters. We demonstrate that these methods provide accurate and cost-effective alternatives to Markov Chain Monte Carlo simulation. We assume a parameterized Gaussian prior on the unknown functions, and approximate the posterior density by a parameterized multivariate Gaussian density. The parameters of the prior and posterior are estimated from sparse observations of the PDE model's states and the unknown functions themselves by maximizing the evidence lower bound (ELBO), a lower bound on the log marginal likelihood of the observations. The first method, Laplace-EM, employs the expectation maximization algorithm to maximize the ELBO, with a Laplace approximation of the posterior on the E-step, and minimization of a Kullback-Leibler divergence on the M-step. The second method, DSVI-EB, employs the doubly stochastic variational inference (DSVI) algorithm, in which the ELBO is maximized via gradient-based stochastic optimization, with nosiy gradients computed via simple Monte Carlo sampling and Gaussian backpropagation. We apply these methods to identifying diffusion coefficients in linear and nonlinear diffusion equations, and we find that both methods provide accurate estimates of posterior densities and the hyperparameters of Gaussian priors. While the Laplace-EM method is more accurate, it requires computing Hessians of the physics model. The DSVI-EB method is found to be less accurate but only requires gradients of the physics model.

References in corpus (8)

• Variational Inference: A Review for Statisticians
• Stochastic Backpropagation and Approximate Inference in Deep Generative Models
• Expectation Propagation for approximate Bayesian inference
• Machine Learning of Linear Differential Equations using Gaussian Processes
• Proceedings of the Seventeenth Conference on Uncertainty in Artificial Intelligence (2001)
• Markov Chain Monte Carlo and Variational Inference: Bridging the Gap
• Bayesian and Variational Bayesian approaches for flows in heterogenous random media
• Improved Laplace Approximation for Marginal Likelihoods

Cited by in corpus (7)

• Physics-Informed Machine Learning Method for Large-Scale Data Assimilation Problems
• Variational Bayesian Approximation of Inverse Problems using Sparse Precision Matrices
• Variational Inference at Glacier Scale
• Multifidelity multilevel Monte Carlo to accelerate approximate Bayesian parameter inference for partially observed stochastic processes
• Learning and Meta-Learning of Stochastic Advection-Diffusion-Reaction Systems from Sparse Measurements
• Physics-Informed Machine Learning with Conditional Karhunen-Loève Expansions
• Modeling the impact of temperature and bird migration on the spread of West Nile virus