机器学习笔记2_决策树

@[toc]

2.1简介

决策树算法经典的机器学习算法,也是使用的很普遍的一类算法,集成学习中的随机森林就是以决策树算法为基础的。决策树是Quinlan(昆兰)在1986年提出来的,最开始的版本是ID3算法,之后他又提出来C4.5算法。后来,有人在昆兰的基础上提出了CART算法,本文主要介绍这三种算法的主要思想。

2.2 信息论基础

可以把决策树当成一系列if-else的集合。比如我们在写程序时总会连用多个if-else,但是哪个特征做if的条件最合适呢?也就是哪个特征最具有区分度?这就是决策树要做的事情。一个叫昆兰的大佬1978年在Standford访问期间选修了图灵的助手D.Michie的课,在完成大作业时,昆兰提出使用信息熵的思路解决,这就是决策树的前身。

我们先介绍信息论中的信息熵,奠基人是另一个大牛香农。在信息论中,熵(entropy)是用来度量事物的不确定性的。所谓不确定性与概率有关,如果一个事情发生的概率越小,不确定性就越大。举个简单的例子:天气预报说明天的降水概率是10%,那就是说明天可能为雨天、晴天、多云...等各种天气,留给人想象的空间很多,但如果天气预报说明天降水概率99%,那基本上可以确定明天会下雨,也就不用多想,带伞就好了。这个例子就是说小概率事件所含的信息量多。信息量可以用负对数来描述,其图像如下:

机器学习笔记2_决策树_第1张图片
在这里插入图片描述

对于一个离散的分布,随机变量的取值可能为多个值,信息熵就表示每个取值所含信息量的加权平均,以下表达式表示量化信息熵:

由于信息熵与X的取值无关,只与不同取值的概率有关,因此上式可以写为:

其中表示随机事件,为X的第个取值发生的概率,若,则定义。对于一个随机变量来说,如果每个取值发生的概率相等的话,则此时的信息熵最大。如对于只有两个取值的伯努利分布:

X 0 1

则信息熵为:

这个就是我们熟悉的交叉熵(cross-entropy)啦,其图像如下所示:

机器学习笔记2_决策树_第2张图片
在这里插入图片描述

也就是说当p=0.5时熵最大,为。这个结论可以推广到对于X取多个离散值的情况,因此有:

对于多个随机变量,有联合熵,

在此基础上又条件熵,注意,用到的概率是联合概率:

如果概率用到是数据的频率,则得到的熵和条件熵称为经验熵和经验条件熵。把信息增益(information gain)定义为,经验熵与经验条件熵的差:

也就是,原来随机变量X的不确定度为,这时Y发生,在这个条件下,X发生的不确定度减小为,不确定度的减少程度就是信息增益,也就是增加了多少信息。计算公式如下:
机器学习笔记2_决策树_第3张图片
在这里插入图片描述

2.3 特征选择

我们绕回决策树,把训练集的划分看成随机变量,特征看成随机变量,表示分类的不确定度,表示加入某个特征A后对D进行新的分类的不确定度,他们的差就是信息增益,也就是特征A使得D不确定度减小的程度,很显然,如果某个特征让不确定度减小的程度多,那就说明这是个好特征。
一个简单的例子如下(来自李航《统计学习方法》):
数据如图:

机器学习笔记2_决策树_第4张图片
在这里插入图片描述

特征选择的过程如下:

人们在实测中发现,相同条件下,取值比较多的特征比取值少的特征信息增益大。为了解决这个问题,大佬昆兰又提出信息增益率的概念:

机器学习笔记2_决策树_第5张图片
在这里插入图片描述

机器学习笔记2_决策树_第6张图片
在这里插入图片描述

2.4 ID3与C4.5

ID3算法的基本思路就是:从根节点开始选择信息增益最大的特征,根据该特征的不同取值建立不同的子节点,每个子节点递归地选择特征,直到没有特征或者信息增益小于某个阈值为止。
算法过程为:
输入:信息增益的阈值,训练的数据集,特征集, 每个特征的取值有个
输出:决策树T

  1. 判断中的所有样本是否为同一类,即所有训练数据的标签是否为同一个。如果是,则T为单结点树,这时返回T
  2. 判断特征是否为空,如果是则返回单节点树T,把训练集中实例树最大的类作为该结点的类标记。否则计算中每个特征的信息增益,选择信息增益最大的特征
  3. 判断刚刚计算的最大信息增益是否小于阈值,如果是,则把T作为单结点树,把中实例数最多的类别作为该结点的类标记
  4. 否则,按照的不同取值把划分为不同的子集,把中实例数最多的类别作为该结点的类标记,返回T
  5. 对于所有子节点,令, 地洞调用1-4

