Bayesian Additive Regression Trees with Model Trees

Noname manuscript No. (will be inserted by the editor) Abstract Bayesian Additive Regression Trees (BART) 1 Introduction is a tree-based machine learning method that has been successfully applied to regression and classification problems. Bayesian Additive Regression Trees (BART) is a statistical BART assumes regularisation priors on a set of method proposed by Chipman et al (2010) that has trees that work as weak learners and is very flexible for become popular in recent years due to its competitive predicting in the presence of non-linearity and highorder performance on regression and classification problems, interactions. In this paper, we introduce an extension when compared to other supervised machine learning of BART, called Model Trees BART (MOTR-methods, such as Random Forests (RF) (Breiman, 2001) BART), that considers piecewise linear functions at node and Gradient Boosting (GB) (Friedman, 2001). In MOTR-BART, differs from other tree-based methods as it controls the rather than having a unique value at node level for the structure of each tree via a prior distribution and generates prediction, a linear predictor is estimated considering the predictions via an MCMC backfitting algorithm the covariates that have been used as the split variables that is responsible for accepting and rejecting the in the corresponding tree. In our approach, local linearities proposed trees along the iterations.