Normal-gamma distribution explained

In probability theory and statistics, the normal-gamma distribution (or Gaussian-gamma distribution) is a bivariate four-parameter family of continuous probability distributions. It is the conjugate prior of a normal distribution with unknown mean and precision.^[1]

Definition

For a pair of random variables, (X,T), suppose that the conditional distribution of X given T is given by

X\mid T \sim N(\mu, 1 /( λ T)) , \mu and precision λ T — equivalently, with variance 1 / ( λ T) .

Suppose also that the marginal distribution of T is given by

T \mid \alpha, \beta \sim \operatorname{Gamma}(\alpha,\beta),

where this means that T has a gamma distribution. Here λ, α and β are parameters of the joint distribution.

Then (X,T) has a normal-gamma distribution, and this is denoted by

(X,T) \sim \operatorname{NormalGamma}(\mu, λ ,\alpha,\beta).

Properties

Probability density function

The joint probability density function of (X,T) is

$$f(x,\backslash tau\backslash mid\backslash mu,\backslash lambda,\backslash alpha,\backslash beta)\; =\; \backslash frac\; \backslash ,\; \backslash tau^\backslash ,e^\backslash exp\backslash left(-\backslash frac\backslash right)$$

Marginal distributions

By construction, the marginal distribution of

\tau is a gamma distribution, and the conditional distribution of x given \tau is a Gaussian distribution. The marginal distribution of x is a three-parameter non-standardized Student's t-distribution with parameters (\nu, \mu, \sigma ^{2)=(2\alpha,} \mu, \beta/( λ \alpha)) .

Exponential family

\alpha -1 / 2 , - \beta -λ \mu ^{2/2,} λ \mu, -λ / 2 and natural statistics ln \tau, \tau, \tau x, \tau x ^{2} .

Moments of the natural statistics

The following moments can be easily computed using the moment generating function of the sufficient statistic:

\operatorname{E}( ln T) = \psi\left(\alpha\right) - ln \beta,

where

\psi\left(\alpha\right) is the digamma function,\begin{align} \operatorname{E}(T) & = (\alpha)/(\beta) , \\[ 5 pt] \operatorname{E}(TX) & = \mu (\alpha)/(\beta) , \\[ 5 pt] \operatorname{E}(TX ^{2)} & = (1)/(λ) + \mu ^{2} (\alpha)/(\beta) . \end{align}

Scaling

If

(X,T) \sim NormalGamma (\mu, λ ,\alpha,\beta), then for any b> 0 , (bX,bT) is distributed as {\rm NormalGamma}(b\mu, λ / b ^{3,} \alpha, \beta/ b ).

Posterior distribution of the parameters

Assume that x is distributed according to a normal distribution with unknown mean

\mu and precision \tau .x \sim l{N}(\mu, \tau ^{-1} ) and that the prior distribution on \mu and \tau , (\mu,\tau) , has a normal-gamma distribution(\mu,\tau) \sim NormalGamma (\mu _{0,λ} _{0,\alpha} _{0,\beta} _{0)} ,

for which the density satisfies

\pi(\mu,\tau) \propto \tau ^{\alpha _{0- (1)/(2)}} \exp[ - \beta _{0\tau]\exp\left[} - (λ _{0 \tau(\mu - \mu} ^{2} ^{0 )})/(2) \right].

Suppose

x _{1,\ldots,x} _{n} \mid \mu,\tau \sim \operatorname{{i.}{i.}{d.}} \operatorname N\left( \mu, \tau ^{-1} \right), i.e. the components of X = (x _{1,\ldots,x} _{n)} are conditionally independent given \mu,\tau and the conditional distribution of each of them given \mu,\tau is normal with expected value \mu and variance 1 / \tau. The posterior distribution of \mu and \tau given this dataset X can be analytically determined by Bayes' theorem^[2] explicitly,P (\tau,\mu \mid X ) \propto L ( X \mid \tau,\mu) \pi(\tau,\mu),

