Gaussian Process Boosting
About
We introduce a novel way to combine boosting with Gaussian process and mixed effects models. This allows for relaxing, first, the zero or linearity assumption for the prior mean function in Gaussian process and grouped random effects models in a flexible non-parametric way and, second, the independence assumption made in most boosting algorithms. The former is advantageous for prediction accuracy and for avoiding model misspecifications. The latter is important for efficient learning of the fixed effects predictor function and for obtaining probabilistic predictions. Our proposed algorithm is also a novel solution for handling high-cardinality categorical variables in tree-boosting. In addition, we present an extension that scales to large data using a Vecchia approximation for the Gaussian process model relying on novel results for covariance parameter inference. We obtain increased prediction accuracy compared to existing approaches on multiple simulated and real-world data sets.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Wage Prediction | PSID wage data 1981–1982 1976–1982 (test) | Test MSE (Fixed Only)0.107 | 12 | |
| ITE estimation | Experiment C Cluster constant (test) | MSE0.011 | 8 | |
| ITE estimation | Experiment A Cluster constant (test) | MSE0.054 | 8 | |
| ITE estimation | Experiment B Cluster constant (test) | MSE0.015 | 8 | |
| ITE estimation | Experiment C Observation-level (test) | MSE0.015 | 8 | |
| ITE estimation | Experiment A Observation-level (test) | MSE0.175 | 8 | |
| ITE estimation | Experiment B Observation-level (test) | MSE0.014 | 8 | |
| Response Prediction | Experiment B Observation-level covariates | MSE0.523 | 7 | |
| Response Prediction | Experiment B Cluster constant covariates | MSE0.518 | 7 | |
| Response Prediction | Experiment C Observation-level covariates | MSE0.605 | 7 |