Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Fabio Sigrist• 2020

Related benchmarks

TaskDatasetResultRank
Wage PredictionPSID wage data 1981–1982 1976–1982 (test)
Test MSE (Fixed Only)0.107
12
ITE estimationExperiment C Cluster constant (test)
MSE0.011
8
ITE estimationExperiment A Cluster constant (test)
MSE0.054
8
ITE estimationExperiment B Cluster constant (test)
MSE0.015
8
ITE estimationExperiment C Observation-level (test)
MSE0.015
8
ITE estimationExperiment A Observation-level (test)
MSE0.175
8
ITE estimationExperiment B Observation-level (test)
MSE0.014
8
Response PredictionExperiment B Observation-level covariates
MSE0.523
7
Response PredictionExperiment B Cluster constant covariates
MSE0.518
7
Response PredictionExperiment C Observation-level covariates
MSE0.605
7
Showing 10 of 20 rows

Other info

Follow for update