The Art of Data Augmentation

TLDR

Data augmentation constructs iterative optimization or sampling algorithms by introducing unobserved data or latent variables, a technique popularized by EM, Tanner–Wong, and Swendsen–Wang, and whose effectiveness depends on model‑specific strategies and often yields positive recurrent subchains of otherwise nonpositive recurrent Markov chains. The authors propose a search strategy that blends marginal and conditional augmentation with deterministic approximations to identify efficient data augmentation schemes. They apply the strategy to multivariate t, probit regression, and mixed‑effects models, producing efficient Markov chain Monte Carlo algorithms for posterior sampling. The resulting algorithms, requiring comparable programming effort, show dramatic improvements over conventional Gibbs samplers for these models.

Abstract

The term data augmentation refers to methods for constructing iterative optimization or sampling algorithms via the introduction of unobserved data or latent variables. For deterministic algorithms, the method was popularized in the general statistical community by the seminal article by Dempster, Laird, and Rubin on the EM algorithm for maximizing a likelihood function or, more generally, a posterior density. For stochastic algorithms, the method was popularized in the statistical literature by Tanner and Wong's Data Augmentation algorithm for posterior sampling and in the physics literature by Swendsen and Wang's algorithm for sampling from the Ising and Potts models and their generalizations; in the physics literature, the method of data augmentation is referred to as the method of auxiliary variables. Data augmentation schemes were used by Tanner and Wong to make simulation feasible and simple, while auxiliary variables were adopted by Swendsen and Wang to improve the speed of iterative simulation. In general, however, constructing data augmentation schemes that result in both simple and fast algorithms is a matter of art in that successful strategies vary greatly with the (observed-data) models being considered. After an overview of data augmentation/auxiliary variables and some recent developments in methods for constructing such efficient data augmentation schemes, we introduce an effective search strategy that combines the ideas of marginal augmentation and conditional augmentation, together with a deterministic approximation method for selecting good augmentation schemes. We then apply this strategy to three common classes of models (specifically, multivariate t, probit regression, and mixed-effects models) to obtain efficient Markov chain Monte Carlo algorithms for posterior sampling. We provide theoretical and empirical evidence that the resulting algorithms, while requiring similar programming effort, can show dramatic improvement over the Gibbs samplers commonly used for these models in practice. A key feature of all these new algorithms is that they are positive recurrent subchains of nonpositive recurrent Markov chains constructed in larger spaces.