Streamlined Variational Inference for Linear Mixed Models with Crossed Random Effects

J Comput Graph Stat. 2023;32(1):99-115. doi: 10.1080/10618600.2022.2096622. Epub 2022 Sep 1.

Abstract

We derive streamlined mean field variational Bayes algorithms for fitting linear mixed models with crossed random effects. In the most general situation, where the dimensions of the crossed groups are arbitrarily large, streamlining is hindered by lack of sparseness in the underlying least squares system. Because of this fact we also consider a hierarchy of relaxations of the mean field product restriction. The least stringent product restriction delivers a high degree of inferential accuracy. However, this accuracy must be mitigated against its higher storage and computing demands. Faster sparse storage and computing alternatives are also provided, but come with the price of diminished inferential accuracy. This article provides full algorithmic details of three variational inference strategies, presents detailed empirical results on their pros and cons and, thus, guides the users on their choice of variational inference approach depending on the problem size and computing resources.

Keywords: Mean field variational Bayes; Rasch analysis; item response theory; scalable statistical methodology; sparse least squares systems.