Hostname: page-component-74656f9884-rngvg Total loading time: 0 Render date: 2026-11-05T12:20:02.598Z Has data issue: false hasContentIssue false

Distill knowledge of additive tree models into generalized linear models: a new learning approach for non-smooth generalized additive models

Published online by Cambridge University Press:  14 November 2024

Arthur Maillart
Affiliation:
Detralytics, Saint-Josse-ten-Noode, Belgium Université Lyon 1, Institut de Science Financière et d’Assurances, Lyon, France
Christian Robert*
Affiliation:
Université Lyon 1, Institut de Science Financière et d’Assurances, Lyon, France Laboratory in Finance and Insurance - LFA CREST - Center for Research in Economics and Statistics, Paris, France
*
Corresponding author: Christian Robert; Email: christian.robert@univ-lyon1.fr
Rights & Permissions [Opens in a new window]

Abstract

Generalized additive models (GAMs) are a leading model class for interpretable machine learning. GAMs were originally defined with smooth shape functions of the predictor variables and trained using smoothing splines. Recently, tree-based GAMs where shape functions are gradient-boosted ensembles of bagged trees were proposed, leaving the door open for the estimation of a broader class of shape functions (e.g. Explainable Boosting Machine (EBM)). In this paper, we introduce a competing three-step GAM learning approach where we combine (i) the knowledge of the way to split the covariates space brought by an additive tree model (ATM), (ii) an ensemble of predictive linear scores derived from generalized linear models (GLMs) using a binning strategy based on the ATM, and (iii) a final GLM to have a prediction model that ensures auto-calibration. Numerical experiments illustrate the competitive performances of our approach on several datasets compared to GAM with splines, EBM, or GLM with binarsity penalization. A case study in trade credit insurance is also provided.

Information

Type
Original Research Paper
Copyright
© The Author(s), 2024. Published by Cambridge University Press on behalf of Institute and Faculty of Actuaries
Figure 0

Figure 1 Schematic overview of the estimation procedure.

Figure 1

Table 1. Shape functions $f_{j}$

Figure 2

Figure 2 Plots of the shape functions $f_j$.

Figure 3

Table 2. Choice of the hyperparameters for the regression task (based on a 5-fold cross-validation)

Figure 4

Table 3. Comparison results based on $R^{2}$

Figure 5

Figure 3 Feature shapes learned using GAM with splines, EBM, Distilltrees, and GLM with binarsity penalization.

Figure 6

Table 4. Mean absolute deviation per variable and learning model

Figure 7

Table 5. Choice of the hyperparameters for the classification task (based on a 5-fold cross-validation)

Figure 8

Figure 4 Predicted probabilities vs observed probabilities for the different learning models.

Figure 9

Table 6. Comparison results based on $AUC$

Figure 10

Figure 5 Feature shapes learned using GAM with splines, EBM, Distilltrees, and GLM with binarsity penalization.

Figure 11

Table 7. Choice of the hyperparameters for the credit insurance case study (based on a 5-fold cross-validation)

Figure 12

Table 8. Variable importance of the XGB used for the ATM

Figure 13

Figure 6 Predicted probabilities vs observed probabilities for the different learning models.

Figure 14

Table 9. Comparison results based on $AUC$

Figure 15

Figure 7 Predicted probabilities vs oberved probabilities for the different learning models.

Figure 16

Figure 8 Feature shapes learned using GAM with splines, EBM, Distilltrees, and GLM with binarsity penalization.

Figure 17

Table A.1 Comparison results based on $R^{2}$. Sample size : $50,000$

Figure 18

Table A.2 Comparison results based on $R^{2}$. Sample size : $1,000$

Figure 19

Figure A.1 Feature shapes learned using GAM with splines, EBM, Distilltrees, and GLM with binarsity penalization, with the presence of multicollinearity. Sample size : $50,000$.

Figure 20

Figure A.2 Feature shapes learned using GAM with splines, EBM, Distilltrees, and GLM with binarsity penalization, with the presence of multicollinearity. Sample size : $1,000$.