References & Citations
Statistics > Other Statistics
Title: mBART: Multidimensional Monotone BART
(Submitted on 6 Dec 2016 (v1), last revised 8 Oct 2021 (this version, v3))
Abstract: For the discovery of regression relationships between Y and a large set of p potential predictors x 1 , . . . , x p , the flexible nonparametric nature of BART (Bayesian Additive Regression Trees) allows for a much richer set of possibilities than restrictive parametric approaches. However, subject matter considerations sometimes warrant a minimal assumption of monotonicity in at least some of the predictors. For such contexts, we introduce mBART, a constrained version of BART that can flexibly incorporate monotonicity in any predesignated subset of predictors using a multivariate basis of monotone trees, while avoiding the further confines of a full parametric form. For such monotone relationships, mBART provides (i) function estimates that are smoother and more interpretable, (ii) better out-of-sample predictive performance, and (iii) less post-data uncertainty. While many key aspects of the unconstrained BART model carry over directly to mBART, the introduction of monotonicity constraints necessitates a fundamental rethinking of how the model is implemented. In particular, the original BART Markov Chain Monte Carlo algorithm relied on a conditional conjugacy that is no longer available in a monotonically constrained space. Various simulated and real examples demonstrate the wide ranging potential of mBART.
Submission history
From: Robert McCulloch [view email][v1] Tue, 6 Dec 2016 01:24:34 GMT (587kb,D)
[v2] Sun, 18 Aug 2019 15:03:18 GMT (3142kb,D)
[v3] Fri, 8 Oct 2021 13:18:55 GMT (3362kb,D)
Link back to: arXiv, form interface, contact.