Empirical Bayes method

Empirical Bayes methods are procedures for statistical inference in which the prior probability distribution is estimated from the data. This approach stands in contrast to standard Bayesian methods, for which the prior distribution is fixed before any data are observed. Despite this difference in perspective, empirical Bayes may be viewed as an approximation to a fully Bayesian treatment of a hierarchical model wherein the parameters at the highest level of the hierarchy are set to their most likely values, instead of being integrated out. ^[1] Empirical Bayes, also known as maximum marginal likelihood , ^[2] represents a convenient approach for setting hyperparameters, but has been mostly supplanted by fully Bayesian hierarchical analyses since the 2000s with the increasing availability of well-performing computation techniques. It is still commonly used, however, for variational methods in Deep Learning, such as variational autoencoders, where latent variable spaces are high-dimensional.

Introduction

Empirical Bayes methods can be seen as an approximation to a fully Bayesian treatment of a hierarchical Bayes model.

In, for example, a two-stage hierarchical Bayes model, observed data ${\displaystyle y=\{y_{1},y_{2},\dots ,y_{n}\}}$ are assumed to be generated from an unobserved set of parameters ${\displaystyle \theta =\{\theta _{1},\theta _{2},\dots ,\theta _{n}\}}$ according to a probability distribution ${\displaystyle p(y\mid \theta )\,}$. In turn, the parameters ${\displaystyle \theta }$ can be considered samples drawn from a population characterised by hyperparameters ${\displaystyle \eta \,}$ according to a probability distribution ${\displaystyle p(\theta \mid \eta )\,}$. In the hierarchical Bayes model, though not in the empirical Bayes approximation, the hyperparameters ${\displaystyle \eta \,}$ are considered to be drawn from an unparameterized distribution ${\displaystyle p(\eta )\,}$.

Information about a particular quantity of interest ${\displaystyle \theta _{i}\;}$ therefore comes not only from the properties of those data ${\displaystyle y}$ that directly depend on it, but also from the properties of the population of parameters ${\displaystyle \theta \;}$ as a whole, inferred from the data as a whole, summarised by the hyperparameters ${\displaystyle \eta \;}$.

Using Bayes' theorem,

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

In general, this integral will not be tractable analytically or symbolically and must be evaluated by numerical methods. Stochastic (random) or deterministic approximations may be used. Example stochastic methods are Markov Chain Monte Carlo and Monte Carlo sampling. Deterministic approximations are discussed in quadrature.

Alternatively, the expression can be written as

${\displaystyle p(\theta \mid y)=\int p(\theta \mid \eta ,y)p(\eta \mid y)\;d\eta =\int {\frac {p(y\mid \theta )p(\theta \mid \eta )}{p(y\mid \eta )}}p(\eta \mid y)\;d\eta \,,}$

and the final factor in the integral can in turn be expressed as

${\displaystyle p(\eta \mid y)=\int p(\eta \mid \theta )p(\theta \mid y)\;d\theta .}$

These suggest an iterative scheme, qualitatively similar in structure to a Gibbs sampler, to evolve successively improved approximations to ${\displaystyle p(\theta \mid y)\;}$ and ${\displaystyle p(\eta \mid y)\;}$. First, calculate an initial approximation to ${\displaystyle p(\theta \mid y)\;}$ ignoring the ${\displaystyle \eta }$ dependence completely; then calculate an approximation to ${\displaystyle p(\eta \mid y)\;}$ based upon the initial approximate distribution of ${\displaystyle p(\theta \mid y)\;}$; then use this ${\displaystyle p(\eta \mid y)\;}$ to update the approximation for ${\displaystyle p(\theta \mid y)\;}$; then update ${\displaystyle p(\eta \mid y)\;}$; and so on.

When the true distribution ${\displaystyle p(\eta \mid y)\;}$ is sharply peaked, the integral determining ${\displaystyle p(\theta \mid y)\;}$ may be not much changed by replacing the probability distribution over ${\displaystyle \eta \;}$ with a point estimate ${\displaystyle \eta ^{*}\;}$ representing the distribution's peak (or, alternatively, its mean),

