Nested $\hat R$: Assessing the convergence of Markov chain Monte Carlo when running many short chains
Abstract
Recent developments in parallel Markov chain Monte Carlo (MCMC) algorithms allow us to run thousands of chains almost as quickly as a single chain, using hardware accelerators such as GPUs. While each chain still needs to forget its initial point during a warmup phase, the subsequent sampling phase can be shorter than in classical settings, where we run only a few chains. To determine if the resulting short chains are reliable, we need to assess how close the Markov chains are to their stationary distribution after warmup. The potential scale reduction factor is a popular convergence diagnostic but unfortunately can require a long sampling phase to work well. We present a nested design to overcome this challenge and a generalization called nested . This new diagnostic works under conditions similar to and completes the workflow for GPU-friendly samplers. In addition, the proposed nesting provides theoretical insights into the utility of , in both classical and short-chains regimes.
Keywords
Cite
@article{arxiv.2110.13017,
title = {Nested $\hat R$: Assessing the convergence of Markov chain Monte Carlo when running many short chains},
author = {Charles C. Margossian and Matthew D. Hoffman and Pavel Sountsov and Lionel Riou-Durand and Aki Vehtari and Andrew Gelman},
journal= {arXiv preprint arXiv:2110.13017},
year = {2024}
}