Bayesian linear regression explained
Bayesian linear regression is a type of conditional modeling in which the mean of one variable is described by a linear combination of other variables, with the goal of obtaining the posterior probability of the regression coefficients (as well as other parameters describing the distribution of the regressand) and ultimately allowing the out-of-sample prediction of the regressand (often labelled
)
conditional on observed values of the regressors (usually
). The simplest and most widely used version of this model is the
normal linear model, in which
given
is distributed
Gaussian. In this model, and under a particular choice of
prior probabilities for the parameters—so-called
conjugate priors—the posterior can be found analytically. With more arbitrarily chosen priors, the posteriors generally have to be approximated.
Model setup
Consider a standard linear regression problem, in which for
we specify the mean of the
conditional distribution of
given a
predictor vector
:
where
is a
vector, and the
are
independent and identically normally distributed random variables:
This corresponds to the following likelihood function:
The ordinary least squares solution is used to estimate the coefficient vector using the Moore–Penrose pseudoinverse:
where
is the
design matrix, each row of which is a predictor vector
; and
is the column
-vector
.
This is a frequentist approach, and it assumes that there are enough measurements to say something meaningful about
. In the
Bayesian approach, the data are supplemented with additional information in the form of a
prior probability distribution. The prior belief about the parameters is combined with the data's likelihood function according to
Bayes theorem to yield the
posterior belief about the parameters
and
. The prior can take different functional forms depending on the domain and the information that is available
a priori.
Since the data comprise both
and
, the focus only on the distribution of
conditional on
needs justification. In fact, a "full" Bayesian analysis would require a joint likelihood
\rho(y,X\mid\boldsymbol\beta,\sigma2,\gamma)
along with a prior
\rho(\beta,\sigma2,\gamma)
, where
symbolizes the parameters of the distribution for
. Only under the assumption of (weak) exogeneity can the joint likelihood be factored into
\rho(y\mid\boldsymbolX,\beta,\sigma2)\rho(X\mid\gamma)
.
[1] The latter part is usually ignored under the assumption of disjoint parameter sets. More so, under classic assumptions
are considered chosen (for example, in a designed experiment) and therefore has a known probability without parameters.
[2] With conjugate priors
Conjugate prior distribution
For an arbitrary prior distribution, there may be no analytical solution for the posterior distribution. In this section, we will consider a so-called conjugate prior for which the posterior distribution can be derived analytically.
A prior
\rho(\boldsymbol\beta,\sigma2)
is
conjugate to this likelihood function if it has the same functional form with respect to
and
. Since the log-likelihood is quadratic in
, the log-likelihood is re-written such that the likelihood becomes normal in
(\boldsymbol\beta-\hat{\boldsymbol\beta})
. Write
The likelihood is now re-written aswherewhere
is the number of regression coefficients.
This suggests a form for the prior:where
is an
inverse-gamma distributionIn the notation introduced in the inverse-gamma distribution article, this is the density of an
distribution with
and
with
and
as the prior values of
and
, respectively. Equivalently, it can also be described as a
scaled inverse chi-squared distribution,
Further the conditional prior density
\rho(\boldsymbol\beta|\sigma2)
is a
normal distribution,
In the notation of the normal distribution, the conditional prior distribution is
l{N}\left(\boldsymbol\mu0,\sigma2
\right).
Posterior distribution
With the prior now specified, the posterior distribution can be expressed as
With some re-arrangement,[3] the posterior can be re-written so that the posterior mean
of the parameter vector
can be expressed in terms of the least squares estimator
and the prior mean
, with the strength of the prior indicated by the prior precision matrix
To justify that
is indeed the posterior mean, the quadratic terms in the exponential can be re-arranged as a
quadratic form in
\boldsymbol\beta-\boldsymbol\mun
.
[4]
Now the posterior can be expressed as a normal distribution times an inverse-gamma distribution:
Therefore, the posterior distribution can be parametrized as follows.where the two factors correspond to the densities of
l{N}\left(\boldsymbol\mun,
\right)
and
Inv-Gamma\left(an,bn\right)
distributions, with the parameters of these given by
which illustrates Bayesian inference being a compromise between the information contained in the prior and the information contained in the sample.
Model evidence
is the probability of the data given the model
. It is also known as the
marginal likelihood, and as the
prior predictive density. Here, the model is defined by the likelihood function
p(y\midX,\boldsymbol\beta,\sigma)
and the prior distribution on the parameters, i.e.
p(\boldsymbol\beta,\sigma)
. The model evidence captures in a single number how well such a model explains the observations. The model evidence of the Bayesian linear regression model presented in this section can be used to compare competing linear models by
Bayesian model comparison. These models may differ in the number and values of the predictor variables as well as in their priors on the model parameters. Model complexity is already taken into account by the model evidence, because it marginalizes out the parameters by integrating
p(y,\boldsymbol\beta,\sigma\midX)
over all possible values of
and
.
This integral can be computed analytically and the solution is given in the following equation.
[5] Here
denotes the
gamma function. Because we have chosen a conjugate prior, the marginal likelihood can also be easily computed by evaluating the following equality for arbitrary values of
and
.
Note that this equation is nothing but a re-arrangement of
Bayes theorem. Inserting the formulas for the prior, the likelihood, and the posterior and simplifying the resulting expression leads to the analytic expression given above.
Other cases
In general, it may be impossible or impractical to derive the posterior distribution analytically. However, it is possible to approximate the posterior by an approximate Bayesian inference method such as Monte Carlo sampling,[6] INLA or variational Bayes.
The special case
is called
ridge regression.
A similar analysis can be performed for the general case of the multivariate regression and part of this provides for Bayesian estimation of covariance matrices: see Bayesian multivariate linear regression.
See also
References
- Book: Box, G. E. P. . George E. P. Box . George Tiao. Tiao . G. C. . 1973 . Bayesian Inference in Statistical Analysis . Wiley . 0-471-57428-7 .
- Book: Carlin, Bradley P. . Louis, Thomas A.. Bayesian Methods for Data Analysis . Third . Boca Raton, FL: Chapman and Hall/CRC . 2008 . 1-58488-697-8.
- Book: Fahrmeir, L. . Kneib, T. . Lang, S.. Regression. Modelle, Methoden und Anwendungen . Second . Springer . Heidelberg . 2009 . 978-3-642-01836-7 . 10.1007/978-3-642-01837-4.
- Book: Gelman, Andrew . Andrew Gelman . Carlin . John B. . Stern . Hal S. . Dunson . David B. . Vehtari . Aki . Rubin . Donald B. . 1 . Introduction to regression models . 353–380 . Bayesian Data Analysis . Third . Boca Raton, FL: Chapman and Hall/CRC . 2013 . 978-1-4398-4095-5 .
- Book: Jackman, Simon . Regression models . 99–124 . Bayesian Analysis for the Social Sciences . Wiley . 2009 . 978-0-470-01154-6 .
- Book: Rossi, Peter E. . Greg M. . Allenby . Robert . McCulloch . Bayesian Statistics and Marketing . John Wiley & Sons . 2006 . 0470863676 .
- Book: O'Hagan, Anthony. Bayesian Inference. 2B . Kendall's Advanced Theory of Statistics. 1994 . First . 0-340-52922-9. Halsted.
External links
- Bayesian estimation of linear models (R programming wikibook). Bayesian linear regression as implemented in R.
Notes and References
- See Jackman (2009), p. 101.
- See Gelman et al. (2013), p. 354.
- The intermediate steps of this computation can be found in O'Hagan (1994) at the beginning of the chapter on Linear models.
- The intermediate steps are in Fahrmeir et al. (2009) on page 188.
- The intermediate steps of this computation can be found in O'Hagan (1994) on page 257.
- Carlin and Louis (2008) and Gelman, et al. (2003) explain how to use sampling methods for Bayesian linear regression.