${\displaystyle p(\theta \mid y)\simeq {\frac {p(y\mid \theta )\;p(\theta \mid \eta ^{*})}{p(y\mid \eta ^{*})}}\,.}$

With this approximation, the above iterative scheme becomes the EM algorithm.

The term "Empirical Bayes" can cover a wide variety of methods, but most can be regarded as an early truncation of either the above scheme or something quite like it. Point estimates, rather than the whole distribution, are typically used for the parameter(s) ${\displaystyle \eta \;}$. The estimates for ${\displaystyle \eta ^{*}\;}$ are typically made from the first approximation to ${\displaystyle p(\theta \mid y)\;}$ without subsequent refinement. These estimates for ${\displaystyle \eta ^{*}\;}$ are usually made without considering an appropriate prior distribution for ${\displaystyle \eta }$.

Point estimation

Robbins' method: non-parametric empirical Bayes (NPEB)

Robbins ^[3] considered a case of sampling from a mixed distribution, where probability for each ${\displaystyle y_{i}}$ (conditional on ${\displaystyle \theta _{i}}$) is specified by a Poisson distribution,

${\displaystyle p(y_{i}\mid \theta _{i})={{\theta _{i}}^{y_{i}}e^{-\theta _{i}} \over {y_{i}}!}}$

while the prior on θ is unspecified except that it is also i.i.d. from an unknown distribution, with cumulative distribution function ${\displaystyle G(\theta )}$. Compound sampling arises in a variety of statistical estimation problems, such as accident rates and clinical trials.^{[ citation needed ]} We simply seek a point prediction of ${\displaystyle \theta _{i}}$ given all the observed data. Because the prior is unspecified, we seek to do this without knowledge of G. ^[4]

Under squared error loss (SEL), the conditional expectation E(θ_i | Y_i = y_i) is a reasonable quantity to use for prediction. For the Poisson compound sampling model, this quantity is

${\displaystyle \operatorname {E} (\theta _{i}\mid y_{i})={\int (\theta ^{y_{i}+1}e^{-\theta }/{y_{i}}!)\,dG(\theta ) \over {\int (\theta ^{y_{i}}e^{-\theta }/{y_{i}}!)\,dG(\theta })}.}$

This can be simplified by multiplying both the numerator and denominator by ${\displaystyle ({y_{i}}+1)}$, yielding

${\displaystyle \operatorname {E} (\theta _{i}\mid y_{i})={{(y_{i}+1)p_{G}(y_{i}+1)} \over {p_{G}(y_{i})}},}$

where p_G is the marginal probability mass function obtained by integrating out θ over G.

