全网最浅显易懂的GBDT(xgboost)算法原理深入剖析

梯度提升(Gradient boosting)是一种用于回归、分类和排序任务的技术,属于Boosting算法族的一部分。Boosting是一族可将弱学习器提升为强学习器的算法,属于集成学习(ensemble learning)的范畴。Boosting方法基于这样一种思想:对于一个复杂任务来说,将多个专家的判断进行适当的综合所得出的判断,要比其中任何一个专家单独的判断要好。通俗地说,就是“三个臭皮匠顶个诸葛亮”的道理。梯度提升同其他boosting方法一样,通过集成(ensemble)多个弱学习器,通常是决策树,来构建最终的预测模型。

Boosting、bagging和stacking是集成学习的三种主要方法。不同于bagging方法,boosting方法通过分步迭代(stage-wise)的方式来构建模型,在迭代的每一步构建的弱学习器都是为了弥补已有模型的不足。Boosting族算法的著名代表是AdaBoost,AdaBoost算法通过给已有模型预测错误的样本更高的权重,使得先前的学习器做错的训练样本在后续受到更多的关注的方式来弥补已有模型的不足。与AdaBoost算法不同,梯度提升方法在迭代的每一步构建一个能够沿着梯度最陡的方向降低损失(steepest-descent)的学习器来弥补已有模型的不足。经典的AdaBoost算法只能处理采用指数损失函数的二分类学习任务,而梯度提升方法通过设置不同的可微损失函数可以处理各类学习任务(多分类、回归、Ranking等),应用范围大大扩展。另一方面,AdaBoost算法对异常点(outlier)比较敏感,而梯度提升算法通过引入bagging思想、加入正则项等方法能够有效地抵御训练数据中的噪音,具有更好的健壮性。这也是为什么梯度提升算法(尤其是采用 决策树 作为弱学习器的GBDT算法)如此流行的原因。之前有种观点认为GBDT是性能最好的 机器学习 算法,这当然有点过于激进又固步自封的味道,但通常各类机器学习算法比赛的赢家们都非常青睐GBDT算法,由此可见该算法的实力不可小觑。

基于梯度提升算法的学习器叫做GBM(Gradient Boosting Machine)。理论上,GBM可以选择各种不同的学习算法作为基学习器。现实中,用得最多的基学习器是决策树。为什么梯度提升方法倾向于选择决策树(通常是CART树)作为基学习器呢?这与决策树算法自身的优点有很大的关系。决策树可以认为是if-then规则的集合,易于理解,可解释性强,预测速度快。同时,决策树算法相比于其他的算法需要更少的特征工程,比如可以不用做特征标准化,可以很好的处理字段缺失的数据,也可以不用关心特征间是否相互依赖等。决策树能够自动组合多个特征,它可以毫无压力地处理特征间的交互关系并且是非参数化的,因此你不必担心异常值或者数据是否线性可分(举个例子,决策树能轻松处理好类别A在某个特征维度x的末端,类别B在中间,然后类别A又出现在特征维度x前端的情况)。不过,单独使用决策树算法时,有容易过拟合的缺点。所幸的是,通过各种方法,抑制决策树的复杂性,降低单颗决策树的拟合能力,再通过梯度提升的方法集成多个决策树,最终能够很好的解决过拟合的问题。由此可见,梯度提升方法和决策树学习算法可以互相取长补短,是一对完美的搭档。至于抑制单颗决策树的复杂度的方法有很多,比如限制树的最大深度、限制叶子节点的最少样本数量、限制节点分裂时的最少样本数量、吸收bagging的思想对训练样本采样(subsample),在学习单颗决策树时只使用一部分训练样本、借鉴随机森林的思路在学习单颗决策树时只采样一部分特征、在目标函数中添加正则项惩罚复杂的树结构等。现在主流的GBDT算法实现中这些方法基本上都有实现,因此GBDT算法的超参数还是比较多的,应用过程中需要精心调参,并用交叉验证的方法选择最佳参数。

