arrow
Return

Variable selection with ABC Bayesian forests

delete2021-04-05
delete7
delete
OA
AI
L
Liu, Yi
R
Rockova, Veronika *
Y
Yuexi Wang
DOI:10.1111/rssb.12423delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Few problems in statistics are as perplexing as variable selection in the presence of very many redundant covariates. The variable selection problem is most familiar in parametric environments such as the linear model or additive variants thereof. In this work, we abandon the linear model framework, which can be quite detrimental when the covariates impact the outcome in a non-linear way, and turn to tree-based methods for variable selection. Such variable screening is traditionally done by pruning down large trees or by ranking variables based on some importance measure. Despite heavily used in practice, these ad hoc selection rules are not yet well understood from a theoretical point of view. In this work, we devise a Bayesian tree-based probabilistic method and show that it is consistent for variable selection when the regression surface is a smooth mix of p > n covariates. These results are the first model selection consistency results for Bayesian forest priors. Probabilistic assessment of variable importance is made feasible by a spike-and-slab wrapper around sum-of-trees priors. Sampling from posterior distributions over trees is inherently very difficult. As an alternative to Markov Chain Monte Carlo (MCMC), we propose approximate Bayesian computation (ABC) Bayesian forests, a new ABC sampling method based on data-splitting that achieves higher ABC acceptance rate. We show that the method is robust and successful at finding variables with high marginal inclusion probabilities. Our ABC algorithm provides a new avenue towards approximating the median probability model in non-parametric setups where the marginal likelihood is intractable.
Keywords:
approximate Bayesian computation
BART
consistency
spike‐ and‐ slab
variable selection

Journal

J
Journal of the Royal Statistical Society Series B-Statistical Methodology
IF:
3.6
Papers:
1.5K
Citations:
3.4W

Organization

U
university of chicago
Scholars:
2.2K
Papers: 853
Citations: 2
Cited Papers

Cited Papers

Dynamic Trees for Learning and Design
err2011-03-01
err76
errOAAI
errTaddy, Matthew A.; Gramacy, Robert B.; Polson, Nicholas G.
errShare
errSave
Effect of Rewetting on Smouldering Combustion of a Tropical Peat
err2018-11-26
err0
errOAAI
errAfiri Dianti; Nadhira Gilang Ratnasari; Pither Palamba; Yulianto Nugroho
errShare
errSave
Evolution of Ty1 copy number control in yeast by horizontal transfer and recombination
err2020-02-21
err0
errOAAI
errWioletta Czaja; Douda Bensasson; Hyo Won Ahn; David J. Garfinkel; Casey M. Bergman
errShare
errSave
Short-course radiation followed by mFOLFOX-6 plus avelumab for locally advanced rectal adenocarcinoma.
err2020-02-01
err0
PREAI
errAli Shamseddine; Youssef Zeidan; Ibrahim Moustafa Khalifeh; Joseph Gergi Kattan; Rim Turfa; Deborah Mukherji; Sally Naji Temraz; Faek Jamali; Yasser Hani Shaib; Assaad Soweid; Samer Deeba; Malek Kreidieh; Ziad zuheir El Husseini; Maya Charafeddine; Monita Al Darazi; Fady B. Geara
errShare
errSave
err
IF0
err
err0
PREAI
err
errShare
errSave
researcher View more