B0777
Title: BAMDT: Bayesian additive semi-multivariate decision trees for spatial nonparametric regression
Authors: Zhao Tang Luo - Apple (United States)
Huiyan Sang - Texas A\&M University (United States) [presenting]
Bani Mallick - Texas A&M University (United States)
Abstract: Bayesian additive regression trees have gained great popularity as flexible nonparametric function estimation and modeling tools. Nearly all existing BART models rely on decision tree-weak learners with axis-parallel univariate split rules to partition the Euclidean feature space into rectangular regions. In practice, however, many regression problems involve features with multivariate structures (e.g., spatial locations) possibly lying in a manifold, where rectangular partitions may fail to respect irregular intrinsic geometry and boundary constraints of the structured feature space. We develop a new class of Bayesian additive multivariate decision tree models that combine univariate split rules for handling possibly high dimensional features without known multivariate structures and novel multivariate split rules for features with multivariate structures in each weak learner. The proposed multivariate split rules are built upon stochastic predictive spanning tree bipartition models on reference knots, which are capable of achieving highly flexible nonlinear decision boundaries on manifold feature spaces while enabling efficient dimension reduction computations. We demonstrate the superior performance of the proposed method using simulation data and a Sacramento housing price data set.
