Sep, 2023

GRANDE: 基于梯度的决策树集成

TL;DR本文提出了 GRANDE,一种基于梯度的决策树集成模型,通过端到端梯度下降学习硬性、轴对齐的决策树集成模型。GRANDE 基于树集成的稠密表示,利用直线传播运算符对所有模型参数进行联合优化。通过结合轴对齐分割和基于梯度的优化,我们的方法既具备了适合表格数据的归纳偏好,又具备了优化的灵活性。此外,我们还引入了一种高级的逐实例加权方法,有助于在单个模型中学习简单和复杂的关系。我们在一个预定义的基准测试集上进行了广泛的评估,并且证明我们的方法在大多数数据集上胜过现有的梯度提升和深度学习框架。