C4.5方法与ID3方法相同,在生成树的过程中用信息增益比代替信息增益。

2.5 决策树的剪枝

如果决策树的分支过多,分类过细则会造成过拟合现象,模型在训练数据上表现很好,但在测试时则表现很差。这时需要主动把分支结点合并到父结点中,达到裁剪决策树的目的,从而降低过拟合。剪枝分为“预剪枝”和“后剪枝”。

预剪枝
所谓“预剪枝”,就是在生成决策树时就用一定的方法控制子树的生成,一种可行的方法,就是用验证集数据,当划分的子树在验证集上仍能保持较好的准确率,则按照特征的不同取值进行划分,否则停止划分。
后剪枝
对于后剪枝,就是先生成一个完整的决策树,接着定义损失函数,按照使得损失函数最小的方法合并叶结点到父结点。

下边介绍一种可行的后剪枝方法:
我们知道,一颗理想的决策树中,所有同一类的训练样本都落入到同一个叶子节点中,但现实 中每个叶子节点中总有错误的分类,假设树中有个子叶子节点,其中第个结点中有个训练样本,由于存在错误的分类,这些样本点中第类有个,,可以用经验熵来衡量这种分类损失,即:

因为前边介绍过,经验熵的取值范围为:

如果叶子节点没有分类错误,则

这是每个叶子节点的损失,而对于整个树的损失则可以表示为叶子节点损失的加权平均,权重就是叶子节点中训练样本(样本点)的个数,所以有:

在之前LR中讲正则项时有提到,权重过大会造成过拟合,而在决策树中,子节点的个数太多同样会造成过拟合,因此我们需要加一个正则项来约束子节点的个数,因此损失函数如下:

假设一个叶结点回缩到父结点之前与之后整体树的损失为与,如果,则进行剪枝。

2.6 CART

CART的全称是classification and regression tree. 由Breiman等人在1984年提出,是应用广泛的决策树学习方法。与ID3和C4.5不同的是,CART构建的是二叉树,我们知道在数据结构中,二叉树有很多优良的性质!sklearn内部默认的决策树算法也是CART。下边分开两部分介绍:

2.6.1 CART分类树

在ID3中,使用信息增益选择特征,在C4.5中使用信息增益比来选择特征,在MART中则使用基尼系数(Gini)来代替信息增益。信息熵由于需要大量的对数运算,计算速度偏慢,而基尼系数则只需要做乘法和加减法运算,其定义如下:

其中表示训练样本属于第类的概率。特别地,在二分类问题中,样本点属于第一类的概率为,则基尼系数为:

如果训练样本集合为,则基尼系数为:

表示属于第类的样本子集,如果根据特征A的取值把样本集合分成和两部分,则在特征A的条件下,训练样本集合的基尼系数为:

也就是两个子集基尼系数的期望(加权平均)

与ID3和C4.5不同,CART分类算法建立的是二叉决策树,比如特征A有三种特征三种特征,CART会把特征分为和,和以及和三种,分别计算着三种组合的基尼系数,对于未分开的特征,后边还有机会做计算。

分类算法的具体流程为:
输入:输入训练集,基尼系数阈值,样本个数阈值
输出:CART决策树T

  1. 对于当前节点的数据集D,如果样本个数小于阈值,返回T
  2. 计算D的基尼系数,如果基尼系数小于阈值,返回T
  3. 对于各个特征A和特征的各个取值a,把D划分成和两部分,根据上述例子计算各个特征各个取值的基尼系数
  4. 选择基尼系数最小的特征A和对应的特征值a,根据是否取a把D分成和两个子树。
  5. 递归调用(1)-(4)
    仍是上述例子,构建的CART决策树为:


    机器学习笔记2_决策树_第7张图片
    在这里插入图片描述

2.6.2 CART回归树

对于回归模型,同样是测试每个特征的每个取值。由于要拟合的函数是连续的,所以考虑输入数据集的划分,用平方误差表示回归树对训练数据的误差:

目标就是如何划分训练数据集,找到划分点,使得划分后的两个子集和的误差最小,同时他们的平方误差和也最小。注意,我们训练集其中我们要找的划分点是其中的第个变量,因此目标函数可以表示为:

其中分别表示中所有的均值。在预测时,回归树输出的也是叶子节点的均值。树建立的过程与分类树类似。

2.6.3 CART的剪枝

与2.5介绍的剪枝策略类似,只不过选用的是损失函数为基尼系数。

你可能感兴趣的:(机器学习笔记2_决策树)