Metropolis–Hastings algorithm explained

In statistics and statistical physics, the Metropolis–Hastings algorithm is a Markov chain Monte Carlo (MCMC) method for obtaining a sequence of random samples from a probability distribution from which direct sampling is difficult. New samples are added to the sequence in two steps: first a new sample is proposed based on the previous sample, then the proposed sample is either added to the sequence or rejected depending on the value of the probability distribution at that point. The resulting sequence can be used to approximate the distribution (e.g. to generate a histogram) or to compute an integral (e.g. an expected value).

Metropolis–Hastings and other MCMC algorithms are generally used for sampling from multi-dimensional distributions, especially when the number of dimensions is high. For single-dimensional distributions, there are usually other methods (e.g. adaptive rejection sampling) that can directly return independent samples from the distribution, and these are free from the problem of autocorrelated samples that is inherent in MCMC methods.

History

The algorithm is named in part for Nicholas Metropolis, the first coauthor of a 1953 paper, entitled Equation of State Calculations by Fast Computing Machines, with Arianna W. Rosenbluth, Marshall Rosenbluth, Augusta H. Teller and Edward Teller. For many years the algorithm was known simply as the Metropolis algorithm.[1] [2] The paper proposed the algorithm for the case of symmetrical proposal distributions, but in 1970, W.K. Hastings extended it to the more general case. The generalized method was eventually identified by both names, although the first use of the term "Metropolis-Hastings algorithm" is unclear.

Some controversy exists with regard to credit for development of the Metropolis algorithm. Metropolis, who was familiar with the computational aspects of the method, had coined the term "Monte Carlo" in an earlier article with Stanisław Ulam, and led the group in the Theoretical Division that designed and built the MANIAC I computer used in the experiments in 1952. However, prior to 2003 there was no detailed account of the algorithm's development. Shortly before his death, Marshall Rosenbluth attended a 2003 conference at LANL marking the 50th anniversary of the 1953 publication. At this conference, Rosenbluth described the algorithm and its development in a presentation titled "Genesis of the Monte Carlo Algorithm for Statistical Mechanics". Further historical clarification is made by Gubernatis in a 2005 journal article recounting the 50th anniversary conference. Rosenbluth makes it clear that he and his wife Arianna did the work, and that Metropolis played no role in the development other than providing computer time.

This contradicts an account by Edward Teller, who states in his memoirs that the five authors of the 1953 article worked together for "days (and nights)". In contrast, the detailed account by Rosenbluth credits Teller with a crucial but early suggestion to "take advantage of statistical mechanics and take ensemble averages instead of following detailed kinematics". This, says Rosenbluth, started him thinking about the generalized Monte Carlo approach – a topic which he says he had discussed often with John Von Neumann. Arianna Rosenbluth recounted (to Gubernatis in 2003) that Augusta Teller started the computer work, but that Arianna herself took it over and wrote the code from scratch. In an oral history recorded shortly before his death, Rosenbluth again credits Teller with posing the original problem, himself with solving it, and Arianna with programming the computer.

Description

P(x)

, provided that we know a function

f(x)

proportional to the density

P

and the values of

f(x)

can be calculated. The requirement that

f(x)

must only be proportional to the density, rather than exactly equal to it, makes the Metropolis–Hastings algorithm particularly useful, because it removes the need to calculate the density's normalization factor, which is often extremely difficult in practice.

The Metropolis–Hastings algorithm generates a sequence of sample values in such a way that, as more and more sample values are produced, the distribution of values more closely approximates the desired distribution. These sample values are produced iteratively in such a way, that the distribution of the next sample depends only on the current sample value, which makes the sequence of samples a Markov chain. Specifically, at each iteration, the algorithm proposes a candidate for the next sample value based on the current sample value. Then, with some probability, the candidate is either accepted, in which case the candidate value is used in the next iteration, or it is rejected in which case the candidate value is discarded, and the current value is reused in the next iteration. The probability of acceptance is determined by comparing the values of the function

f(x)

of the current and candidate sample values with respect to the desired distribution.

The method used to propose new candidates is characterized by the probability distribution

g(x\midy)

(sometimes written

Q(x\midy)

) of a new proposed sample

x

given the previous sample

y

. This is called the proposal density, proposal function, or jumping distribution. A common choice for

g(x\midy)

is a Gaussian distribution centered at

y

, so that points closer to

y

are more likely to be visited next, making the sequence of samples into a Gaussian random walk. In the original paper by Metropolis et al. (1953),

g(x\midy)

was suggested to be a uniform distribution limited to some maximum distance from

y

. More complicated proposal functions are also possible, such as those of Hamiltonian Monte Carlo, Langevin Monte Carlo, or preconditioned Crank–Nicolson.

