Heterogeneous large datasets integration using Bayesian factor regression

Avalos-Pacheco, Alejandra; Rossell Ribera, David; Savage, Richard S.

doi:http://dx.doi.org/10.1214/20-BA1240

Heterogeneous large datasets integration using Bayesian factor regression

License

Citation

Avalos-Pacheco A, Rossell D, Savage RS. Heterogeneous large datasets integration using Bayesian factor regression. Bayesian Anal. 2022;17(1):33-66. DOI: 10.1214/20-BA1240

Abstract

Two key challenges in modern statistical applications are the large amount of information recorded per individual, and that such data are often not collected all at once but in batches. These batch effects can be complex, causing distortions in both mean and variance. We propose a novel sparse latent factor regression model to integrate such heterogeneous data. The model provides a tool for data exploration via dimensionality reduction and sparse low-rank covariance estimation while correcting for a range of batch effects. We study the use of several sparse priors (local and non-local) to learn the dimension of the latent factors. We provide a flexible methodology for sparse factor regression which is not limited to data with batch effects. Our model is fitted in a deterministic fashion by means of an EM algorithm for which we derive closed-form updates, contributing a novel scalable algorithm for non-local priors of interest beyond the immediate scope of this paper. We present several examples, with a focus on bioinformatics applications. Our results show an increase in the accuracy of the dimensionality reduction, with non-local priors substantially improving the reconstruction of factor cardinality. The results of our analyses illustrate how failing to properly account for batch effects can result in unreliable inference. Our model provides a novel approach to latent factor regression that balances sparsity with sensitivity in scenarios both with and without batch effects and is highly computationally efficient.