Bayesian experimental design

Part of a series on
Bayesian statistics
Posterior = Likelihood × Prior ÷ Evidence
Background
Bayesian inference Bayesian probability Bayes' theorem Bernstein–von Mises theorem Coherence Cox's theorem Cromwell's rule Principle of indifference Principle of maximum entropy
Model building
Strong prior Conjugate prior Linear regression Empirical Bayes Hierarchical model
Posterior approximation
Markov chain Monte Carlo Laplace's approximation Integrated nested Laplace approximations Variational inference Approximate Bayesian computation
Estimators
Bayesian estimator Credible interval Maximum a posteriori estimation
Evidence approximation
Evidence lower bound Nested sampling
Model evaluation
Bayes factor Model averaging Posterior predictive
Mathematics portal
v t e

Bayesian experimental design provides a general probability-theoretical framework from which other theories on experimental design can be derived. It is based on Bayesian inference to interpret the observations/data acquired during the experiment. This allows accounting for both any prior knowledge on the parameters to be determined as well as uncertainties in observations.

The theory of Bayesian experimental design is to a certain extent based on the theory for making

) but may also involve factors such as the financial cost of performing the experiment. What will be the optimal experiment design depends on the particular utility criterion chosen.

Relations to more specialized optimal design theory

Linear theory

If the model is linear, the

optimal experimental design theory

.

Approximate normality

In numerous publications on Bayesian experimental design, it is (often implicitly) assumed that all posterior probabilities will be approximately normal. This allows for the expected utility to be calculated using linear theory, averaging over the space of model parameters.^[1] Caution must however be taken when applying this method, since approximate normality of all possible posteriors is difficult to verify, even in cases of normal observational errors and uniform prior probability.

Posterior distribution

In many cases, the posterior distribution is not available in closed form and has to be approximated using numerical methods. The most common approach is to use Markov chain Monte Carlo methods to generate samples from the posterior, which can then be used to approximate the expected utility.

Another approach is to use a

approximation of the posterior, which can often be calculated in closed form. This approach has the advantage of being computationally more efficient than Monte Carlo methods, but the disadvantage that the approximation might not be very accurate.

Some authors proposed approaches that use the posterior predictive distribution to assess the effect of new measurements on prediction uncertainty,^[2][3] while others suggest maximizing the mutual information between parameters, predictions and potential new experiments.^[4]

Mathematical formulation

Notation ${\displaystyle \theta \,}$ parameters to be determined ${\displaystyle y\,}$ observation or data ${\displaystyle \xi \,}$ design ${\displaystyle p(y\mid \theta ,\xi )\,}$ PDF for making observation ${\displaystyle y}$, given parameter values ${\displaystyle \theta }$ and design ${\displaystyle \xi }$ ${\displaystyle p(\theta )\,}$ prior PDF ${\displaystyle p(y\mid \xi )\,}$ marginal PDF in observation space ${\displaystyle p(\theta \mid y,\xi )\,}$    posterior PDF ${\displaystyle U(\xi )\,}$    utility of the design ${\displaystyle \xi }$ ${\displaystyle U(y,\xi )\,}$    utility of the experiment outcome after observation ${\displaystyle y}$ with design ${\displaystyle \xi }$

Given a vector ${\displaystyle \theta }$ of parameters to determine, a prior probability ${\displaystyle p(\theta )}$ over those parameters and a likelihood ${\displaystyle p(y\mid \theta ,\xi )}$ for making observation ${\displaystyle y}$, given parameter values ${\displaystyle \theta }$ and an experiment design ${\displaystyle \xi }$, the posterior probability can be calculated using Bayes' theorem

${\displaystyle p(\theta \mid y,\xi )={\frac {p(y\mid \theta ,\xi )p(\theta )}{p(y\mid \xi )}}\,,}$

where ${\displaystyle p(y\mid \xi )}$ is the marginal probability density in observation space

${\displaystyle p(y\mid \xi )=\int p(\theta )p(y\mid \theta ,\xi )\,d\theta \,.}$

The expected utility of an experiment with design ${\displaystyle \xi }$ can then be defined

${\displaystyle U(\xi )=\int p(y\mid \xi )U(y,\xi )\,dy,}$

where ${\displaystyle U(y,\xi )}$ is some real-valued functional of the posterior probability ${\displaystyle p(\theta \mid y,\xi )}$ after making observation ${\displaystyle y}$ using an experiment design ${\displaystyle \xi }$.

Gain in Shannon information as utility

Utility may be defined as the prior-posterior gain in Shannon information

${\displaystyle U(y,\xi )=\int \log(p(\theta \mid y,\xi ))\,p(\theta |y,\xi )\,d\theta -\int \log(p(\theta ))\,p(\theta )\,d\theta \,.}$

Another possibility is to define the utility as

${\displaystyle U(y,\xi )=D_{KL}(p(\theta \mid y,\xi )\|p(\theta ))\,,}$

the Kullback–Leibler divergence of the prior from the posterior distribution. Lindley (1956) noted that the expected utility will then be coordinate-independent and can be written in two forms

${\displaystyle {\begin{alignedat}{2}U(\xi )&=\int \int \log(p(\theta \mid y,\xi ))\,p(\theta ,y\mid \xi )\,d\theta \,dy-\int \log(p(\theta ))\,p(\theta )\,d\theta \\&=\int \int \log(p(y\mid \theta ,\xi ))\,p(\theta ,y\mid \xi )\,dy\,d\theta -\int \log(p(y\mid \xi ))\,p(y\mid \xi )\,dy,\end{alignedat}}\,}$

of which the latter can be evaluated without the need for evaluating individual posterior probability ${\displaystyle p(\theta \mid y,\xi )}$ for all possible observations ${\displaystyle y}$.^[5] It is worth noting that the second term on the second equation line will not depend on the design ${\displaystyle \xi }$, as long as the observational uncertainty doesn't. On the other hand, the integral of ${\displaystyle p(\theta )\log p(\theta )}$ in the first form is constant for all ${\displaystyle \xi }$, so if the goal is to choose the design with the highest utility, the term need not be computed at all. Several authors have considered numerical techniques for evaluating and optimizing this criterion.^[6][7] Note that

${\displaystyle U(\xi )=I(\theta ;y)\,,}$

the expected information gain being exactly the mutual information between the parameter θ and the observation y. An example of Bayesian design for linear dynamical model discrimination is given in Bania (2019).^[8] Since ${\displaystyle I(\theta ;y)\,,}$ was difficult to calculate, its lower bound has been used as a utility function. The lower bound is then maximized under the signal energy constraint. Proposed Bayesian design has been also compared with classical average D-optimal design. It was shown that the Bayesian design is superior to D-optimal design.

The Kelly criterion also describes such a utility function for a gambler seeking to maximize profit, which is used in gambling and information theory; Kelly's situation is identical to the foregoing, with the side information, or "private wire" taking the place of the experiment.

See also

• Bayesian optimization
• Optimal design
• Active Learning
• Expected value of sample information

References

Further reading

• DasGupta, A. (1996), "Review of optimal Bayes designs" (PDF), in Ghosh, S.;
  ISBN 978-0-444-82061-7
• Rainforth, Tom; et al. (2023), Modern Bayesian Experimental Design,
  arXiv:2302.14545