利用随机森林对特征重要性进行评估（公式原理）_随机森林特征重要性

本文参考来源于：
杨凯, 侯艳, 李康. 随机森林变量重要性评分及其研究进展[J]. 2015.

码字不易，各位看官大大的赞是我更细的动力！

一、引言

随机森林（$randomforest，RF$）由$Breiman$等人在2001年提出。

$RF$具有很高的预测准确率，对异常值和噪声有很强的容忍度，能够处理高维数据（变量个数远大于观测个数），有效地分析非线性、具有共线性和交互作用的数据， 并能够在分析数据的同时给出变量重要性评分（$variableimportancemeasures，VIM$）。这些特点使得$RF$特别适用于高维组学数据的研究，即在对疾病进行分类的同时通过$VIM$值筛选出潜在的生物标志物，研究疾病发生、发展的生物学机制。然而， 由于在实际中由RF筛选变量使用的统计量不同，可能会使结果有较大的差异，为此本文在简单介绍$RF$筛选变量的基础上，拟针对目前随机森林变量重要性评分的不同计算方法和新近提出的改进方法， 说明其在高维组学数据分析中的应用。

二、随机森林的基本原理

了解决策树的算法，那么随机森林是相当容易理解的。随机森林的算法可以用如下几个步骤概括：

1.用有抽样放回的方法（bootstrap）从样本集中选取$n$个样本作为一个训练集；

2.用抽样得到的样本集生成一棵决策树。在生成的每一个结点：

（1）随机不重复地选择$d$个特征

（2）利用这$d$个特征分别对样本集进行划分，找到最佳的划分特征（可用基尼系数、增益率或者信息增益判别）

3.重复步骤1到步骤2共$k$次，$k$即为随机森林中决策树的个数。

4.用训练得到的随机森林对测试样本进行预测，并用投票法决定预测的结果。

下图比较直观地展示了随机森林算法：

图片来源：
杨凯, 侯艳, 李康. 随机森林变量重要性评分及其研究进展[J]. 2015.

$RF$中的每一棵分类树为二叉树，根节点包含全部训练自助样本，按照一定的原则，在每个节点从一组随机选取的变量中选择使分枝后节点“不纯度” 最小的变量作为分枝变量，分裂为左节点和右节点， 它们分别包含训练数据的一个子集， 分裂后的节点按照同样规则继续分裂， 直到满足分枝停止规则而停止生长， 具体过程见图 1。

“不纯度” 的衡量标准包括$Gini$不纯度、 熵和错误率等。 变量筛选使用$VIM$统计量。

三、随机森林常规的变量重要性评分

现假定有变量$X_1,X_2,L,X_M$，需要计算出$M$个$VIM$得分统计量。

$RF$常规的$VIM$计算方法分为两种，即根据$Gini$指数和袋外数据（$OOB$） 错误率计算得到， 变量$X_j$的得分统计量分别用$VI{M}_j^{(Gini)}$和$VI{M}_j^{(OOB)}$表示。

2.1 Gini指数

统计量$VI{M}_j^{(Gini)}$表示第$j$个变量在$RF$所有树中节点分裂不纯度的平均改变量。$Gini$指数的计算公式为:

$$G{I}_m=\sum _{k=1}^K{\hat{p}}_{mk}(1-{\hat{p}}_{mk})(1)$$

$K$为自助样本集的类别数，${\hat{p}}_{mk}$为节点$m$样本属于第$k$类的概率估计值，当样本为二分类数据时（$K=2$） ，节点$m$的$Gini$指数为:

$$G{I}_m=2{\hat{p}}_m(1-{\hat{p}}_m)(2)$$

${\hat{p}}_m$为样本在节点$m$属于任意一类的概率估计值。

变量$X_j$在节点$m$的重要性，即节点$m$分枝前后$Gini$指数变化量为:

$$VI{M}_{jm}^{(Gini)}=G{I}_m-G{I}_l-G{I}_r(3)$$

$G{I}_l$和$G{I}_r$分别表示由节点$m$分裂的两新节点的$Gini$指数。

如果变量$X_j$在第$i$棵树中出现$M$次，则变量$X_j$在第$i$棵树的重要性为:

$$VI{M}_{ij}^{(Gini)}=\sum _{m=1}^{M}VI{M}_{jm}^{(Gini)}(4)$$

变量$X_j$在$RF$中的$Gini$重要性定义为:

$$VI{M}_j^{(Gini)}=\frac{1}{n}\sum _{i=1}^{n}VI{M}_{ij}^{(Gini)}(5)$$

其中，$n$为$RF$中分类树的数量。

2.2 OOB 错误率

$VI{M}_j^{(OOB)}$的定义为:在$RF$的每棵树中，使用随机抽取的训练自助样本建树，并计算袋外数据（$OOB$）的预测错误率，然后随机置换变量$X_j$的观测值后再次建树并计算$OOB$的预测错误率，最后计算两次$OOB$错误率的差值经过标准化处理后在所有树中的平均值即为变量$X_j$的置换重要性($VI{M}_j^{(OOB)}$)。

变量$X_j$在第$i$棵树的$VI{M}_j^{(OOB)}$为：

$$VI{M}_j^{(OOB)}=\frac{{\sum }_{p=1}^{n_o^i}I(Y_p=Y_p^i)}{n_o^i}-\frac{{\sum }_{p=1}^{n_o^i}I(Y_p=Y_{p,{\pi }_j}^i)}{n_o^i}$$