To take advantage of this, Robbins ^[3] suggested estimating the marginals with their empirical frequencies (${\displaystyle \#\{Y_{j}\}}$), yielding the fully non-parametric estimate as:

${\displaystyle \operatorname {E} (\theta _{i}\mid y_{i})\approx (y_{i}+1){{\#\{Y_{j}=y_{i}+1\}} \over {\#\{Y_{j}=y_{i}\}}},}$

where ${\displaystyle \#}$ denotes "number of". (See also Good–Turing frequency estimation.)

Example – Accident rates

Suppose each customer of an insurance company has an "accident rate" Θ and is insured against accidents; the probability distribution of Θ is the underlying distribution, and is unknown. The number of accidents suffered by each customer in a specified time period has a Poisson distribution with expected value equal to the particular customer's accident rate. The actual number of accidents experienced by a customer is the observable quantity. A crude way to estimate the underlying probability distribution of the accident rate Θ is to estimate the proportion of members of the whole population suffering 0, 1, 2, 3, ... accidents during the specified time period as the corresponding proportion in the observed random sample. Having done so, it is then desired to predict the accident rate of each customer in the sample. As above, one may use the conditional expected value of the accident rate Θ given the observed number of accidents during the baseline period. Thus, if a customer suffers six accidents during the baseline period, that customer's estimated accident rate is 7 × [the proportion of the sample who suffered 7 accidents] / [the proportion of the sample who suffered 6 accidents]. Note that if the proportion of people suffering k accidents is a decreasing function of k, the customer's predicted accident rate will often be lower than their observed number of accidents.

This shrinkage effect is typical of empirical Bayes analyses.

Gaussian

Suppose ${\displaystyle X,Y}$ are random variables, such that ${\displaystyle Y}$ is observed, but ${\displaystyle X}$ is hidden. The problem is to find the expectation of ${\displaystyle X}$, conditional on ${\displaystyle Y}$. Suppose further that ${\displaystyle Y|X\sim {\mathcal {N}}(X,\Sigma )}$, that is, ${\displaystyle Y=X+Z}$, where ${\displaystyle Z}$ is a multivariate gaussian with variance ${\displaystyle \Sigma }$.

Then, we have the formula $${\displaystyle \Sigma \nabla _{y}\rho (y|x)=\rho (y|x)(x-y)}$$by direct calculation with the probability density function of multivariate gaussians. Integrating over ${\displaystyle \rho (x)dx}$, we obtain$${\displaystyle \Sigma \nabla _{y}\rho (y)=(\mathbb {E} [x|y]-y)\rho (y)\implies \mathbb {E} [x|y]=y+\Sigma \nabla _{y}\ln \rho (y)}$$In particular, this means that one can perform Bayesian estimation of ${\displaystyle X}$ without access to either the prior density of ${\displaystyle X}$ or the posterior density of ${\displaystyle Y}$. The only requirement is to have access to the score function of ${\displaystyle Y}$. This has applications in score-based generative modeling. ^[5]

Parametric empirical Bayes

If the likelihood and its prior take on simple parametric forms (such as 1- or 2-dimensional likelihood functions with simple conjugate priors), then the empirical Bayes problem is only to estimate the marginal ${\displaystyle m(y\mid \eta )}$ and the hyperparameters ${\displaystyle \eta }$ using the complete set of empirical measurements. For example, one common approach, called parametric empirical Bayes point estimation, is to approximate the marginal using the maximum likelihood estimate (MLE), or a moments expansion, which allows one to express the hyperparameters ${\displaystyle \eta }$ in terms of the empirical mean and variance. This simplified marginal allows one to plug in the empirical averages into a point estimate for the prior ${\displaystyle \theta }$. The resulting equation for the prior ${\displaystyle \theta }$ is greatly simplified, as shown below.

There are several common parametric empirical Bayes models, including the Poisson–gamma model (below), the Beta-binomial model, the Gaussian–Gaussian model, the Dirichlet-multinomial model, as well specific models for Bayesian linear regression (see below) and Bayesian multivariate linear regression. More advanced approaches include hierarchical Bayes models and Bayesian mixture models.

Gaussian–Gaussian model

For an example of empirical Bayes estimation using a Gaussian-Gaussian model, see Empirical Bayes estimators.

Poisson–gamma model

For example, in the example above, let the likelihood be a Poisson distribution, and let the prior now be specified by the conjugate prior, which is a gamma distribution (${\displaystyle G(\alpha ,\beta )}$) (where ${\displaystyle \eta =(\alpha ,\beta )}$):

${\displaystyle \rho (\theta \mid \alpha ,\beta )\,d\theta ={\frac {(\theta /\beta )^{\alpha -1}\,e^{-\theta /\beta }}{\Gamma (\alpha )}}\,(d\theta /\beta ){\text{ for }}\theta >0,\alpha >0,\beta >0\,\!.}$

It is straightforward to show the posterior is also a gamma distribution. Write

${\displaystyle \rho (\theta \mid y)\propto \rho (y\mid \theta )\rho (\theta \mid \alpha ,\beta ),}$

where the marginal distribution has been omitted since it does not depend explicitly on ${\displaystyle \theta }$. Expanding terms which do depend on ${\displaystyle \theta }$ gives the posterior as:

${\displaystyle \rho (\theta \mid y)\propto (\theta ^{y}\,e^{-\theta })(\theta ^{\alpha -1}\,e^{-\theta /\beta })=\theta ^{y+\alpha -1}\,e^{-\theta (1+1/\beta )}.}$

So the posterior density is also a gamma distribution ${\displaystyle G(\alpha ',\beta ')}$, where ${\displaystyle \alpha '=y+\alpha }$, and ${\displaystyle \beta '=(1+1/\beta )^{-1}}$. Also notice that the marginal is simply the integral of the posterior over all ${\displaystyle \Theta }$, which turns out to be a negative binomial distribution.

To apply empirical Bayes, we will approximate the marginal using the maximum likelihood estimate (MLE). But since the posterior is a gamma distribution, the MLE of the marginal turns out to be just the mean of the posterior, which is the point estimate ${\displaystyle \operatorname {E} (\theta \mid y)}$ we need. Recalling that the mean ${\displaystyle \mu }$ of a gamma distribution ${\displaystyle G(\alpha ',\beta ')}$ is simply ${\displaystyle \alpha '\beta '}$, we have

${\displaystyle \operatorname {E} (\theta \mid y)=\alpha '\beta '={\frac {{\bar {y}}+\alpha }{1+1/\beta }}={\frac {\beta }{1+\beta }}{\bar {y}}+{\frac {1}{1+\beta }}(\alpha \beta ).}$

To obtain the values of ${\displaystyle \alpha }$ and ${\displaystyle \beta }$, empirical Bayes prescribes estimating mean ${\displaystyle \alpha \beta }$ and variance ${\displaystyle \alpha \beta ^{2}}$ using the complete set of empirical data.

The resulting point estimate ${\displaystyle \operatorname {E} (\theta \mid y)}$ is therefore like a weighted average of the sample mean ${\displaystyle {\bar {y}}}$ and the prior mean ${\displaystyle \mu =\alpha \beta }$. This turns out to be a general feature of empirical Bayes; the point estimates for the prior (i.e. mean) will look like a weighted averages of the sample estimate and the prior estimate (likewise for estimates of the variance).

See also

• Bayes estimator
• Bayesian network
• Hyperparameter
• Hyperprior
• Best linear unbiased prediction
• Robbins lemma
• Spike-and-slab variable selection

References

1. Carlin, Bradley P.; Louis, Thomas A. (2002). "Empirical Bayes: Past, Present, and Future". In Raftery, Adrian E.; Tanner, Martin A.; Wells, Martin T. (eds.). Statistics in the 21st Century. Chapman & Hall. pp. 312–318. ISBN   1-58488-272-7.
2. C.M. Bishop (2005). Neural networks for pattern recognition. Oxford University Press ISBN   0-19-853864-2
3. Robbins, Herbert (1956). "An Empirical Bayes Approach to Statistics". Proceedings of the Third Berkeley Symposium on Mathematical Statistics and Probability, Volume 1: Contributions to the Theory of Statistics. Springer Series in Statistics: 157–163. doi:10.1007/978-1-4612-0919-5_26. ISBN   978-0-387-94037-3. MR   0084919.
4. Carlin, Bradley P.; Louis, Thomas A. (2000). Bayes and Empirical Bayes Methods for Data Analysis (2nd ed.). Chapman & Hall/CRC. pp. Sec. 3.2 and Appendix B. ISBN   978-1-58488-170-4.
5. Saremi, Saeed; Hyvärinen, Aapo (2019). "Neural Empirical Bayes". Journal of Machine Learning Research. 20 (181): 1–23. ISSN   1533-7928.

Further reading

• Peter E. Rossi; Greg M. Allenby; Rob McCulloch (14 May 2012). Bayesian Statistics and Marketing. John Wiley & Sons. ISBN   978-0-470-86368-8.
• Casella, George (May 1985). "An Introduction to Empirical Bayes Data Analysis" (PDF). American Statistician. 39 (2): 83–87. doi:10.2307/2682801. hdl: 1813/32886 . JSTOR   2682801. MR   0789118.
• Nikulin, Mikhail (1987). "Bernstein's regularity conditions in a problem of empirical Bayesian approach". Journal of Soviet Mathematics. 36 (5): 596–600. doi: 10.1007/BF01093293 . S2CID   122405908.

External links

• Use of empirical Bayes Method in estimating road safety (North America)
• Empirical Bayes methods for missing data analysis
• Using the Beta-Binomial distribution to assess performance of a biometric identification device
• A Hierarchical Naive Bayes Classifiers (for continuous and discrete variables).