Preconditioned Crank–Nicolson algorithm explained

In computational statistics, the preconditioned Crank–Nicolson algorithm (pCN) is a Markov chain Monte Carlo (MCMC) method for obtaining random samples – sequences of random observations – from a target probability distribution for which direct sampling is difficult.

The most significant feature of the pCN algorithm is its dimension robustness, which makes it well-suited for high-dimensional sampling problems. The pCN algorithm is well-defined, with non-degenerate acceptance probability, even for target distributions on infinite-dimensional Hilbert spaces. As a consequence, when pCN is implemented on a real-world computer in large but finite dimension N, i.e. on an N-dimensional subspace of the original Hilbert space, the convergence properties (such as ergodicity) of the algorithm are independent of N. This is in strong contrast to schemes such as Gaussian random walk Metropolis–Hastings and the Metropolis-adjusted Langevin algorithm, whose acceptance probability degenerates to zero as N tends to infinity.

The algorithm as named was highlighted in 2013 by Cotter, Roberts, Stuart and White,[1] and its ergodicity properties were proved a year later by Hairer, Stuart and Vollmer.[2] In the specific context of sampling diffusion bridges, the method was introduced in 2008.[3]

Description of the algorithm

Overview

See also: Metropolis–Hastings algorithm.

The pCN algorithm generates a Markov chain

(Xn)n

on a Hilbert space

l{H}

whose invariant measure is a probability measure

\mu

of the form

\mu(E)=

1
Z

\intE\exp(-\Phi(x))\mu0(dx)

for each measurable set

E\subseteql{H}

, with normalising constant

Z

given by

Z=\intl{H

} \exp(- \Phi(x)) \, \mu_ (\mathrm x),where

\mu0=l{N}(0,C0)

is a Gaussian measure on

l{H}

with covariance operator

C0

and

\Phi\colonl{H}\toR

is some function. Thus, the pCN method applied to target probability measures that are re-weightings of a reference Gaussian measure.

The Metropolis–Hastings algorithm is a general class of methods that try to produce such Markov chains

(Xn)n

, and do so by a two-step procedure of first proposing a new state

X'n

given the current state

Xn

and then accepting or rejecting this proposal, according to a particular acceptance probability, to define the next state

Xn

. The idea of the pCN algorithm is that a clever choice of (non-symmetric) proposal for a new state

X'n

given

Xn

might have an associated acceptance probability function with very desirable properties.

The pCN proposal

The special form of this pCN proposal is to take

X'n=\sqrt{1-\beta2}Xn+\beta\Xin,

\Xin\sim\mu0i.i.d.

or, equivalently,

X'n|Xn\siml{N}\left(\sqrt{1-\beta2}Xn,\beta2C0\right).

The parameter

0<\beta<1

is a step size that can be chosen freely (and even optimised for statistical efficiency). One then generates

Zn\simUnif([0,1])

and sets

Xn=X'nifZn\leq\alpha(Xn,X'n),

Xn=XnifZn>\alpha(Xn,X'n).

The acceptance probability takes the simple form

\alpha(x,x')=min(1,\exp(\phi(x)-\phi(x'))).

It can be shown[2] that this method not only defines a Markov chain that satisfies detailed balance with respect to the target distribution

\mu

, and hence has

\mu

as an invariant measure, but also possesses a spectral gap that is independent of the dimension of

l{H}

, and so the law of

Xn

converges to

\mu

as

n\toinfty

. Thus, although one may still have to tune the step size parameter

\beta

to achieve a desired level of statistical efficiency, the performance of the pCN method is robust to the dimension of the sampling problem being considered.

Contrast with symmetric proposals

This behaviour of pCN is in stark contrast to the Gaussian random walk proposal

X'n\midXn\siml{N}\left(Xn,\beta\Gamma\right)

with any choice of proposal covariance

\Gamma

, or indeed any symmetric proposal mechanism. It can be shown using the Cameron–Martin theorem that for infinite-dimensional

l{H}

this proposal has acceptance probability zero for

\mu

-almost all

X'n+1

and

Xn

. In practice, when one implements the Gaussian random walk proposal in dimension

N

, this phenomenon can be seen in the way that

\beta

, the acceptance probability tends to zero as

N\toinfty

, and

\beta\to0

as

N\toinfty

.

References

  1. Cotter. S. L.. Roberts. G. O.. Stuart. A. M.. White. D.. MCMC methods for functions: modifying old algorithms to make them faster. Statist. Sci.. 28. 2013. 3. 424 - 446. 0883-4237. 10.1214/13-STS421. 1202.0709. 36562755.
  2. Hairer. M.. Stuart. A. M.. Vollmer. S. J.. Spectral gaps for a Metropolis - Hastings algorithm in infinite dimensions. Ann. Appl. Probab.. 24. 2014. 6. 2455 - 2490. 1050-5164. 10.1214/13-AAP982. 1112.1392. 73662504.
  3. Beskos. A.. Roberts. G. O.. Stuart. A. M.. Voss. J.. MCMC Methods for Diffusion Bridges. Stochastics and Dynamics. 8. 2008. 3. 319 - 350. 10.1142/S0219493708002378.