【机器学习】决策树学习下篇（详解）

X.AI666

已于 2024-04-02 19:22:14 修改

阅读量3.8k

点赞数 93

分类专栏：机器学习文章标签：机器学习

于 2024-03-24 08:00:00 首次发布

本文链接：https://blog.csdn.net/chen695969/article/details/136941271

版权

机器学习专栏收录该内容

27 篇文章 10 订阅

订阅专栏

本文探讨了决策树在数据驱动时代的重要性，介绍了过拟合问题及其在决策树中的表现。详细讲解了预剪枝和后剪枝技术，包括实施策略、优缺点及实际应用。同时，讨论了决策树在处理连续属性、缺失值和成本敏感问题的方法，以及其在扩展学习框架如随机森林和GBDT中的应用。

摘要由CSDN通过智能技术生成

引言

在当今数据驱动的时代，机器学习技术已成为解决复杂问题不可或缺的工具。其中，决策树学习作为一种基础且强大的算法，广泛应用于各种领域，包括但不限于金融风控、医疗诊断、客户关系管理等。决策树以其简单直观、易于理解和实现的特点，受到了数据科学家和业界专家的青睐。

过拟合问题

在决策树学习中，过拟合是一个经常遇到的问题。过拟合发生时，模型在训练数据上表现得非常好，但在新的、未见过的数据上表现较差。这是因为模型学习到了训练数据中的噪声，而不仅仅是底层的数据分布。过拟合问题的一个极端例子是决策树模型将每个训练样本都完美分类，每个叶节点对应单个训练样本，这种情况下的决策树仅仅是实现了一个对训练数据的查表，而对新数据几乎没有预测能力。

如何避免过拟合

预剪枝（Pre-pruning）和后剪枝（Post-pruning）是决策树算法中两种主要的剪枝技术，用于防止模型过拟合，提高模型的泛化能力。下面详细介绍这两种技术的原理、方法及其优缺点。

预剪枝（Pre-pruning）

预剪枝是指在决策树生成过程中，提前停止树的增长的一种策略。其核心思想是在树完全生成之前就停止其进一步分裂，以防止模型复杂度过高，避免过拟合。

实施策略

基于样本数的停止分裂：设定一个阈值，当到达某个节点的训练样本数量少于这个阈值时，停止分裂。这个策略的基本假设是，基于过少的数据样本做出的决策可能会引入更多的误差和泛化错误。
基于信息增益的停止分裂：在每次分裂前计算分裂带来的信息增益，如果这个增益小于某个预设的阈值，则停止分裂。这种方法利用了所有的训练数据，并允许叶节点存在于树的任何层级。

优缺点

优点：计算效率高，因为它阻止了树的进一步增长，减少了模型的复杂度。
缺点：可能因为过早停止树的增长而导致模型欠拟合，从而影响模型的准确率。

后剪枝（Post-pruning）

后剪枝是在决策树完全生成之后，通过去除一些子树或者节点来简化模型的一种方法。相比预剪枝，后剪枝允许树完全生长，然后再评估是否有必要进行剪枝。

实施策略

错误降低剪枝（Error Reduction Pruning）：将数据集分为训练集和验证集。使用验证集来评估剪枝前后的模型性能。如果剪枝能够在验证集上减少错误，则进行剪枝。这种方法是贪心算法，每次尝试剪去每个可能的节点（以及其子树），保留那些能够提高验证集准确率的剪枝。
规则后剪枝（Rule Post-Pruning）：首先将决策树转换为一组规则，然后对每条规则进行评估和简化。规则简化的过程中，去除某些条件，如果这样做能够提高规则的准确性，则保留简化后的规则。

优缺点

优点：通常能够得到更加精确的模型。因为它允许树完全生长，然后再基于实际的模型性能来进行剪枝，有更大的空间找到最优的模型结构。
缺点：计算开销较大，特别是对于大型数据集，因为需要完整地构建树模型，然后再进行复杂的剪枝操作。

总结来说，预剪枝通过提前停止树的增长来防止过拟合，具有实施简单和计算高效的优点；而后剪枝则是允许树完全生长后，通过剪枝优化模型，虽然计算复杂度高，但通常能够得到更好的模型性能。在实践中，选择哪种剪枝技术取决于具体问题的需求、数据集的大小和计算资源的限制。

实际场景中的决策树学习

决策树在解决实际问题时，经常会遇到一些特殊情况，如连续属性值的处理、具有过多取值的属性、未知（缺失）属性值处理，以及有代价的属性等。对于连续属性，一种常见的处理方法是通过确定一个阈值将其离散化。而对于具有过多取值的属性，可以采用信息增益比来减少偏差。对于缺失属性值，一种策略是根据数据中该属性的分布来推断缺失值。对于属性获取成本高的情况，可以通过考虑属性的代价来优化决策树的构建过程。