【技术分享】GBDT算法-原理篇

本文原作者：蒋凯，经授权后发布。

导语 ：工业界机器学习大杀器解读。

GBDT是常用的机器学习算法之一，因其出色的特征自动组合能力和高效的运算大受欢迎。

这里简单介绍一下GBDT算法的原理，后续再写一个实战篇。

1、决策树的分类

决策树分为两大类，分类树和回归树。

分类树用于分类标签值，如晴天/阴天/雾/雨、用户性别、网页是否是垃圾页面；

回归树用于预测实数值，如明天的温度、用户的年龄、网页的相关程度；

两者的区别：

分类树的结果不能进行加减运算，晴天+晴天没有实际意义；

回归树的结果是预测一个数值，可以进行加减运算，例如20岁+3岁=23岁。

GBDT中的决策树是回归树，预测结果是一个数值，在点击率预测方面常用GBDT，例如用户点击某个内容的概率。

2、GBDT概念

GBDT的全称是Gradient Boosting Decision Tree，梯度提升决策树。

要理解GBDT，首先就要理解这个B(Boosting)。

Boosting是一族可将弱学习器提升为强学习器的算法，属于集成学习（ensemble learning）的范畴。Boosting方法基于这样一种思想：对于一个复杂任务来说，将多个专家的判断进行适当的综合所得出的判断，要比其中任何一个专家单独的判断要好。通俗地说，就是“三个臭皮匠顶个诸葛亮”的道理。

基于梯度提升算法的学习器叫做GBM(Gradient Boosting Machine)。理论上，GBM可以选择各种不同的学习算法作为基学习器。GBDT实际上是GBM的一种情况。

为什么梯度提升方法倾向于选择决策树作为基学习器呢？(也就是GB为什么要和DT结合，形成GBDT) 决策树可以认为是if-then规则的集合，易于理解，可解释性强，预测速度快。同时，决策树算法相比于其他的算法需要更少的特征工程，比如可以不用做特征标准化，可以很好的处理字段缺失的数据，也可以不用关心特征间是否相互依赖等。决策树能够自动组合多个特征。

不过，单独使用决策树算法时，有容易过拟合缺点。所幸的是，通过各种方法，抑制决策树的复杂性，降低单颗决策树的拟合能力，再通过梯度提升的方法集成多个决策树，最终能够很好的解决过拟合的问题。由此可见，梯度提升方法和决策树学习算法可以互相取长补短，是一对完美的搭档。

至于抑制单颗决策树的复杂度的方法有很多，比如限制树的最大深度、限制叶子节点的最少样本数量、限制节点分裂时的最少样本数量、吸收bagging的思想对训练样本采样（subsample），在学习单颗决策树时只使用一部分训练样本、借鉴随机森林的思路在学习单颗决策树时只采样一部分特征、在目标函数中添加正则项惩罚复杂的树结构等。

演示例子：

考虑一个简单的例子来演示GBDT算法原理

下面是一个二分类问题，1表示可以考虑的相亲对象，0表示不考虑的相亲对象

特征维度有3个维度，分别对象 身高，金钱，颜值。

对应这个例子，训练结果是perfect的，全部正确， 特征权重可以看出，对应这个例子训练结果颜值的重要度最大，看一下训练得到的树。

Tree 0:

Tree 1:

3、原理推导

3.1 目标函数

监督学习的关键概念：模型（model）、参数（parameters）、目标函数（objective function）

模型就是所要学习的条件概率分布或者决策函数，它决定了在给定特征向量时如何预测出目标。

参数就是我们要从数据中学习得到的内容。模型通常是由一个参数向量决定的函数。

目标函数通常定义为如下形式：

3.2 加法模型

GBDT算法可以看成是由K棵树组成的加法模型：

如何来学习加法模型呢？

解这一优化问题，可以用前向分布算法（forward stagewise algorithm）。因为学习的是加法模型，如果能够从前往后，每一步只学习一个基函数及其系数（结构），逐步逼近优化目标函数，那么就可以简化复杂度。这一学习过程称之为Boosting。具体地，我们从一个常量预测开始，每次学习一个新的函数，过程如下：

在第t步，这个时候目标函数可以写为：

举例说明，假设损失函数为平方损失（square loss），则目标函数为：

其中，称

之为残差（residual）。因此，使用平方损失函数时，GBDT算法的每一步在生成决策树时只需要拟合前面的模型的残差。

3.3 泰勒公式

定义：