本文对GBDT算法原理进行介绍,从机器学习的关键元素出发,一步一步推导出GBDT算法背后的理论基础,读者可以从这个过程中了解到GBDT算法的来龙去脉。对于该算法的工程实现,本文也有较好的指导意义,实际上对 机器学习关键概念元素的区分对应了软件工程中的“开放封闭原则”的思想,基于此思想的实现将会具有很好的模块独立性和扩展性。

机器学习的关键元素

先复习下监督学习的关键概念:模型(model)、参数(parameters)、目标函数(objective function)。

模型就是所要学习的条件概率分布或者决策函数,它决定了在给定特征向量x时如何预测出目标y。定义 x_i \in R^d 为训练集中的第i个训练样本,则线性模型(linear model)可以表示为:\hat{y}_i = \sum_j{w_j x_{ij}}。模型预测的分数\hat{y}_i在不同的任务中有不同的解释。例如在逻辑回归任务中,1/(1+exp(-\hat{y}_i))表示模型预测为正例的概率;而在排序学习任务中,\hat{y}_i表示排序分。

参数就是我们要从数据中学习得到的内容。模型通常是由一个参数向量决定的函数。例如,线性模型的参数可以表示为:\Theta={w_j|j=1,\cdots,d}

目标函数通常定义为如下形式:Obj(\Theta)=L(\Theta)+\Omega(\Theta) 其中,L(\Theta)是损失函数,用来衡量模型拟合训练数据的好坏程度;\Omega(\Theta)称之为正则项,用来衡量学习到的模型的复杂度。训练集上的损失(Loss)定义为:L=\sum_{i=1}^n l(y_i, \hat{y}_i)

常用的损失函数有:

  • 平方损失(square loss): l(y_i, \hat{y}_i)=(y_i - \hat{y}_i)^2
  • Logistic损失: l(y_i, \hat{y}_i)=y_i ln(1+e^{y_i}) + (1-y_i)ln(1+e^{\hat{y}_i})

常用的正则项有L1范数\Omega(w)=\lambda \Vert w \Vert_1和L2范数\Omega(w)=\lambda \Vert w \Vert_2。Ridge regression就是指使用平方损失和L2范数正则项的线性回归模型;Lasso regression就是指使用平方损失和L1范数正则项的线性回归模型;逻辑回归(Logistic regression)指使用logistic损失和L2范数或L1范数正则项的线性模型。

目标函数之所以定义为损失函数和正则项两部分,是为了尽可能平衡模型的偏差和方差(Bias Variance Trade-off)。最小化目标函数意味着同时最小化损失函数和正则项,损失函数最小化表明模型能够较好的拟合训练数据,一般也预示着模型能够较好地拟合真实数据(groud true);另一方面,对正则项的优化鼓励算法学习到较简单的模型,简单模型一般在测试样本上的预测结果比较稳定、方差较小(奥坎姆剃刀原则)。也就是说,优化损失函数尽量使模型走出欠拟合的状态,优化正则项尽量使模型避免过拟合。

从概念上区分模型、参数和目标函数给学习算法的工程实现带来了益处,使得机器学习的各个组成部分之间耦合尽量松散。

加法模型(additive model)

GBDT算法可以看成是由K棵树组成的加法模型。一般我们使用CART(classification and regression tree)树作为集成学习的基础模块,下图是一颗用来预测某用户是否喜欢电脑游戏X的例子,该例子中人群被划分到不同的叶子节点,每个叶子节点被赋予一个分数来表示其与目标的关联度。

cart.png

通常,一棵树并不足以强大到能够解决问题,我们使用基于多棵树的集成模型来建模。在GBDT算法中,该集成模型即加法模型,它的形式化如下:

\hat{y}_i=\sum_{k=1}^K f_k(x_i), f_k \in F \tag 0

其中F为所有树组成的函数空间,以回归任务为例,回归树可以看作为一个把特征向量映射为某个score的函数。该模型的参数为:\Theta={f_1,f_2, \cdots, f_K }。与一般的机器学习算法不同的是,加法模型不是学习d维空间中的权重,而是直接学习函数(决策树)集合。

twocart.png

如上图所示,每个样本的最终得分是由每颗子树的得分相加得到的。

上述加法模型的目标函数定义为:

Obj=\sum_{i=1}^n l(y_i, \hat{y}i) + \sum_{k=1}^K \Omega(f_k),

