1. ID3 算法的核心思想
ID3(Iterative Dichotomiser 3)是一种 基于信息增益的决策树生成算法。它的目标是:
从给定训练数据中,选择最佳特征作为节点划分,从而生成一棵尽可能高效的决策树。
它的步骤概括如下:
-
计算当前数据集的熵 (Entropy):
H(D) = - \\sum_{i=1}\^{n} p_i \\log_2 p_i
其中 (p_i) 是第 i 类样本占比。熵越高,表示数据集越混乱。
-
计算每个特征的信息增益 (Information Gain):
Gain(D, A) = H(D) - \\sum_{v \\in Values(A)} \\frac{\|D_v\|}{\|D\|} H(D_v)
其中 (D_v) 是按特征 A 的取值 v 划分后的子集。
-
选择信息增益最大的特征作为当前节点划分特征。
-
递归构建子树,直到:
- 子集属于同一类别(纯净),或者
- 没有特征可用,则取多数类别作为叶节点。
2. 为什么可以优化决策树
"优化"主要体现在 减少树的复杂度,提高分类效率:
-
优先划分"最能区分类别"的特征:
- ID3 通过信息增益衡量特征的重要性。
- 每次选择信息增益最大的特征,保证每次划分都最大程度减少数据集的不确定性。
- 这意味着决策树会尽可能 快速到达叶子节点,减少树的深度,提高分类速度。
-
避免不必要的分支:
- 当一个节点的数据已经纯净(所有样本属于同一类),ID3 就停止划分,不会继续创建分支。
- 这避免了生成冗余节点,优化了树的结构。
-
自顶向下贪心策略:
- ID3 使用 贪心算法,每次都选择当前最优划分特征。
- 虽然不保证全局最优,但在大多数情况下能生成 较短且信息效率高的树。
3. 局限性与进一步优化
ID3 虽然在构建树时优化了信息增益,但存在问题:
- 过拟合:信息增益偏向取值多的特征,可能生成过深的树。
- 缺乏剪枝机制:树生成后仍可能包含冗余分支。
解决方法:
- 改进算法,如 C4.5(使用信息增益比)
- 后剪枝策略(Post-pruning)减少过拟合
✅ 总结
ID3 算法优化决策树的核心原因在于:
- 信息增益驱动:每次选择最能区分类别的特征,减少树的不确定性。
- 贪心递归构建:快速生成高效结构,尽可能少的分支和深度。
- 提前停止划分:避免冗余节点,提高分类效率。
简单理解:ID3 用"最优划分特征优先"的策略,让树更短、更准确、更高效。