where

L is the likelihood of the parameters given the data.

Since the data are i.i.d, the likelihood of the entire dataset is equal to the product of the likelihoods of the individual data samples:

L ( X \mid \tau, \mu) = \prod ^{n} _{i =1} L (x _{i} \mid \tau, \mu).

This expression can be simplified as follows:

\begin{align} L ( X \mid \tau, \mu) & \propto \prod ^{n} _{i =1} \tau ^{1/2} \exp\left[ (- \tau)/(2) (x ^{2 \right]} _{i - \mu)} \\[ 5 pt] & \propto \tau ^{n/2} \exp\left[ (- \tau)/(2) \sum ^{2 \right]} _{i - \mu)} \\[ 5 pt] & \propto \tau ^{n/2} \exp\left[ (- \tau)/(2) \sum ^{n(x} _{i - \bar{x}} + \bar{x} - \mu) ^{2} \right] \\[ 5 pt] & \propto \tau ^{n/2} \exp\left[ (- \tau)/(2) \sum ^{n} _{i =1} \left((x ^{2} _{i - \bar{x})} + (\bar{x} - \mu) ^{2} \right)\right] \\[ 5 pt] & \propto \tau ^{n/2} \exp\left[ (- \tau)/(2) \left(n s + n(\bar{x} - \mu) ^{2\right)\right], \end{align}}

where

\bar{x} = (1)/(n) \sum ^{n} _{i =1} x _{i} , the mean of the data samples, and s = (1)/(n) \sum ^{2} _{i - \bar{x})} , the sample variance.

The posterior distribution of the parameters is proportional to the prior times the likelihood.

\begin{align} P (\tau, \mu \mid X ) &\propto L ( X \mid \tau,\mu) \pi(\tau,\mu) \\ &\propto \tau ^{n/2} \exp \left[ (- \tau)/(2) \left(n s + n(\bar{x} - \mu) ^{2\right)} \right] \tau ^{\alpha _{0- (1)/(2)}} \exp[{ - \beta _{0 \tau}]\exp\left[ - (λ _{0 \tau(\mu - \mu} ^{2} ^{0 )})/(2)} \right] \ &\propto \tau ^{(n)/(+ \alpha _{0} - (1)/(2) 2)} \exp\left[ - \tau \left( (1)/(2) n s + \beta _{0} \right) \right] \exp\left[ - (\tau)/(2) \left( λ _{0(\mu-\mu} ^{2} ^{0 )} + n(\bar{x} - \mu) ^{2\right)\right]} \end{align}

The final exponential term is simplified by completing the square.

\begin{align} λ _{0(\mu-\mu} ^{2} ^{0 )} + n(\bar{x} - \mu) ^{2 & =λ} _{0} \mu ^{2} - 2 λ _{0} \mu \mu _{0} + λ _{0} \mu ^{2} _{0} + n \mu ^{2} - 2 n \bar{x} \mu + n \bar{x} ^{2} \\ & = ( λ _{0} + n) \mu ^{2} - 2 ( λ _{0} \mu _{0} + n \bar{x}) \mu + λ _{0} \mu ^{2} _{0} + n \bar{x} ^{2} \\ & = ( λ _{0} + n)( \mu ^{2} - 2 (λ _{0} \mu _{0} + n \bar{x)/() } \mu) + \lambda_0 \mu_0^2 +n \bar^2 \\&= (\lambda_0 + n)\left(\mu - \frac \right) ^2 + \lambda_0 \mu_0^2 +n \bar^2 - \frac \\&= (\lambda_0 + n)\left(\mu - \frac \right) ^2 + \frac\end

On inserting this back into the expression above,

