Scalable Bayesian computation for crossed and nested hierarchical models

Abstract

We develop sampling algorithms to fit Bayesian hierarchical models, the computational complexity of which scales linearly with the number of observations and the number of parameters in the model. We focus on crossed random effect and nested multilevel models, which are used ubiquitously in applied sciences. The posterior dependence in both classes is sparse: in crossed random effects models it resembles a random graph, whereas in nested multilevel models it is tree-structured. For each class we identify a framework for scalable computation, building on previous work. Methods for crossed models are based on extensions of appropriately designed collapsed Gibbs samplers, where we introduce the idea of local centering; while methods for nested models are based on sparse linear algebra and data augmentation. We provide a theoretical analysis of the proposed algorithms in some simplified settings, including a comparison with previously proposed methodologies and an average-case analysis based on random graph theory. Numerical experiments, including two challenging real data analyses on predicting electoral results and real estate prices, compare with off-the-shelf Hamiltonian Monte Carlo, displaying drastic improvement in performance.

References in corpus (10)

• A General Framework for the Parametrization of Hierarchical Models
• Log-concave sampling: Metropolis-Hastings algorithms are fast
• ${\ell }_p$-Spread and Restricted Isometry Properties of Sparse Random Matrices
• Efficient moment calculations for variance components in large unbalanced crossed random effects models
• Minimax Mixing Time of the Metropolis-Adjusted Langevin Algorithm for Log-Concave Sampling
• Fast and Accurate Estimation of Non-Nested Binomial Hierarchical Models Using Variational Inference
• Bayesian forecasting of electoral outcomes with new parties' competition
• A note on MCMC for nested multilevel regression models via belief propagation
• Hierarchical array priors for ANOVA decompositions of cross-classified data
• Estimation and Inference for Very Large Linear Mixed Effects Models

Cited by in corpus (2)

• Partially factorized variational inference for high-dimensional mixed models
• Bayesian Nonparametric Models for Multiple Raters: a General Statistical Framework