Journal of Machine Learning Research 17 (2016) 1-47 Submitted 11/14; Revised 11/15; Published 4/16 Bayesian group factor analysis with structured sparsity Shiwen Zhao
[email protected] Computational Biology and Bioinformatics Program Department of Statistical Science Duke University Durham, NC 27708, USA Chuan Gao
[email protected] Department of Statistical Science Duke University Durham, NC 27708, USA Sayan Mukherjee
[email protected] Departments of Statistical Science, Computer Science, Mathematics Duke University Durham, NC 27708, USA Barbara E Engelhardt
[email protected] Department of Computer Science Center for Statistics and Machine Learning Princeton University Princeton, NJ 08540, USA Editor: Samuel Kaski Abstract Latent factor models are the canonical statistical tool for exploratory analyses of low- dimensional linear structure for a matrix of p features across n samples. We develop a structured Bayesian group factor analysis model that extends the factor model to multiple coupled observation matrices; in the case of two observations, this reduces to a Bayesian model of canonical correlation analysis. Here, we carefully define a structured Bayesian prior that encourages both element-wise and column-wise shrinkage and leads to desirable behavior on high-dimensional data. In particular, our model puts a structured prior on the joint factor loading matrix, regularizing at three levels, which enables element-wise sparsity and unsupervised recovery of latent factors corresponding to structured variance across arbitrary subsets of the observations. In addition, our structured prior allows for both dense and sparse latent factors so that covariation among either all features or only a subset of features can be recovered.