Skip to main navigation Skip to search Skip to main content

XBART: Accelerated Bayesian Additive Regression Trees

Research output: Chapters, Conference Papers, Creative and Literary WorksRGC 32 - Refereed conference paper (with host publication)peer-review

Abstract

Bayesian additive regression trees (BART) (Chipman et. al., 2010) is a powerful predictive model that often outperforms alternative models at out-of-sample prediction. BART is especially well-suited to settings with unstructured predictor variables and substantial sources of unmeasured variation as is typical in the social, behavioral and health sciences. This paper develops a modified version of BART that is amenable to fast posterior estimation. We present a stochastic hill climbing algorithm that matches the remarkable predictive accuracy of previous BART implementations, but is many times faster and less memory intensive. Simulation studies show that the new method is comparable in computation time and more accurate at function estimation than both random forests and gradient boosting. © 2019 by the author(s).
Original languageEnglish
Title of host publicationThe 22nd International Conference on Artificial Intelligence and Statistics
EditorsKamalika Chaudhuri, Masashi Sugiyama
PublisherPMLR
Pages1130-1138
Number of pages9
Publication statusPublished - Apr 2019
Externally publishedYes
Event22nd International Conference on Artificial Intelligence and Statistics (AISTATS 2019) - Naha, Japan
Duration: 16 Apr 201918 Apr 2019
https://www.aistats.org/aistats2019/

Publication series

NameAISTATS - International Conference on Artificial Intelligence and Statistics
NameProceedings of Machine Learning Research
Volume89
ISSN (Print)2640-3498

Conference

Conference22nd International Conference on Artificial Intelligence and Statistics (AISTATS 2019)
Abbreviated titleAISTATS2019
PlaceJapan
CityNaha
Period16/04/1918/04/19
Internet address

Fingerprint

Dive into the research topics of 'XBART: Accelerated Bayesian Additive Regression Trees'. Together they form a unique fingerprint.

Cite this