其中\Omega(\cdot)是正则项,表示决策树的复杂度,那么该如何定义树的复杂度呢?比如,可以考虑树的节点数量、树的深度或者叶子节点所对应的分数的L2范数等等。

如何来学习加法模型呢?

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

\begin{split} \hat{y}_i^0 &= 0 \\ \hat{y}_i^1 &= f_1(x_i) = \hat{y}_i^0 + f_1(x_i) \\ \hat{y}_i^2 &= f_1(x_i) + f_2(x_i) = \hat{y}_i^1 + f_2(x_i) \\ & \cdots \\ \hat{y}_i^t &= \sum_{k=1}^t f_k(x_i) = \hat{y}_i^{t-1} + f_t(x_i) \ \end{split}

那么,在每一步如何决定哪一个函数f被加入呢?指导原则还是最小化目标函数。 在第t步,模型对x_i的预测为:\hat{y}_i^t= \hat{y}_i^{t-1} + f_t(x_i),其中f_t(x_i)为这一轮我们要学习的函数(决策树)。这个时候目标函数可以写为:

\begin{split} Obj^{(t)} &= \sum_{i=1}^nl(y_i, \hat{y}_i^t) + \sum_{i=i}^t \Omega(f_i) \\ &= \sum_{i=1}^n l\left(y_i, \hat{y}_i^{t-1} + f_t(x_i) \right) + \Omega(f_t) + constant \end{split}\tag{1}

举例说明,假设损失函数为平均平方损失(mean square loss),则目标函数为:

\begin{split} Obj^{(t)} &= \sum_{i=1}^n \left(y_i - (\hat{y}_i^{t-1} + f_t(x_i)) \right)^2 + \Omega(f_t) + constant \\ &= \sum_{i=1}^n \left[2(\hat{y}_i^{t-1} - y_i)f_t(x_i) + f_t(x_i)^2 \right] + \Omega(f_t) + constant \end{split}\tag{2}

其中,(\hat{y}_i^{t-1} - y_i)称之为残差(residual)。因此,在使用平方损失函数时,GBDT算法的每一步在生成决策树时只需要拟合前面的模型的残差。

在使用其他损失函数时,如logistic loss,就很难得到如MSE损失那样简洁的解析表达式。因此,Xgboost使用泰勒公式的二阶展开式来近似目标函数,详细过程如下:

泰勒公式:设n是一个正整数,如果定义在一个包含 a 的区间上的函数 fa 点处 n+1 次可导,那么对于这个区间上的任意 x 都有:\displaystyle f(x)=\sum_{n=0}^{N}\frac{f^{(n)}(a)}{n!}(x-a)^ n+R_n(x),其中的多项式称为函数在 a 处的泰勒展开式,R_n(x)是泰勒公式的余项且是(x-a)^ n的高阶无穷小。
其中,零阶导数理解为本身,常数0阶导数仍为本身,函数的0阶导数为函数本身;并且规定0!=1。