泰勒公式简单的理解，就是函数某个点的取值可以用参考点取值和n+1阶导数的来表示，而且这个公式是有规律的比较好记。

根据泰勒公式把函数

在

点处二阶展开，可得到如下等式：

则等式(1)可转化为：

假设损失函数为平方损失函数，把对应的一阶导数和二阶导数代入等式(4)即得等式(2)。

由于函数中的常量在函数最小化的过程中不起作用，因此我们可以从等式(4)中移除掉常量项，得：

3.4 GBDT算法

一颗生成好的决策树，假设其叶子节点个数为

，

决策树的复杂度可以由正则项

来定义，即决策树模型的复杂度由生成的树的叶子节点数量和叶子节点对应的值向量的L2范数决定。

定义集合

为所有被划分到叶子节点的训练样本的集合。等式(5)可以根据树的叶子节点重新组织为T个独立的二次函数的和：

定义

，则等式(6)可写为：

因为一元二次函数最小值处，一阶导数等于0:

此时，目标函数的值为

综上，为了便于理解，单颗决策树的学习过程可以大致描述为： 1. 枚举所有可能的树结构 q 2. 用等式(8)为每个q计算其对应的分数Obj，分数越小说明对应的树结构越好 3. 根据上一步的结果，找到最佳的树结构，用等式(7)为树的每个叶子节点计算预测值

然而，可能的树结构数量是无穷的，所以实际上我们不可能枚举所有可能的树结构。通常情况下，我们采用贪心策略来生成决策树的每个节点。 1. 从深度为0的树开始，对每个叶节点枚举所有的可用特征 2. 针对每个特征，把属于该节点的训练样本根据该特征值升序排列，通过线性扫描的方式来决定该特征的最佳分裂点，并记录该特征的最大收益（采用最佳分裂点时的收益） 3. 选择收益最大的特征作为分裂特征，用该特征的最佳分裂点作为分裂位置，把该节点生长出左右两个新的叶节点，并为每个新节点关联对应的样本集 4. 回到第1步，递归执行到满足特定条件为止

3.5 收益的计算

如何计算每次分裂的收益呢？假设当前节点记为C,分裂之后左孩子节点记为L，右孩子节点记为R，则该分裂获得的收益定义为当前节点的目标函数值减去左右两个孩子节点的目标函数值之和：

根据等式(8)可得：

其中，

项表示因为增加了树的复杂性（该分裂增加了一个叶子节点）带来的惩罚。

最后，总结一下GBDT的学习算法： 1. 算法每次迭代生成一颗新的决策树 ; 2. 在每次迭代开始之前，计算损失函数在每个训练样本点的一阶导数和二阶导数 ; 3. 通过贪心策略生成新的决策树，通过等式(7)计算每个叶节点对应的预测值 4. 把新生成的决策树

添加到模型中：

保持简单

易经中说道“易则易知，简则易从”，就是越是简易的东西，越是容易被理解和得到执行。很多机器学习模型都会尽量让学习到的模型尽量简单，尽量减少参数，越是简单的模型，通用性越好，也是这个道理。

Xgboost和GBDT的区别：

GBDT:

 GBDT它的非线性变换比较多，表达能力强，而且不需要做复杂的特征工程和特征变换。

 GBDT的缺点也很明显，Boost是一个串行过程，不好并行化，而且计算复杂度高，同时不太适合高维稀疏特征；

 传统GBDT在优化时只用到一阶导数信息。

Xgboost：

它 有以下几个优良的特性：

       1. 显示的把树模型复杂度作为正则项加到优化目标中。

       2. 公式推导中用到了二阶导数，用了二阶泰勒展开。（GBDT用牛顿法貌似也是二阶信息）

       3. 实现了分裂点寻找近似算法。

       4. 利用了特征的稀疏性。

       5. 数据事先排序并且以block形式存储，有利于并行计算。

       6. 基于分布式通信框架rabit，可以运行在MPI和yarn上。（最新已经不基于rabit了）

       7. 实现做了面向体系结构的优化，针对cache和内存做了性能优化。

腾讯云一站式机器学习平台智能钛TI-ONE试运营阶段限时0折，欢迎大家积极试用。

https://cloud.tencent.com/product/tio

更多优质技术文章请关注官方知乎机构号：

https://www.zhihu.com/org/teng-xun-zhi-neng-tai-ji-qi-xue-xi-ping-tai/activities

更多优质技术文章请关注官方微信公众号：