信息增益，信息增益率，Gini

最新推荐文章于 2024-03-11 15:14:33 发布

转载最新推荐文章于 2024-03-11 15:14:33 发布 · 639 阅读

信息增益（Information Divergence）专栏收录该内容

1 篇文章

订阅专栏

本文介绍决策树中的三种特征选择方法：信息增益、信息增益率及Gini指数，并通过实例详细解析每种方法的计算过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在看Random Forest的时候，最优特征选择的常用方法查了下一般有三种：

信息增益（Info Gain）用于ID3，Gini用于CART，信息增益率（Info Gain Ratio）用于C4.5。提到前两个指标的计算时，首先要讲到的是关于熵（Entropy）的计算。

1、熵（Entropy）

理论上来说用于决策树的属性选择函数，为方便计算，往往是定义为其属性的不纯性度量，那么必须满足如下三个条件：

当结点很纯时，其度量值应为0
当不纯性最大时（比如所有类都有同样的可能），其度量值应最大
度量应该服从多级特性，这样决策树才能分阶段建立起来
$measure([2,3,4])=measure([2,7])+frac79timesmeasure([3,4])$

而熵（Entropy）能够满足以上三点特性。熵（Entropy）是由“信息论之父”香农提出的，更多的各种历史、数学理论请查看参考[1]。接下来，看看熵的计算公式如下：

e n t r o p y (p 1, p 2, \dots, p n) = - p 1 l o g 2 (p 1) - p 2 l o g 2 (p 2) - \dots - p n l o g 2 (p n)

其中，( p_i )为比例值。其实，熵也可以用另外一种意思来解释：

Given a probability distribution, the info required to predict an event is the distribution’s entropy. Entropy gives the information required in bits (this can involve fractions of bits!)

可以简单的理解为“熵”描述了用来预测的信息位数。接下来看个例子：

如下表所述的天气数据，学习目标是预测Play or not play?

表1 天气预报数据集例子

Outlook	Temperature	Humidity	Windy	Play?
sunny	hot	high	false	no
sunny	hot	high	true	no
overcast	hot	high	false	yes
rain	mild	high	false	yes
rain	cool	normal	false	yes
rain	cool	normal	true	no
overcast	cool	normal	true	yes
sunny	mild	high	false	no
sunny	cool	normal	false	yes
rain	mild	normal	false	yes
sunny	mild	normal	true	yes
overcast	mild	high	true	yes
overcast	hot	normal	false	yes
rain	mild	high	true	no

共14个实例，9个正例（yes），5个负例（no）。

这样当前数据的信息量（原始状态）用熵来计算就是：

info(play?)=info([9,5])=entropy(frac914,frac514) =–frac914log2(frac914)–frac514log2(frac514)=0.410+0.530=0.940

有了上面计算熵（信息）的基础，接下来看信息增益了。

2、信息增益（Info Gain）

信息增益，按名称来理解的话，就是前后信息的差值，在决策树分类问题中，即就是决策树在进行属性选择划分前和划分后的信息差值，即可以写成：

g a i n () = i n f o b e f o r e S p l i t () - i n f o a f t e r S p l i t ()

如上面的例子，通过使用Outlook属性来划分成下图：

图1 使用Outlook属性划分决策树

在划分后，可以看到数据被分成三份，则各分支的信息计算如下：

info([2,3])=−frac25log2(frac25)–frac35log2(frac35)=0.971bits

info([4,0])=−frac44log2(frac44)–frac04log2(frac04)=0bits

，此处虽然( log_2(0) )没有定义，但( 0 times log_2(0) )仍然计算为0。

info([3,2])=−frac35log2(frac35)–frac25log2(frac25)=0.971bits

因此，划分后的信息总量应为：

info([2,3],[4,0],[3,2])=frac514times0.971+frac414times0+frac514times0.971=0.693bits

如果把上面的公式整理到一起的话，就是：

$$ info([2,3], [4,0], [3,2]) = frac{5}{14}times info([2,3]) + frac{4}{14}times info([4,0]) + frac{5}{14}times info([3,2])

= 0.347 + 0 + 0.347 = 0.694 bits $$

从上面可以看出，划分后的信息总量计算公式为：

info(S1,…,Sn)=sumnifrac|Si||S|timesentropy(Si)

