For high-dimensional hierarchical models, consider exchangeability of effects across covariates instead of across datasets
Brian L. Trippe, Hilary K. Finucane, Tamara Broderick
Abstract
Hierarchical Bayesian methods enable information sharing across multiple related regression problems. While standard practice is to model regression parameters (effects) as (1) exchangeable across datasets and (2) correlated to differing degrees across covariates, we show that this approach exhibits poor statistical performance when the number of covariates exceeds the number of datasets. For instance, in statistical genetics, we might regress dozens of traits (defining datasets) for thousands of individuals (responses) on up to millions of genetic variants (covariates). When an analyst has more covariates than datasets, we argue that it is often more natural to instead model effects as (1) exchangeable across covariates and (2) correlated to differing degrees across datasets. To this end, we propose a hierarchical model expressing our alternative perspective. We devise an empirical Bayes estimator for learning the degree of correlation between datasets. We develop theory that demonstrates that our method outperforms the classic approach when the number of covariates dominates the number of datasets, and corroborate this result empirically on several high-dimensional multiple regression and classification problems.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers1
Ask how each one uses itRelated papers
- Deep Hierarchy in BanditsJoey Hong, Branislav Kveton, Sumeet Katariya, Manzil Zaheer et al.ICML 2022 · 21 citations
- Amortized Variational Inference for Simple Hierarchical ModelsAbhinav Agrawal, Justin DomkeNeurIPS 2021 · 32 citations
- Double and Single Descent in Causal Inference with an Application to High-Dimensional Synthetic ControlJann Spiess, Guido Imbens, Amar VenugopalNeurIPS 2023 · 10 citations
- Addressing Instrument-Outcome Confounding in Mendelian Randomization through Representation LearningShimeng Huang, Matthew Robinson, Francesco LocatelloICML 2026
- Differentially Private Distributed Bayesian Linear Regression with MCMCBaris Alparslan, Sinan Yildirim, S. Ilker BirbilICML 2023 · 1 citation
