RoBART: Bayesian Additive Regression Trees with Tree-Specific Rotations

Jeongung Heo, Seonghyun Jeong

Bayesian additive regression trees (BART) can require many splits to approximate boundaries misaligned with the predictor axes. RoBART assigns each tree a rotation shared by all in…