Sharded Bayesian Additive Regression Trees
Not on the map. Why?
Abstract (arXiv)
In this paper we develop the randomized Sharded Bayesian Additive Regression Trees (SBT) model. We introduce a randomization auxiliary variable and a sharding tree to decide partitioning of data, and fit each partition component to a sub-model using Bayesian Additive Regression Tree (BART). By observing that the optimal design of a sharding tree can determine optimal sharding for sub-models on a product space, we introduce an intersection tree structure to completely specify both the sharding and modeling using only tree structures. In addition to experiments, we also derive the theoretical optimal weights for minimizing posterior contractions and prove the worst-case complexity of SBT.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 15 of its 52 references that are in the map. First posted on arXiv 1 Jun 2023.
Builds on
15 of 52 references.
References · 15
- 2021Interpretable Machine Learning: Fundamental Principles and 10 Grand Challengescited by 367
- 2016Coresets for Scalable Bayesian Logistic Regressioncited by 79
- 2017A general framework for Vecchia approximations of Gaussian processescited by 67
- 2023Randomized Numerical Linear Algebra : A Perspective on the Field With an Eye to Softwarecited by 38
- 2013bartMachine: Machine Learning with Bayesian Additive Regression Treescited by 24
- 2018Distributed linear regression by averagingcited by 21
- 2018On Theory for BARTcited by 17
- 2019Bayesian experimental design using regularized determinantal point processescited by 17
- 2019Optimizing Ensemble Weights and Hyperparameters of Machine Learning Models for Regression Problemscited by 15
- 2013Parallelizing MCMC via Weierstrass Samplercited by 11
- 2021Non-smooth Bayesian Optimization in Tuning Problemscited by 8
- 2020A Distance-preserving Matrix Sketchcited by 4
Built on by
Citing papers per year: 2023: 1; 2024: 3; 2025: 0; 2026: 2.
6 papers.
- 2023A Unifying Perspective on Non-Stationary Kernels for Deeper Gaussian Processescited by 8
- 2024Compactly-supported nonstationary kernels for computing exact Gaussian processes on big datacited by 2
- 2024Efficient Decision Trees for Tensor Regressionscited by 1
- 2024Ranking Perspective for Tree-based Methods with Applications to Symbolic Feature Selectioncited by 1
- 2026GGMPs: Generalized Gaussian Mixture Processescited by 1
- 2026Asymptotic Optimism for Tensor Regression Models with Applications to Neural Network Compressioncited by 0
- 2026Asymptotic Optimism for Tensor Regression Models with Applications to Neural Network Compression27 Mar 2026
- 2026GGMPs: Generalized Gaussian Mixture Processes11 Mar 2026
- 2024Compactly-supported nonstationary kernels for computing exact Gaussian processes on big data7 Nov 2024
- 2024Ranking Perspective for Tree-based Methods with Applications to Symbolic Feature Selection3 Oct 2024
- 2024Efficient Decision Trees for Tensor Regressions4 Aug 2024
- 2023A Unifying Perspective on Non-Stationary Kernels for Deeper Gaussian Processes18 Sep 2023