\begin{align} P (\tau, \mu \mid X ) & \propto \tau ^{(n)/(+ \alpha _{0} - (1)/(2) 2)} \exp \left[ - \tau \left( (1)/(2) n s + \beta _{0} \right) \right] \exp \left[ - (\tau)/(2) \left( \left( λ _{0} + n \right) \left(\mu - (λ _{0} \mu _{0} + n \bar{x)/() } \right)^2 + \frac \right) \right]\\& \propto \tau^ \exp \left[-\tau \left(\frac{1}{2} n s + \beta_0 + \frac{\lambda_0 n (\bar{x} - \mu_0)^2}{2(\lambda_0 +n)} \right) \right] \exp \left[- \frac{\tau}{2} \left(\lambda_0 + n \right) \left(\mu- \frac{\lambda_0 \mu_0 + n \bar{x}}{\lambda_0 + n} \right)^2 \right]\end

This final expression is in exactly the same form as a Normal-Gamma distribution, i.e.,

P (\tau, \mu \mid X ) =NormalGamma \left( (λ _{0} \mu _{0} + n \bar{x)/() }, \lambda_0 + n, \alpha_0+\frac, \beta_0+ \frac\left(n s + \frac \right) \right)

Interpretation of parameters

The interpretation of parameters in terms of pseudo-observations is as follows:

• The new mean takes a weighted average of the old pseudo-mean and the observed mean, weighted by the number of associated (pseudo-)observations.
• The precision was estimated from

2 \alpha pseudo-observations (i.e. possibly a different number of pseudo-observations, to allow the variance of the mean and precision to be controlled separately) with sample mean \mu and sample variance (\beta)/(\alpha) (i.e. with sum of squared deviations 2 \beta ).

• The posterior updates the number of pseudo-observations (

λ _{0} ) simply by adding the corresponding number of new observations (n ).

• The new sum of squared deviations is computed by adding the previous respective sums of squared deviations. However, a third "interaction term" is needed because the two sets of squared deviations were computed with respect to different means, and hence the sum of the two underestimates the actual total squared deviation.

As a consequence, if one has a prior mean of

\mu _{0} from n _{\mu} samples and a prior precision of \tau _{0} from n _{\tau} samples, the prior distribution over \mu and \tau isP (\tau,\mu \mid X ) = \operatorname{NormalGamma} \left(\mu _{0,} n _{\mu} , (n _{\tau})/(2) , (n _{\tau})/(2 \tau _{0}) \right)

and after observing

n samples with mean \mu and variance s , the posterior probability isP (\tau,\mu \mid X ) = NormalGamma \left( (n _{\mu} \mu _{0} + n \mu)/(n _{\mu} + n) , n _{\mu} + n , (1)/(2) (n _{\tau+n),} (1)/(\left( 2) (n _{\tau})/(\tau _{0}) + n s + (n _{\mu} n (\mu - \mu ^{2} _{0 )})/(n _{\mu+n}) \right) \right)

Note that in some programming languages, such as Matlab, the gamma distribution is implemented with the inverse definition of

\beta , so the fourth argument of the Normal-Gamma distribution is 2 \tau _{0} /n _{\tau} .

Generating normal-gamma random variates

Generation of random variates is straightforward:

1. Sample

\tau from a gamma distribution with parameters \alpha and \beta

1. Sample

x from a normal distribution with mean \mu and variance 1 /( λ \tau)

Related distributions

• The normal-inverse-gamma distribution is essentially the same distribution parameterized by variance rather than precision
• The normal-exponential-gamma distribution

References

• Bernardo, J.M.; Smith, A.F.M. (1993) Bayesian Theory, Wiley.
• Dearden et al. "Bayesian Q-learning", Proceedings of the Fifteenth National Conference on Artificial Intelligence (AAAI-98), July 26–30, 1998, Madison, Wisconsin, USA.

Notes and References

1. Bernardo & Smith (1993, pages 136, 268, 434)
2. Web site: Bayes' Theorem: Introduction . 2014-08-05 . live . https://web.archive.org/web/20140807091855/http://www.trinity.edu/cbrown/bayesweb/ . 2014-08-07 .