决策树ID3算法的信息论基础
作为一个码农经常会不停的敲if, else if, else,其实就已经在用到决策树的思想了。只是你有没有想过,有这么多条件,用哪个条件特征先做if,哪个条件特征后做if比较优呢?怎么准确的定量选择这个标准就是决策树机器学习算法的关键了。1970年代,一个叫昆兰的大牛找到了用信息论中的熵来度量决策树的决策选择过程,方法一出,它的简洁和高效就引起了轰动,昆兰把这个算法叫做ID3。下面我们就看看ID3算法是怎么选择特征的。
首先,我们需要熟悉信息论中熵的概念。熵度量了事物的不确定性,越不确定的事物,它的熵就越大。具体的,随机变量X的熵的表达式如下:
H
(
X
)
=
−
∑
i
=
1
n
p
i
l
o
g
p
i
H(X)=−∑_{i=1}^np_ilogp_i
H(X)=−i=1∑npilogpi
其中n代表X的n种不同的离散取值。而
p
i
p_i
pi代表了
X
X
X取值为i的概率,
l
o
g
log
log为以
2
2
2或者
e
e
e为底的对数。比如X有2个可能的取值,而这两个取值各为
1
2
\frac12
21时X的熵最大,此时X具有最大的不确定性。值为
H
(
X
)
=
−
(
1
2
l
o
g
1
2
+
1
2
l
o
g
1
2
)
=
l
o
g
2
H(X)=−(\frac12log\frac12+\frac12log\frac12)=log2
H(X)=−(21log21+21log21)=log2。如果一个值概率大于1/2,另一个值概率小于1/2,则不确定性减少,对应的熵也会减少。比如一个概率1/3,一个概率2/3,则对应熵为
H
(
X
)
=
−
(
1
3
l
o
g
1
3
+
2
3
l
o
g
2
3
)
=
l
o
g
3
−
2
3
l
o
g
2
<
l
o
g
2
)
H(X)=−(\frac13log\frac13+\frac23log\frac23)=log3−\frac23log2<log2)
H(X)=−(31log31+32log32)=log3−32log2<log2).
熟悉了一个变量X的熵,很容易推广到多个个变量的联合熵,这里给出两个变量X和Y的联合熵表达式:
H
(
X
,
Y
)
=
−
∑
x
i
∈
X
∑
y
i
∈
Y
p
(
x
i
,
y
i
)
l
o
g
p
(
x
i
,
y
i
)
H(X,Y)=−∑_{{x_i}∈X}∑_{y_i∈Y}p(x_i,y_i)logp(x_i,y_i)
H(X,Y)=−xi∈X∑yi∈Y∑p(xi,yi)logp(xi,yi)
有了联合熵,又可以得到条件熵的表达式
H
(
X
∣
Y
)
H(X|Y)
H(X∣Y),条件熵类似于条件概率,它度量了我们的X在知道Y以后剩下的不确定性。表达式如下:
H
(
X
∣
Y
)
=
∑
j
=
1
n
p
(
y
j
)
H
(
X
∣
y
j
)
=
−
∑
x
i
∈
X
∑
y
i
∈
Y
p
(
x
i
,
y
i
)
l
o
g
p
(
x
i
∣
y
i
)
H(X|Y)=∑_{j=1}^np(y_j)H(X|y_j)=−∑_{x_i∈X}∑_{y_i∈Y}p(x_i,y_i)logp(x_i|y_i)
H(X∣Y)=j=1∑np(yj)H(X∣yj)=−xi∈X∑yi∈Y∑p(xi,yi)logp(xi∣yi)
好吧,绕了一大圈,终于可以重新回到ID3算法了。我们刚才提到H(X)度量了X的不确定性,条件熵H(X|Y)度量了我们在知道Y以后X剩下的不确定性,那么H(X)-H(X|Y)呢?从上面的描述大家可以看出,它度量了X在知道Y以后不确定性减少程度,这个度量我们在信息论中称为互信息,,记为I(X,Y)。在决策树ID3算法中叫做信息增益。ID3算法就是用信息增益来判断当前节点应该用什么特征来构建决策树。信息增益大,则越适合用来分类。
上面一堆概念,大家估计比较晕,用下面这个图很容易明白他们的关系。左边的椭圆代表H(X),右边的椭圆代表H(Y),中间重合的部分就是我们的互信息或者信息增益I(X,Y), 左边的椭圆去掉重合部分就是H(X|Y),右边的椭圆去掉重合部分就是H(Y|X)。两个椭圆的并就是H(X,Y)。
上面提到ID3算法就是用信息增益大小来判断当前节点应该用什么特征来构建决策树,用计算出的信息增益最大的特征来建立决策树的当前节点。这里我们举一个信息增益计算的具体的例子。比如我们有15个样本D,输出为0或者1。其中有9个输出为1, 6个输出为0。 样本中有个特征A,取值为A1,A2和A3。在取值为A1的样本的输出中,有3个输出为1, 2个输出为0,取值为A2的样本输出中,2个输出为1,3个输出为0, 在取值为A3的样本中,4个输出为1,1个输出为0.
上面提到ID3算法就是用信息增益大小来判断当前节点应该用什么特征来构建决策树,用计算出的信息增益最大的特征来建立决策树的当前节点。这里我们举一个信息增益计算的具体的例子。比如我们有15个样本D,输出为0或者1。其中有9个输出为1, 6个输出为0。 样本中有个特征A,取值为A1,A2和A3。在取值为A1的样本的输出中,有3个输出为1, 2个输出为0,取值为A2的样本输出中,2个输出为1,3个输出为0, 在取值为A3的样本中,4个输出为1,1个输出为0.
样本D的熵为:
H
(
D
)
=
−
(
9
15
l
o
g
9
15
+
6
15
l
o
g
6
15
)
=
0.971
H(D)=−(\frac9{15}log\frac9{15}+\frac6{15}log\frac6{15})=0.971
H(D)=−(159log159+156log156)=0.971
样本D在特征下的条件熵为:
H
(
D
∣
A
)
=
5
15
H
(
D
1
)
+
5
15
H
(
D
2
)
+
5
15
H
(
D
3
)
=
−
5
15
(
3
5
l
o
g
3
5
+
2
5
l
o
g
2
5
)
−
5
15
(
2
5
l
o
g
2
5
+
3
5
l
o
g
3
5
)
−
5
15
(
4
5
l
o
g
4
5
+
1
5
l
o
g
1
5
)
=
0.888
H(D|A)=\frac5{15}H(D1)+\frac5{15}H(D2)+\frac5{15}H(D3) =−\frac5{15}(\frac35log\frac35+\frac25log\frac25)−\frac5{15}(\frac25log\frac25+\frac35log\frac35)−\frac5{15}(\frac45log\frac45+\frac15log\frac15)=0.888
H(D∣A)=155H(D1)+155H(D2)+155H(D3)=−155(53log53+52log52)−155(52log52+53log53)−155(54log54+51log51)=0.888
对应的信息增益为 I(D,A)=H(D)−H(D|A)=0.083
下面我们看看具体算法过程大概是怎么样的。
输入的是m个样本,样本输出集合为D,每个样本有n个离散特征,特征集合即为A,输出为决策树T。
算法的过程为:
1)初始化信息增益的阈值ϵ
2)判断样本是否为同一类输出Di,如果是则返回单节点树T。标记类别为Di
3) 判断特征是否为空,如果是则返回单节点树T,标记类别为样本中输出类别D实例数最多的类别。
4)计算A中的各个特征(一共n个)对输出D的信息增益,选择信息增益最大的特征Ag
5) 如果Ag的信息增益小于阈值ϵ,则返回单节点树T,标记类别为样本中输出类别D实例数最多的类别。
6)否则,按特征Ag的不同取值Agi将对应的样本输出D分成不同的类别Di。每个类别产生一个子节点。对应特征值为Agi。返回增加了节点的数T。
7)对于所有的子节点,令D=Di,A=A−{Ag}递归调用2-6步,得到子树Ti并返回。
决策树ID3算法的不足
ID3算法虽然提出了新思路,但是还是有很多值得改进的地方。
a)ID3没有考虑连续特征,比如长度,密度都是连续值,无法在ID3运用。这大大限制了ID3的用途。
b)ID3采用信息增益大的特征优先建立决策树的节点。很快就被人发现,在相同条件下,取值比较多的特征比取值少的特征信息增益大。比如一个变量有2个值,各1/2,另一个变量为3个值,各为1/3,其实他们都是完全不确定的变量,但是取3个值的比取2个值的信息增益大。如果校正这个问题呢?
c) ID3算法对于缺失值的情况没有做考虑
d) 没有考虑过拟合的问题
ID3 算法的作者昆兰基于上述不足,对ID3算法做了改进,这就是C4.5算法,也许你会问,为什么不叫ID4,ID5之类的名字呢?那是因为决策树太火爆,他的ID3一出来,别人二次创新,很快 就占了ID4, ID5,所以他另辟蹊径,取名C4.0算法,后来的进化版为C4.5算法。下面我们就来聊下C4.5算法
决策树C4.5算法的改进
对于第一个问题,不能处理连续特征, C4.5的思路是将连续的特征离散化。比如m个样本的连续特征A有m个,从小到大排列为
a
1
,
a
2
,
.
.
.
,
a
m
a1,a2,...,am
a1,a2,...,am,则C4.5取相邻两样本值的平均数,一共取得m-1个划分点,其中第i个划分点Ti表示为:
T
i
=
a
i
+
a
i
+
1
2
Ti=\frac{a_i+a_{i+1}}2
Ti=2ai+ai+1。对于这
m
−
1
m-1
m−1个点,分别计算以该点作为二元分类点时的信息增益。选择信息增益最大的点作为该连续特征的二元离散分类点。比如取到的增益最大的点为at,则小于at的值为类别1,大于at的值为类别2,这样我们就做到了连续特征的离散化。要注意的是,与离散属性不同的是,如果当前节点为连续属性,则该属性后面还可以参与子节点的产生选择过程。
对于第二个问题,信息增益作为标准容易偏向于取值较多的特征的问题。我们引入一个信息增益比的变量IR(X,Y),它是信息增益和特征熵的比值。表达式如下:
I
R
(
D
,
A
)
=
I
(
A
,
D
)
H
A
H
A
=
−
∑
i
=
1
n
D
i
D
l
o
g
D
i
D
其
中
n
为
特
征
A
的
类
别
数
,
D
i
为
特
征
A
的
第
i
个
取
值
对
应
的
样
本
个
数
。
D
为
样
本
个
数
。
I_R(D,A)=\frac{I(A,D)}{H_A} \\ H_A =−∑_{i=1}^n\frac{D_i}Dlog\frac{D_i}D\\ 其中n为特征A的类别数, D_i为特征A的第i个取值对应的样本个数。D为样本个数。
IR(D,A)=HAI(A,D)HA=−i=1∑nDDilogDDi其中n为特征A的类别数,Di为特征A的第i个取值对应的样本个数。D为样本个数。
特征数越多的特征对应的特征熵越大,它作为分母,可以校正信息增益容易偏向于取值较多的特征的问题。
对于第三个缺失值处理的问题,主要需要解决的是两个问题,一是在样本某些特征缺失的情况下选择划分的属性,二是选定了划分属性,对于在该属性上缺失特征的样本的处理。
对于第一个子问题,对于某一个有缺失特征值的特征A。C4.5的思路是将数据分成两部分,对每个样本设置一个权重(初始可以都为1),然后划分数据,一部分是有特征值A的数据D1,另一部分是没有特征A的数据D2. 然后对于没有缺失特征A的数据集D1来和对应的A特征的各个特征值一起计算加权重后的信息增益比,最后乘上一个系数,这个系数是无特征A缺失的样本加权后所占加权总样本的比例。
对于第二个子问题,可以将缺失特征的样本同时划分入所有的子节点,不过将该样本的权重按各个子节点样本的数量比例来分配。比如缺失特征A的样本a之前权重为1,特征A有3个特征值A1,A2,A3。 3个特征值对应的无缺失A特征的样本个数为2,3,4.则a同时划分入A1,A2,A3。对应权重调节为2/9,3/9, 4/9。
决策树C4.5算法的不足与思考
C4.5虽然改进或者改善了ID3算法的几个主要的问题,仍然有优化的空间。
1)由于决策树算法非常容易过拟合,因此对于生成的决策树必须要进行剪枝。剪枝的算法有非常多,C4.5的剪枝方法有优化的空间。思路主要是两种,一种是预剪枝,即在生成决策树的时候就决定是否剪枝。另一个是后剪枝,即先生成决策树,再通过交叉验证来剪枝。后面在下篇讲CART树的时候我们会专门讲决策树的减枝思路,主要采用的是后剪枝加上交叉验证选择最合适的决策树。
2)C4.5生成的是多叉树,即一个父节点可以有多个节点。很多时候,在计算机中二叉树模型会比多叉树运算效率高。如果采用二叉树,可以提高效率。
3)C4.5只能用于分类,如果能将决策树用于回归的话可以扩大它的使用范围。
4)C4.5由于使用了熵模型,里面有大量的耗时的对数运算,如果是连续值还有大量的排序运算。如果能够加以模型简化可以减少运算强度但又不牺牲太多准确性的话,那就更好了。