Gradient-boosted decision trees (GBDT) are widely used and highly effective machine learning approach for tabular data modeling. However, their complex structure may lead to low robustness against small covariate perturbation in unseen data. In this study, we apply one-hot encoding to convert a GBDT model into a linear framework, through encoding of each tree leaf to one dummy variable. This allows for the use of linear regression techniques, plus a novel risk decomposition for assessing the robustness of a GBDT model against covariate perturbations. We propose to enhance the robustness of GBDT models by refitting their linear regression forms with $L_1$ or $L_2$ regularization. Theoretical results are obtained about the effect of regularization on the model performance and robustness. It is demonstrated through numerical experiments that the proposed regularization approach can enhance the robustness of the one-hot-encoded GBDT models.
翻译:梯度提升决策树(GBDT)是一种广泛应用于表格数据建模且高效能的机器学习方法。然而,其复杂结构可能导致对未见数据中微小协变量扰动的鲁棒性较低。本研究通过将每棵树的叶子节点编码为单一虚拟变量,应用独热编码将GBDT模型转换为线性框架。这使得能够运用线性回归技术,并引入一种新颖的风险分解方法,用于评估GBDT模型对协变量扰动的鲁棒性。我们提出通过使用$L_1$或$L_2$正则化重新拟合其线性回归形式,从而增强GBDT模型的鲁棒性。从理论上推导了正则化对模型性能与鲁棒性的影响,并通过数值实验证明,所提出的正则化方法能够增强经独热编码后的GBDT模型的鲁棒性。