Tuning an MCMC sampler

April 30, 2020 β€” April 30, 2020

Figure 1: The process of adapting to the target optimally

Designing MCMC transition density, possibly via the proposal density in rejection sampling, by optimisation for optimal mixing.

The simplest way to do this is to do a β€œpilot” run to estimate optimal mixing kernels then use the adapted mixing kernels, discarding the suspect samples from the pilot run as suspect. This wastes some effort but is theoretically simple. Alternatively you could do this dynamically, online, which is called Adaptive MCMC. There are then some theoretical wrinkles.

I do wish to maximise the mixing rate by some criterion. If I already know my mixing rate is bad without optimising, which is why I am optimising, how do I get the simulations against which to conduct the optimisation? How do we optimise simultaneously for maximising mixing rate and minimising rejection rate? Fearnhead and Taylor (2013) summarise some options here for an objective function. One that seems sufficient for publication of typical MCMC papers is Expected Squared Jump Distance, ESJD (which is more precisely an expected squared Mahalanobis distance) between samples, which minimises the lag-1 autocorrelation, which is in practice most of what we do.

1 Proposal density

Designing the proposal density is often easy for an independent rejection sampler. That is precisely the cross-entropy method. For Markov chain, though the success criterion is muddier. AFAICT the cross entropy trick does not apply for non-i.i.d. samples.

2 Transition density

πŸ—

3 Adaptive SMC

In Sequential Monte Carlo, which is not MCMC, we do not need to be so sensitive to changing the proposal parameters, since there is no stationary distribution argument. See Fearnhead and Taylor (2013).

4 Variational inference

What is Hamiltonian Variational Inference? Does that fit under this heading? πŸ— (Caterini, Doucet, and Sejdinovic 2018; Salimans, Kingma, and Welling 2015)

5 References

Caterini, Doucet, and Sejdinovic. 2018. β€œHamiltonian Variational Auto-Encoder.” In Advances in Neural Information Processing Systems.
Fearnhead, and Taylor. 2013. β€œAn Adaptive Sequential Monte Carlo Sampler.” Bayesian Analysis.
Mathew, Bauer, Koistinen, et al. 2012. β€œBayesian Adaptive Markov Chain Monte Carlo Estimation of Genetic Parameters.” Heredity.
Norton, and Fox. 2016. β€œTuning of MCMC with Langevin, Hamiltonian, and Other Stochastic Autoregressive Proposals.” arXiv:1610.00781 [Math, Stat].
Roberts, and Rosenthal. 2009. β€œExamples of Adaptive MCMC.” Journal of Computational and Graphical Statistics.
β€”β€”β€”. 2014. β€œMinimising MCMC Variance via Diffusion Limits, with an Application to Simulated Tempering.” Annals of Applied Probability.
Salimans, Kingma, and Welling. 2015. β€œMarkov Chain Monte Carlo and Variational Inference: Bridging the Gap.” In Proceedings of the 32nd International Conference on Machine Learning (ICML-15). ICML’15.
Schuster, Strathmann, Paige, et al. 2017. β€œKernel Sequential Monte Carlo.” In ECML-PKDD 2017.