其中，n表示划分后的分支数目，( |S| )表示划分前的实例个数，( |S_i| )表示划分后某个分支的实例个数。

最后，信息增益

g a i n (o u t l o o k | p l a y ?) = i n f o b e f o r e S p l i t () - i n f o a f t e r S p l i t () = i n f o (p l a y ?) - i n f o ([3, 2], [4, 0], [3, 2]) = 0.940 - 0.694 = 0.246 b i t s

通过划分其他属性，可得到其他的树如下：

图2 使用其他属性划分决策树

同样计算，

gain(Temperature|play?)=info([9,5])–info([2,2],[4,2],[3,1]) =0.940–(frac414timesinfo([2,2])+frac614timesinfo([4,2])+frac414timesinfo([3,1]))=0.028bits

g a i n (H u m i d i t y | p l a y ?) = 0.152 b i t s

g a i n (W i n d y | p l a y ?) = 0.048 b i t s

这样，算法会选择最大的信息增益属性来进行划分，在此处为Outlook属性。接下来在Outlook=sunny结点，按照同样的判断方法，选择其他属性进行划分，

图3 继续划分决策树

计算如下：

i n f o b e f o r e S p l i t () = i n f o ([2, 3]) = 0.971 b i t s

2个正例，3个负例，计算其熵

g a i n (T e m p e r a t u r e | O u t l o o k, P l a y ?) = 0.571 b i t s

g a i n (H u m i d i t y | O u t l o o k, P l a y ?) = 0.971 b i t s

g a i n (W i n d a y | O u t l o o k, P l a y ?) = 0.020 b i t s

因而可以得到下一个划分结点是Humidity属性。在这里的Temperature有三个值，因而其info([2,2,1])使用简便方法计算为：

info([0,2],[1,1],[1,0])=frac25timesinfo([0,2])+frac25timesinfo([1,1])+frac15timesinfo([1,0])=0.4bits

因此( gain(Temperature|Outlook, Play?) = 0.971 – 0.4 = 0.571 bits )。另外，由于要计算的是信息增益最大，在划分前信息总量( info_{beforeSplit}() )一定的情况下，我们完全可以直接求划分后信息量最小的特性即可。

3、信息增益率（Info Gain Ratio）

通过上面的例子，想必对于信息增益的计算步骤已经熟悉些了。那么下面来看信息增益存在的一个问题：假设某个属性存在大量的不同值，如ID编号（在上面例子中加一列为ID，编号为a~n），在划分时将每个值成为一个结点，这样就形成了下面的图：

图4 信息增益偏向

最终计算得到划分后的信息量为：( info(S_1, …, S_n) = sum_{i=1}^n{frac{|S_i|}{|S|}times entropy(S_i)} = 0 )，因为( entropy(S_i) = info([1,0]) ) 或 ( info([0,1]) )，只含一个纯结点。这样决策树在选择属性时，将偏向于选择该属性，但这肯定是不正确（导致过拟合）的。因此有必要使用一种更好的方法，那就是C4.5中使用的信息增益率（Info Gain Ratio），其考虑了分支数量和尺寸的因素，使用称为内在信息（Intrinsic Information）的概念。

Gain ratio takes number and size of branches into account when choosing an attribute, and corrects the information gain by taking the intrinsic information of a split into account (i.e. how much info do we need to tell which branch an instance belongs to).

内在信息（Intrinsic Information），可简单地理解为表示信息分支所需要的信息量，其计算公式如下：

IntrinsicInfo(S,A)=−sumfrac|Si||S|log2frac|Si||S|

则针对上图中的例子，

info([1,1,…,1])=14times(−frac114log2frac114)=3.807bits

信息增益率则计算如下：

gainratio(Attribute)=fracgain(Attribute)IntrinsicInfo(Attribute)

依然如上例：

gainratio(IDCode)=frac0.940bits3.807bits=0.246

实际上可以看出，属性的重要性会随着其内在信息（Intrinsic Information）的增大而减小。信息增益率作为一种补偿（Compensate）措施来解决信息增益所存在的问题，但是它也有可能导致过分补偿，而选择那些内在信息很小的属性，这一点可以尝试：首先，仅考虑那些信息增益超过平均值的属性，其次再比较信息增益。

4、Gini Index

在CART里面划分决策树的条件是采用Gini Index，定义如下：