Hamiltonian Monte Carlo within Stan Daniel Lee Columbia University, Statistics Department
[email protected] BayesComp mc-stan.org 1 Why MCMC? • Have data. • Have a rich statistical model. • No analytic solution. • (Point estimate not adequate.) 2 Review: MCMC • Markov Chain Monte Carlo. The samples form a Markov Chain. • Markov property: Pr(θn+1 j θ1; : : : ; θn) = Pr(θn+1 j θn) • Invariant distribution: π × Pr = π • Detailed balance: sufficient condition: R Pr(θn+1; A) = A q(θn+1; θn)dy π(θn+1)q(θn+1; θn) = π(θn)q(θn; θn+1) 3 Review: RWMH • Want: samples from posterior distribution: Pr(θjx) • Need: some function proportional to joint model. f (x, θ) / Pr(x, θ) • Algorithm: Given f (x, θ), x, N, Pr(θn+1 j θn) For n = 1 to N do Sample θˆ ∼ q(θˆ j θn−1) f (x,θ)ˆ With probability α = min 1; , set θn θˆ, else θn θn−1 f (x,θn−1) 4 Review: Hamiltonian Dynamics • (Implicit: d = dimension) • q = position (d-vector) • p = momentum (d-vector) • U(q) = potential energy • K(p) = kinectic energy • Hamiltonian system: H(q; p) = U(q) + K(p) 5 Review: Hamiltonian Dynamics • for i = 1; : : : ; d dqi = @H dt @pi dpi = − @H dt @qi • kinectic energy usually defined as K(p) = pT M−1p=2 • for i = 1; : : : ; d dqi −1 dt = [M p]i dpi = − @U dt @qi 6 Connection to MCMC • q, position, is the vector of parameters • U(q), potential energy, is (proportional to) the minus the log probability density of the parameters • p, momentum, are augmented variables • K(p), kinectic energy, is calculated • Hamiltonian dynamics used to update q.