其中，$n_o^i$为第$i$棵树$OOB$数据的观测例数，$I(g)$为指示函数，即两值相等时取1，不等时取0； Y p ∈ { 0 , 1 } Y_p\in\{0,1\} Yp​∈{0,1}为第$p$个观测的真实结果， Y p i ∈ { 0 , 1 } Y_p^i\in\{0,1\} Ypi​∈{0,1}为随机置换前第$i$棵树对$OOB$数据第$p$个观测的预测结果， Y p , π j i ∈ { 0 , 1 } Y_{p,\pi_j}^i\in\{0,1\} Yp,πj​i​∈{0,1}为随机置换后第$i$棵树对$OOB$数据第$p$个观测的预测结果。

当变量$j$没有在第$i$棵树中出现时，$VI{M}_{ij}^{(OOB)}=0$

变量$X_j$在$RF$中的置换重要性定义为:

$$VI{M}_j^{(OOB)}=\frac{{\sum }_{i=1}^{n}VI{M}_{ij}^{(OOB)}}{n}$$

其中，$n$为$RF$中分类树的数量。

2.3 常规变量重要性评分的优缺点

$VI{M}_j^{(Gini)}$在数据挖掘中估计变量重要性时有着广泛的应用。 当变量为连续型变量且互不相关时，$VI{M}_j^{(Gini)}$的估计是无偏的；

当信噪比较低时，$VI{M}_j^{(Gini)}$的准确性也高于$VI{M}_j^{(OOB)}$，因此，$VI{M}_j^{(Gini)}$有更高的稳定性。

然而， 当同时存在连续变量和分类变量或者分类变量的水平数不同时，$VI{M}_j^{(Gini)}$估计则不够准确。由于$K$个水平的分类变量在节点处可以有$2^{K-1}-1$种分割， 连续变量在节点处可以有$b-1$种分割（$b$为样本量）。在备选分割较多时， 即使变量没有分类作用， 也可能使$Gini$指数降低， 从而更容易被选为分枝变量， 即$VI{M}_j^{(Gini)}$被高估。 实际上， 由于$Gini$指数的计算问题， 水平数多的分类变量的$Gini$指数降低会大于水平数少的分类变量。

当分类变量水平数相同时， 如果不同变量的水平间差别不同， 容易过高估计水平间差别大的变量的$VI{M}_j^{(Gini)}$ 。 在 SNP 数据分析中， Nicodemus 等人指出当所有 SNP 都不具有分类能力时，不相关 SNP 的$VI{M}_j^{(Gini)}$高于高度相关 SNP 的$VI{M}_j^{(Gini)}$

事实上，$VI{M}_j^{(OOB)}$在实际中的应用范围更加广泛。 由于$VI{M}_j^{(OOB)}$是通过$OOB$数据计算的， 因此可以看作变量具有的分类能力， 没有分类能力的变量在观测值置换前后的$OOB$错误率不会发生改变， 即数学期望$E(VI{M}_j^{(OOB)})=0$，此外，$VI{M}_j^{(OOB)}$不仅能够衡量变量单独的分类能力， 还能够衡量变量间交互作用的分类能力。

当同时存在连续变量和分类变量或者分类变量水平数不同时， 并不会影响$VI{M}_j^{(OOB)}$准确性。$VI{M}_j^{(OOB)}$是通过随机置换变量观测值前后$OOB$错误率的差值计算的， 即使没有分类作用， 水平数多的变量也更容易被选为分枝变量， 但置换前后并不会影响$OOB$错误率，同时会使$VI{M}_j^{(OOB)}$的变异增大。

在分类数据不平衡时，$VI{M}_j^{(OOB)}$的准确性会受到一定影响。例如多数为正常人， 无论变量是否置换， 大部分的数据都会被预测为正常， 虽然$OOB$错误率受到的影响可能不大，但却会严重低估所有变量的$VI{M}_j^{(OOB)}$。

当单棵树的预测准确率较低时（如$OOB$错误率达到 50%） ， 会低估变量的$VI{M}_j^{(OOB)}$

当置换前$OOB$错误率已经很大时， 置换变量观测值使得 OOB 错误率变大的可能性降低， 从而低估变量的$VI{M}_j^{(OOB)}$

当没有分类能力的变量与有分类能力的变量相关时， 可能低估相关有分类能力变量的$VI{M}_j^{(OOB)}$，且估计方差变异增加， 而此时没有分类能力变量的$VI{M}_j^{(OOB)}$则被高估。

组学数据中存在大量的噪声变量， 当变量数目巨大而具有分类能力的变量所占比例很小时，$RF$建模容易受到大量噪声变量的干扰， 使变量的$VIM$计算受到影响， 变量筛选的结果不可信。

2.4 变量重要性（ VIM ） 的显著性检验

$VIM$给出了变量的重要性排序， 但无法在变量筛选时给出显著性假设检验的阈值。 事实上， 当所有变量都没有分类能力时，$VIM$也会给出变量重要性排序， 筛选出不具有分类能力的变量； 而当有分类能力的变量很多时， 仅通过变量的排序进行变量筛选， 可能漏掉一些具有分类能力的变量。

$Breiman$等人提出通过$Z$值直接计算$VI{M}_j^{(OOB)}$的显著性， 即:

$$Z=VI{M}_j^{(OOB)}/(\hat{\sigma }/\sqrt{n})$$

其中，$\hat{\sigma }$为$RF$中各棵树$VI{M}_j^{(OOB)}$的标准差，$n$为随机森林（$RF$）中树的数量。 然而，$Strobl$等人通过模拟实验表明：$Z$值的检验效能与样本量$b$及树的数量$n$有很大的关系， 当样本量$b$很大而树的数量$n$小时， 检验效能接近于0。