Overfitted high-dimensional matrix factorizations via adaptive spectral shrinkage
Lorenzo Mauri, David B. Dunson
stat.ME, stat.CO, stat.ML
Submitted: 2026-06-17
License: http://creativecommons.org/licenses/by/4.0/
The gist: Factor models are popular approaches for analyzing high-dimensional data to extract low-rank signals and estimate covariances.
Terminology
Abstract
Factor models are popular approaches for analyzing high-dimensional data to extract low-rank signals and estimate covariances. They decompose the covariance matrix as the sum of low-rank and diagonal components. A key issue is how to choose the latent dimension k, which is particularly challenging when the factor model only holds approximately and in low signal-to-noise scenarios. Bayesian overfitted factor models specify an upper bound on k and rely on structured shrinkage priors to effectively remove extra components. Such approaches are popular and effective, but computationally expensive. We propose a much faster EigenBayes approach that provides valid uncertainty quantification, based on spectral estimation of latent factors and adaptive empirical Bayes calibration of key hyperparameters. The resulting posterior distribution factorizes across outcomes and is analytically tractable, bypassing Markov chain Monte Carlo. We show that EigenBayes adapts to the signal-to-noise ratio of each outcome and latent dimension, while shrinking superfluous latent components to zero. We establish favorable asymptotic properties and demonstrate strong empirical performance in numerical experiments and a genomics application, where EigenBayes outperforms state-of-the-art alternatives.
Sources
- Simpler Proofs for Approximate Factor Models of Large Dimensions
- Blessing of dimension in Bayesian inference on covariance matrices
- Spectral decomposition-assisted multi-study factor analysis
- Inference on covariance structure in high-dimensional multi-view data
- Pathway-based Bayesian factor models for 'omics data
- SANVI: A Fast Spectral-Assisted Network Variational Inference Method with an Extended Surrogate Likelihood Function
- Compressive Bayesian non-negative matrix factorization for mutational signatures analysis
- Multivariate Causal Effects: a Bayesian Causal Regression Factor Model
Related papers
- Doubly robust inference via calibration
- Bayesian Empirical Bayes: Simultaneous Inference from Probabilistic Symmetries
- Optimal Watermark Localization in Mixed-Source Large Language Model Texts
- Flexible Nonparametric Inference for Causal Effects under the Front-Door Model
- A Greedy PDE Router for Blending Neural Operators and Classical Methods
- Policy Learning with Weak Signals