Variable Selection with ABC Bayesian Forests

TL;DR

This paper introduces a Bayesian tree-based variable selection method that is consistent in high-dimensional non-linear models, and proposes an ABC sampling approach to efficiently identify important variables.

Contribution

It develops the first model selection consistency results for Bayesian forest priors and introduces ABC Bayesian Forests, a novel data-splitting ABC method for non-parametric variable selection.

Findings

Method is consistent for variable selection with p > n.

ABC Bayesian Forests achieve higher acceptance rates.

Successfully identifies variables with high marginal inclusion probabilities.

Abstract

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 MCMC, we propose 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.