For the purpose of illustration, the Metropolis algorithm, a special case of the Metropolis–Hastings algorithm where the proposal function is symmetric, is described below.

Metropolis algorithm (symmetric proposal distribution):Let

f(x)

be a function that is proportional to the desired probability density function

P(x)

(a.k.a. a target distribution).
  1. Initialization: Choose an arbitrary point

xt

to be the first observation in the sample and choose a proposal function

g(x\midy)

. In this section,

g

is assumed to be symmetric; in other words, it must satisfy

g(x\midy)=g(y\midx)

.
  1. For each iteration t:
    • Propose a candidate

x'

for the next sample by picking from the distribution

g(x'\midxt)

.
    • Calculate the acceptance ratio

\alpha=f(x')/f(xt)

, which will be used to decide whether to accept or reject the candidate. Because f is proportional to the density of P, we have that

\alpha=f(x')/f(xt)=P(x')/P(xt)

.
    • Accept or reject:
      • Generate a uniform random number

u\in[0,1]

.
      • If

u\le\alpha

, then accept the candidate by setting

xt+1=x'

,
      • If

u>\alpha

, then reject the candidate and set

xt+1=xt

instead.

This algorithm proceeds by randomly attempting to move about the sample space, sometimes accepting the moves and sometimes remaining in place. Note that the acceptance ratio

\alpha

indicates how probable the new proposed sample is with respect to the current sample, according to the distribution whose density is

P(x)

. If we attempt to move to a point that is more probable than the existing point (i.e. a point in a higher-density region of

P(x)

corresponding to an

\alpha>1\geu

), we will always accept the move. However, if we attempt to move to a less probable point, we will sometimes reject the move, and the larger the relative drop in probability, the more likely we are to reject the new point. Thus, we will tend to stay in (and return large numbers of samples from) high-density regions of

P(x)

, while only occasionally visiting low-density regions. Intuitively, this is why this algorithm works and returns samples that follow the desired distribution with density

P(x)

.

Compared with an algorithm like adaptive rejection sampling[3] that directly generates independent samples from a distribution, Metropolis–Hastings and other MCMC algorithms have a number of disadvantages:

P(x)

, a set of nearby samples will be correlated with each other and not correctly reflect the distribution. This means that effective sample sizes can be significantly lower than the number of samples actually taken, leading to large errors.

On the other hand, most simple rejection sampling methods suffer from the "curse of dimensionality", where the probability of rejection increases exponentially as a function of the number of dimensions. Metropolis–Hastings, along with other MCMC methods, do not have this problem to such a degree, and thus are often the only solutions available when the number of dimensions of the distribution to be sampled is high. As a result, MCMC methods are often the methods of choice for producing samples from hierarchical Bayesian models and other high-dimensional statistical models used nowadays in many disciplines.

In multivariate distributions, the classic Metropolis–Hastings algorithm as described above involves choosing a new multi-dimensional sample point. When the number of dimensions is high, finding the suitable jumping distribution to use can be difficult, as the different individual dimensions behave in very different ways, and the jumping width (see above) must be "just right" for all dimensions at once to avoid excessively slow mixing. An alternative approach that often works better in such situations, known as Gibbs sampling, involves choosing a new sample for each dimension separately from the others, rather than choosing a sample for all dimensions at once. That way, the problem of sampling from potentially high-dimensional space will be reduced to a collection of problems to sample from small dimensionality.[5] This is especially applicable when the multivariate distribution is composed of a set of individual random variables in which each variable is conditioned on only a small number of other variables, as is the case in most typical hierarchical models. The individual variables are then sampled one at a time, with each variable conditioned on the most recent values of all the others. Various algorithms can be used to choose these individual samples, depending on the exact form of the multivariate distribution: some possibilities are the adaptive rejection sampling methods, the adaptive rejection Metropolis sampling algorithm,[6] a simple one-dimensional Metropolis–Hastings step, or slice sampling.

Formal derivation

The purpose of the Metropolis–Hastings algorithm is to generate a collection of states according to a desired distribution

P(x)

. To accomplish this, the algorithm uses a Markov process, which asymptotically reaches a unique stationary distribution

\pi(x)

such that

\pi(x)=P(x)

.

A Markov process is uniquely defined by its transition probabilities

P(x'\midx)

, the probability of transitioning from any given state

x

to any other given state

x'

. It has a unique stationary distribution

\pi(x)

when the following two conditions are met:
  1. Existence of stationary distribution: there must exist a stationary distribution

\pi(x)

. A sufficient but not necessary condition is detailed balance, which requires that each transition

x\tox'

is reversible: for every pair of states

x,x'

, the probability of being in state

x

and transitioning to state

x'

must be equal to the probability of being in state

x'

and transitioning to state

x

,

\pi(x)P(x'\midx)=\pi(x')P(x\midx')

.
  1. Uniqueness of stationary distribution: the stationary distribution

\pi(x)

must be unique. This is guaranteed by ergodicity of the Markov process, which requires that every state must (1) be aperiodic—the system does not return to the same state at fixed intervals; and (2) be positive recurrent—the expected number of steps for returning to the same state is finite.

The Metropolis–Hastings algorithm involves designing a Markov process (by constructing transition probabilities) that fulfills the two above conditions, such that its stationary distribution

\pi(x)

is chosen to be

P(x)

. The derivation of the algorithm starts with the condition of detailed balance:

P(x'\midx)P(x)=P(x\midx')P(x'),

which is re-written as

P(x'\midx)
P(x\midx')

=

P(x')
P(x)

.

The approach is to separate the transition in two sub-steps; the proposal and the acceptance-rejection. The proposal distribution

g(x'\midx)

is the conditional probability of proposing a state

x'

given

x

, and the acceptance distribution

A(x',x)

is the probability to accept the proposed state

x'

. The transition probability can be written as the product of them:

P(x'\midx)=g(x'\midx)A(x',x).

Inserting this relation in the previous equation, we have

A(x',x)
A(x,x')

=

P(x')
P(x)
g(x\midx')
g(x'\midx)

.

The next step in the derivation is to choose an acceptance ratio that fulfills the condition above. One common choice is the Metropolis choice:

A(x',x)=min\left(1,

P(x')
P(x)
g(x\midx')
g(x'\midx)

\right).

For this Metropolis acceptance ratio

A

, either

A(x',x)=1

or

A(x,x')=1

and, either way, the condition is satisfied.

The Metropolis–Hastings algorithm can thus be written as follows:

  1. Initialise
    1. Pick an initial state

x0

.
    1. Set

t=0

.
  1. Iterate
    1. Generate a random candidate state

x'

according to

g(x'\midxt)

.
    1. Calculate the acceptance probability

A(x',xt)=min\left(1,

P(x')
P(xt)
g(xt\midx')
g(x'\midxt)

\right)

.
    1. Accept or reject:
      1. generate a uniform random number

u\in[0,1]

;
      1. if

u\leA(x',xt)

, then accept the new state and set

xt+1=x'

;
      1. if

u>A(x',xt)

, then reject the new state, and copy the old state forward

xt+1=xt

.
    1. Increment: set

t=t+1

.

Provided that specified conditions are met, the empirical distribution of saved states

x0,\ldots,xT

will approach

P(x)

. The number of iterations (

T

) required to effectively estimate

P(x)

depends on the number of factors, including the relationship between

P(x)

and the proposal distribution and the desired accuracy of estimation.[7] For distribution on discrete state spaces, it has to be of the order of the autocorrelation time of the Markov process.

It is important to notice that it is not clear, in a general problem, which distribution

g(x'\midx)

one should use or the number of iterations necessary for proper estimation; both are free parameters of the method, which must be adjusted to the particular problem in hand.

Use in numerical integration

See main article: Monte Carlo integration.

A common use of Metropolis–Hastings algorithm is to compute an integral. Specifically, consider a space

\Omega\subsetR

and a probability distribution

P(x)

over

\Omega

,

x\in\Omega

. Metropolis–Hastings can estimate an integral of the form of

P(E)=\int\OmegaA(x)P(x)dx,

where

A(x)

is a (measurable) function of interest.

E(x)

and its probability distribution

P(E)

, which is a marginal distribution. Suppose that the goal is to estimate

P(E)

for

E

on the tail of

P(E)

. Formally,

P(E)

can be written as

P(E)=\int\OmegaP(E\midx)P(x)dx=\int\Omega\delta(E-E(x))P(x)dx=E(P(E\midX))

and, thus, estimating

P(E)

can be accomplished by estimating the expected value of the indicator function

AE(x)\equiv1E(x)

, which is 1 when

E(x)\in[E,E+\DeltaE]

and zero otherwise.Because

E

is on the tail of

P(E)

, the probability to draw a state

x

with

E(x)

on the tail of

P(E)

is proportional to

P(E)

, which is small by definition. The Metropolis–Hastings algorithm can be used here to sample (rare) states more likely and thus increase the number of samples used to estimate

P(E)

on the tails. This can be done e.g. by using a sampling distribution

\pi(x)

to favor those states (e.g.

\pi(x)\proptoea

with

a>0

).

Step-by-step instructions

Suppose that the most recent value sampled is

xt

. To follow the Metropolis–Hastings algorithm, we next draw a new proposal state

x'

with probability density

g(x'\midxt)

and calculate a value

a=a1a2,

where

a1=

P(x')
P(xt)

is the probability (e.g., Bayesian posterior) ratio between the proposed sample

x'

and the previous sample

xt

, and

a2=

g(xt\midx')
g(x'\midxt)

is the ratio of the proposal density in two directions (from

xt

to

x'

and conversely).This is equal to 1 if the proposal density is symmetric.Then the new state

xt+1

is chosen according to the following rules.

If

a\geq1{:}

xt+1=x',

else:

xt+1= \begin{cases} x'&withprobabilitya,\\ xt&withprobability1-a. \end{cases}

The Markov chain is started from an arbitrary initial value

x0

, and the algorithm is run for many iterations until this initial state is "forgotten". These samples, which are discarded, are known as burn-in. The remaining set of accepted values of

x

represent a sample from the distribution

P(x)

.

The algorithm works best if the proposal density matches the shape of the target distribution

P(x)

, from which direct sampling is difficult, that is

g(x'\midxt)P(x')

.If a Gaussian proposal density

g

is used, the variance parameter

\sigma2

has to be tuned during the burn-in period.This is usually done by calculating the acceptance rate, which is the fraction of proposed samples that is accepted in a window of the last

N

samples.The desired acceptance rate depends on the target distribution, however it has been shown theoretically that the ideal acceptance rate for a one-dimensional Gaussian distribution is about 50%, decreasing to about 23% for an

N

-dimensional Gaussian target distribution. These guidelines can work well when sampling from sufficiently regular Bayesian posteriors as they often follow a multivariate normal distribution as can be established using the Bernstein–von Mises theorem.[8]

If

\sigma2

is too small, the chain will mix slowly (i.e., the acceptance rate will be high, but successive samples will move around the space slowly, and the chain will converge only slowly to

P(x)

). On the other hand,if

\sigma2

is too large, the acceptance rate will be very low because the proposals are likely to land in regions of much lower probability density, so

a1

will be very small, and again the chain will converge very slowly. One typically tunes the proposal distribution so that the algorithms accepts on the order of 30% of all samples – in line with the theoretical estimates mentioned in the previous paragraph.

Bayesian Inference

See main article: article and Bayesian Inference. MCMC can be used to draw samples from the posterior distribution of a statistical model.The acceptance probability is given by:

Pacc(\thetai\to\theta*)=min\left(1,

l{L
(y|\theta

*)P(\theta

*)}{l{L}(y|\theta
i)P(\theta
i)}
*)
Q(\theta
i|\theta
*|\theta
Q(\theta
i)

\right),

where

l{L}

is the likelihood,

P(\theta)

the prior probability density and

Q

the (conditional) proposal probability.

See also

Further reading

Notes and References

  1. Book: Kalos . Malvin H. . Monte Carlo Methods Volume I: Basics . Whitlock . Paula A. . Wiley . 1986 . New York . 78–88.
  2. Tierney . Luke . 1994 . Markov chains for exploring posterior distributions . The Annals of Statistics . 22 . 4 . 1701–1762. 10.1214/aos/1176325750 .
  3. Gilks . W. R. . Wild . P. . 1992-01-01 . Adaptive Rejection Sampling for Gibbs Sampling . Journal of the Royal Statistical Society. Series C (Applied Statistics) . 41 . 2 . 337–348 . 10.2307/2347565 . 2347565.
  4. Book: Bayesian data analysis . 2004 . Chapman & Hall / CRC . Gelman, Andrew . 978-1584883883 . 2nd . Boca Raton, Fla. . 51991499.
  5. Lee . Se Yoon . 2021 . Gibbs sampler and coordinate ascent variational inference: A set-theoretical review . Communications in Statistics - Theory and Methods . 51 . 6 . 1549–1568 . 2008.01006 . 10.1080/03610926.2021.1921214 . 220935477.
  6. Gilks . W. R. . Best . N. G. . Nicky Best . Tan . K. K. C. . 1995-01-01 . Adaptive Rejection Metropolis Sampling within Gibbs Sampling . Journal of the Royal Statistical Society. Series C (Applied Statistics) . 44 . 4 . 455–472 . 10.2307/2986138 . 2986138.
  7. Raftery, Adrian E., and Steven Lewis. "How Many Iterations in the Gibbs Sampler?" In Bayesian Statistics 4. 1992.
  8. Schmon . Sebastian M. . Gagnon . Philippe . 2022-04-15 . Optimal scaling of random walk Metropolis algorithms using Bayesian large-sample asymptotics . Statistics and Computing . en . 32 . 2 . 28 . 10.1007/s11222-022-10080-8 . 0960-3174 . 8924149 . 35310543.