Sparse Linear Mixed Model Selection via Streamlined Variational Bayes

TL;DR

This paper introduces a streamlined variational Bayes approach for efficient and accurate fixed effects selection in high-dimensional linear mixed models, significantly reducing computational costs while maintaining inference quality.

Contribution

It generalizes recent streamlined variational inference methods to linear mixed models with nested random effects and global-local priors, enabling fast Bayesian fixed effects selection.

Findings

Achieves convergence similar to standard methods with lower computational effort.

Provides automated fixed effects selection without hyperparameter tuning.

Demonstrates high accuracy of variational approximations compared to MCMC.

Abstract

Linear mixed models are a versatile statistical tool to study data by accounting for fixed effects and random effects from multiple sources of variability. In many situations, a large number of candidate fixed effects is available and it is of interest to select a parsimonious subset of those being effectively relevant for predicting the response variable. Variational approximations facilitate fast approximate Bayesian inference for the parameters of a variety of statistical models, including linear mixed models. However, for models having a high number of fixed or random effects, simple application of standard variational inference principles does not lead to fast approximate inference algorithms, due to the size of model design matrices and inefficient treatment of sparse matrix problems arising from the required approximating density parameters updates. We illustrate how recently developed streamlined variational inference procedures can be generalized to make fast and accurate inference for the parameters of linear mixed models with nested random effects and global-local priors for Bayesian fixed effects selection. Our variational inference algorithms achieve convergence to the same optima of their standard implementations, although with significantly lower computational effort, memory usage and time, especially for large numbers of random effects. Using simulated and real data examples, we assess the quality of automated procedures for fixed effects selection that are free from hyperparameters tuning and only rely upon variational posterior approximations. Moreover, we show high accuracy of variational approximations against model fitting via Markov Chain Monte Carlo sampling.