根据泰勒公式把函数f(x+\Delta x)在点x处二阶展开,可得到如下等式: f(x+\Delta x) \approx f(x) + f'(x)\Delta x + \frac12 f''(x)\Delta x^2 \tag 3
(即用x+\Delta x代替泰勒公式中的x,并且用x代替泰勒公式中的a

由等式(1)可知,目标函数是关于变量 \hat{y}_i^{t-1} + f_t(x_i) 的函数,若把变量\hat{y}_i^{t-1}看成是等式(3)中的x,把变量f_t(x_i)看成是等式(3)中的\Delta x,则等式(1)可转化为:

Obj^{(t)} = \sum_{i=1}^n \left[ l(y_i, \hat{y}_i^{t-1}) + g_if_t(x_i) + \frac12h_if_t^2(x_i) \right] + \Omega(f_t) + constant \tag 4

其中,g_i定义为损失函数的一阶导数,即g_i=\partial_{\hat{y}_i^{t-1}}l(y_i,\hat{y}_i^{t-1})h_i定义为损失函数的二阶导数,即h_i=\partial_{\hat{y}_i^{t-1}}^2l(y_i,\hat{y}_i^{t-1})。 这里需要注意的是,损失函数是对前一步的预测结果\hat{y}_i^{t-1}求导。

假设损失函数为平方损失函数,则g_i=\partial_{\hat{y}^{t-1}}(\hat{y}^{t-1} - y_i)^2 = 2(\hat{y}^{t-1} - y_i)h_i=\partial_{\hat{y}^{t-1}}^2(\hat{y}^{t-1} - y_i)^2 = 2,把g_ih_i代入等式(4)即得等式(2)。

由于函数中的常量在函数最小化的过程中不起作用,因此我们可以从等式(4)中移除掉常量项,得:
Obj^{(t)} \approx \sum_{i=1}^n \left[ g_if_t(x_i) + \frac12h_if_t^2(x_i) \right] + \Omega(f_t) \tag 5

由于要学习的函数仅仅依赖于目标函数,从等式(5)可以看出只需为学习任务定义好损失函数,并为每个训练样本计算出损失函数的一阶导数和二阶导数,通过在训练样本集上最小化等式(5)即可求得每步要学习的函数f(x),从而根据加法模型等式(0)可得最终要学习的模型。

GBDT算法

一颗生成好的决策树,假设其叶子节点个数为T,该决策树是由所有叶子节点对应的值组成的向量w \in R^T,以及一个把特征向量映射到叶子节点索引(Index)的函数q:R^d \to {1,2,\cdots,T}组成的。因此,策树可以定义为f_t(x)=w_{q(x)}

决策树的复杂度可以由正则项 \Omega(f_t)=\gamma T + \frac12 \lambda \sum_{j=1}^T w_j^2 来定义,即 决策树模型的复杂度由生成的树的叶子节点数量和叶子节点对应的值向量的L2范数决定。

定义集合 I_j=\{ i \vert q(x_i)=j \} 为所有被划分到叶子节点 j 的训练样本的集合。等式(5)可以根据树的叶子节点重新组织为T个独立的二次函数的和:
\begin{split} Obj^{(t)} &\approx \sum_{i=1}^n \left[ g_if_t(x_i) + \frac12h_if_t^2(x_i) \right] + \Omega(f_t) \\ &= \sum_{i=1}^n \left[ g_iw_{q(x_i)} + \frac12h_iw_{q(x_i)}^2 \right] + \gamma T + \frac12 \lambda \sum_{j=1}^T w_j^2 \\ &= \sum_{j=1}^T \left[(\sum_{i \in I_j}g_i)w_j + \frac12(\sum_{i \in I_j}h_i + \lambda)w_j^2 \right] + \gamma T \end{split}\tag 6

定义G_j=\sum_{i \in I_j}g_iH_j=\sum_{i \in I_j}h_i,则等式(6)可写为:
Obj^{(t)} = \sum_{j=1}^T \left[G_iw_j + \frac12(H_i + \lambda)w_j^2 \right] + \gamma T

假设树的结构是固定的,即由函数q(x)确定,令函数Obj^{(t)}对参数 w_j 的一阶导数等于0,即可求得叶子节点 j 对应的值为:w_j^*=-\frac{G_j}{H_j+\lambda} \tag 7

此时,目标函数的值为
Obj^* = -\frac12 \sum_{j=1}^T \frac{G_j^2}{H_j+\lambda} + \gamma T \tag 8
该函数度量了一颗树结构的好坏程度,其类似于决策树算法中的不纯度度量(impurity measure),只是额外考虑了正则项。

struct_score.png

综上,为了便于理解,单颗决策树的学习过程可以大致描述为:

  1. 枚举所有可能的树结构 q
  2. 用等式(8)为每个 q 计算其对应的分数Obj,分数越小说明对应的树结构越好
  3. 根据上一步的结果,找到最佳的树结构,用等式(7)为树的每个叶子节点计算预测值

然而,可能的树结构数量是无穷的,所以实际上我们不可能枚举所有可能的树结构。通常情况下,我们采用贪心策略来生成决策树的每个节点。

  1. 从深度为0的树开始,对每个叶节点枚举所有的可用特征
  2. 针对每个特征,把属于该节点的训练样本根据该特征值升序排列,通过线性扫描的方式来决定该特征的最佳分裂点,并记录该特征的最大收益(采用最佳分裂点时的收益);这里假设类别型特征已经做了one-hot编码
  3. 选择收益最大的特征作为分裂特征,用该特征的最佳分裂点作为分裂位置,把该节点生长出左右两个新的叶节点,并为每个新节点关联对应的样本集
  4. 回到第1步,递归执行到满足特定条件为止

在上述算法的第二步,样本排序的时间复杂度为O(n \log n),假设共有 K 个特征,那么生成一颗深度为 d 的树的时间复杂度为O(dKn\log n)。具体实现可以进一步优化计算复杂度,比如可以缓存每个特征的排序结果等。

如何计算每次分裂的收益呢?假设当前节点记为C,分裂之后左孩子节点记为L,右孩子节点记为R,则该分裂获得的收益定义为当前节点的目标函数值减去左右两个孩子节点的目标函数值之和:Gain=Obj_C-Obj_L-Obj_R,具体地,根据等式(8)可得:Gain=\frac12 \left[ \frac{G_L^2}{H_L+\lambda} + \frac{G_R^2}{H_R+\lambda} - \frac{(G_L+G_R)^2}{H_L+H_R+\lambda}\right] - \gamma 其中,-\gamma 项表示因为增加了树的复杂性(该分裂增加了一个叶子节点)带来的惩罚。当节点的最佳分裂收益小于0时(也就是不考虑增加树的复杂性惩罚时的收益小于\gamma时),意味着此时最好不要分裂该节点。这也就是决策树剪枝算法的原理。搜索某特征最佳分裂点的过程可参考下面的例子。

split_find.png

A left to right scan is sufficient to calculate the structure score of all possible split solutions, and we can find the best split efficiently.

关于如何计算特征的重要度,请参考《Tree ensemble算法的特征重要度计算》。

最后,总结一下GBDT的学习算法:

  1. 算法每次迭代生成一颗新的 决策树
  2. 在每次迭代开始之前,计算损失函数在每个训练样本点的一阶导数g_i和二阶导数h_i
  3. 通过贪心策略生成新的 决策树,通过等式(7)计算每个叶节点对应的预测值
  4. 把新生成的 决策树f_t(x)添加到模型中:\hat{y}_i^t = \hat{y}_i^{t-1} + f_t(x_i)

通常在第四步,我们把模型更新公式替换为:\hat{y}_i^t = \hat{y}_i^{t-1} + \epsilon f_t(x_i),其中\epsilon称之为步长或者学习率。增加\epsilon因子的目的是为了避免模型过拟合。

参考资料

  • [1] Gradient Boosting 的更多内容
  • [2] XGBoost 是一个优秀的GBDT开源软件库,有多种语言接口
  • [3] Pyramid 是一个基于Java语言的机器学习库,里面也有GBDT算法的介绍和实现
  • [4] Friedman的论文《Greedy function approximation: a gradient boosting machine》是比较早的GBDT算法文献,但是比较晦涩难懂,不适合初学者,高阶选手可以进一步学习
  • [5] "A Gentle Introduction to Gradient Boosting"是关于Gradient Boosting的一个通俗易懂的解释,比较适合初学者或者是已经对GBDT算法原理印象不深的从业者
  • [6] 关于GBDT算法调参的经验和技巧可以参考这两篇博文:《GBM调参指南》、 《XGBoost调参指南》,作者使用的算法实现工具来自于著名的Python机器学习工具scikit-learn
  • [7] GBDT算法在搜索引擎排序中的应用可以查看这篇论文《Web-Search Ranking with Initialized Gradient Boosted Regression Trees 》,这篇论文提出了一个非常有意思的方法,用一个已经训练好的随机森林模型作为GBDT算法的初始化,再用GBDT算法优化最终的模型,取得了很好的效果
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,835评论 4 364
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,598评论 1 295
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 109,569评论 0 244
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 44,159评论 0 213
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,533评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,710评论 1 222
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,923评论 2 313
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,674评论 0 203
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,421评论 1 246
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,622评论 2 245
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,115评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,428评论 2 254
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,114评论 3 238
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,097评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,875评论 0 197
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,753评论 2 276
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,649评论 2 271

推荐阅读更多精彩内容