Gradient-based stochastic optimization methods in Bayesian experimental design

TL;DR

This paper develops and compares gradient-based stochastic optimization methods for Bayesian experimental design, focusing on nonlinear models and using Monte Carlo estimators of expected information gain.

Contribution

It introduces gradient-based stochastic optimization techniques for experimental design using Monte Carlo estimators and polynomial chaos acceleration, with empirical performance analysis.

Findings

Gradient-based methods effectively optimize experimental design.

Polynomial chaos accelerates objective and gradient evaluations.

Tradeoffs between computational cost and solution robustness are characterized.

Abstract

Optimal experimental design (OED) seeks experiments expected to yield the most useful data for some purpose. In practical circumstances where experiments are time-consuming or resource-intensive, OED can yield enormous savings. We pursue OED for nonlinear systems from a Bayesian perspective, with the goal of choosing experiments that are optimal for parameter inference. Our objective in this context is the expected information gain in model parameters, which in general can only be estimated using Monte Carlo methods. Maximizing this objective thus becomes a stochastic optimization problem. This paper develops gradient-based stochastic optimization methods for the design of experiments on a continuous parameter space. Given a Monte Carlo estimator of expected information gain, we use infinitesimal perturbation analysis to derive gradients of this estimator. We are then able to formulate two gradient-based stochastic optimization approaches: (i) Robbins-Monro stochastic approximation, and (ii) sample average approximation combined with a deterministic quasi-Newton method. A polynomial chaos approximation of the forward model accelerates objective and gradient evaluations in both cases. We discuss the implementation of these optimization methods, then conduct an empirical comparison of their performance. To demonstrate design in a nonlinear setting with partial differential equation forward models, we use the problem of sensor placement for source inversion. Numerical results yield useful guidelines on the choice of algorithm and sample sizes, assess the impact of estimator bias, and quantify tradeoffs of computational cost versus solution quality and robustness.