图解机器学习 | 回归树模型详解

本文涉及的产品
模型训练 PAI-DLC,5000CU*H 3个月
模型在线服务 PAI-EAS,A10/V100等 500元 1个月
交互式建模 PAI-DSW,每月250计算时 3个月
简介: 用于回归任务的决策树称作回归树,属性选择与生长方式与分类决策树不同。本文讲解决策树回归算法的核心思想、启发式切分、最优属性选择、过拟合、正则化、以及缺失值处理等关键知识点。

ShowMeAI研究中心

作者:韩信子@ShowMeAI
教程地址http://www.showmeai.tech/tutorials/34
本文地址http://www.showmeai.tech/article-detail/192
声明:版权所有,转载请联系平台与作者并注明出处
收藏 ShowMeAI 查看更多


引言

大家在前面的部分学习到了使用决策树进行分类,实际决策树也可以用作回归任务,我们叫作回归树。而回归树的结构还是树形结构,但是属性选择与生长方式和分类的决策树有不同,我们一起来看看它的原理知识吧。

(本篇回归树模型部分内容涉及到机器学习基础知识、决策树算法,没有先序知识储备的宝宝可以查看ShowMeAI的文章 图解机器学习 | 机器学习基础知识决策树模型详解)。

1.决策树回归算法核心思想

1)决策树结构回顾

我们一起来回顾一下决策树的结构,决策树的典型结构如下图所示

回归树模型详解; 决策树回归算法; 结构回顾; 8-1

决策树的学习过程预测过程如下图所示。详细内容可以参考ShowMeAI的文章 决策树模型详解

回归树模型详解; 决策树回归算法; 结构回顾; 8-2

主流的决策树算法有

  • ID3:基于信息增益来选择分裂属性(每步选择信息增益最大的属性作为分裂节点,树可能是多叉的)。
  • C4.5:基于信息增益率来选择分裂属性(每步选择信息增益率最大的属性作为分裂节点,树可能是多叉的)。
  • CART:基于基尼系数来构建决策树(每步要求基尼系数最小,树是二叉的)。

    • 其中:CART树全称Classification And Regression Tree,即可以用于分类,也可以用于回归,这里指的回归树就是 CART 树,ID3和C4.5不能用于回归问题。

2)回归树的核心思想

要讲回归树,我们一定会提到CART树,CART树全称Classification And Regression Trees,包括分类树与回归树

CART的特点是:假设决策树是二叉树,内部结点特征的取值为「是」和「否」,右分支是取值为「是」的分支,左分支是取值为「否」的分支。这样的决策树等价于「递归地二分每个特征」,将输入空间(特征空间)划分为有限个单元,并在这些单元上确定预测的概率分布,也就是在输入给定的条件下输出的条件概率分布。

设有数据集 equation?tex=D,构建回归树的大体思路如下:

  • ① 考虑数据集 equation?tex=D 上的所有特征 equation?tex=j,遍历每一个特征下所有可能的取值或者切分点 equation?tex=s,将数据集 equation?tex=D 划分成两部分 equation?tex=D_1equation?tex=D_2
  • ② 分别计算 equation?tex=D_1equation?tex=D_2 的平方误差和,选择最小的平方误差对应的特征与分割点,生成两个子节点(将数据划分为两部分)。
  • ③ 对上述两个子节点递归调用步骤 ① ②,直到满足停止条件。

回归树构建完成后,就完成了对整个输入空间的划分(即完成了回归树的建立)。将整个输入空间划分为多个子区域,每个子区域输出为该区域内所有训练样本的平均值。

回归树模型详解; 回归树的核心思想; CART示例1; 8-3

回归树模型详解; 回归树的核心思想; CART示例2; 8-4

我们知道了回归树其实是将输入空间划分为 equation?tex=M 个单元,每个区域的输出值是该区域内所有点 equation?tex=y 值的平均数。但我们希望构建最有效的回归树:预测值与真实值差异度最小。下面部分我们展开讲讲,回归树是如何生长的。

2.启发式切分与最优属性选择

1)回归树模型示例

我们用一个经典的棒球案例来解释回归树:根据从业年限和表现,去预估棒球运动员的工资。如下所示,有 equation?tex=1987 个数据样本,包含 equation?tex=322 个棒球运动员。

  • 红黄表示高收入,蓝绿表示低收入。
  • 横坐标是年限,纵坐标是表现。

回归树模型详解; 回归树模型示例; 经典的棒球案例; 8-5

这个简单案例中,每个样本数据有两个特征:从业年限 equation?tex=Years和成绩表现 equation?tex=Hits,回归树的决策过程由最终生成的回归树决定,如右图所示:

  • 根决策节点为特征 equation?tex=Years,其划分阈值为 equation?tex=4.5equation?tex=Years 小于 equation?tex=4.5 的样本划分到左边,大于或等于 equation?tex=4.5 的样本划分到右边;
  • 第二个决策节点的特征为 equation?tex=Hits,其划分阈值为 equation?tex=117.5equation?tex=Hits 小于 equation?tex=117.5 的样本划分到左边,大于或等于 equation?tex=117.5 的样本划分到右边。
  • 一个样本顺着决策树的决策条件,走到叶子节点,即可获得预测工资,这里的预测工资总共就 equation?tex=3 种取值,分别为 equation?tex=5.11equation?tex=6.00equation?tex=6.74

回归树模型详解; 回归树模型示例; 经典的棒球案例; 8-6

我们来深入拆解和对应一下,其实回归树构建完成后,实现了对整个空间的划分(如下图所示)。实际预测时,新样本会按照回归树的决策过程,被划分到下图 equation?tex=R_1equation?tex=R_2equation?tex=R_3 之中的一个区域 equation?tex=R_i,而这个新样本的预测值(本案例中为棒球运动员的工资)就是它所在的区域。

  • equation?tex=R_i 中所有训练样本的工资平均值。

回归树模型详解; 回归树模型示例; 经典的棒球案例; 8-76

回归树背后的含义:对空间的划分。整个平面被划分成 equation?tex=3 部分:

equation?tex=R1%20%3D%20%7BX%20%7CYears%20%3C%204.5%7D

equation?tex=R2%20%3D%20%7BX%20%7CYears%20%E2%89%A5%204.5%2C%20Hits%20%3C%20117.5%7D

equation?tex=R3%20%3D%20%7BX%20%7CYears%20%E2%89%A5%204.5%2C%20Hits%20%E2%89%A5%20117.5%7D

2)回归树构建方法

下面切到回归树构建的核心:切分方式与属性选择。

假设一回归问题,预估结果 equation?tex=y%20%5Cin%20R,特征向量为 equation?tex=X%20%3D%20%5Bx_1%2Cx_2%2Cx_3%2C%20%5Cdots%20%2C%20x_p%20%5D,回归树 equation?tex=2 个步骤是:

  • ① 把整个特征空间 equation?tex=X 切分成 equation?tex=J 个没有重叠的区域 equation?tex=R_1%2CR_2%2CR_3%2C%20%5Cdots%20%2CR_J
  • ② 其中区域 equation?tex=R_J 中的每个样本我们都给一样的预测结果 equation?tex=%5Ctilde%7By%7D_%7BR_%7Bj%7D%7D%3D%5Cfrac%7B1%7D%7Bn%7D%20%5Csum%20j%20%5Cin%20R%20j%20y_%7Bj%7D,其中 equation?tex=nequation?tex=R_J 中的总样本数。


仔细观察一下上面的过程,实际上我们希望能找到如下的 RSS 最小的化划分方式 equation?tex=R_1%2CR_2%2CR_3%2C%20%5Cdots%20%2CR_J

equation?tex=R%20S%20S%3D%5Csum_%7Bj%3D1%7D%5E%7BJ%7D%20%5Csum_%7Bi%20%5Cin%20R%20j%7D%5Cleft%28y_%7Bi%7D-%5Ctilde%7By%7D_%7BR_%7Bj%7D%7D%5Cright%29%5E%7B2%7D

回归树模型详解; 回归树构建方法; 切分方式 / 属性选择; 8-8

  • equation?tex=y :为每个训练样本的标签构成的标签向量,向量中的每个元素 equation?tex=y_j 对应的是每个样本的标签。
  • equation?tex=X :为特征的集合,equation?tex=x_1%2Cx_2%2C%20%5Cdots%20%2C%20x_p 为第 equation?tex=1 个特征到第 equation?tex=p 个特征。
  • equation?tex=R_1%2CR_2%2CR_3%2C%20%5Cdots%20%2CR_J 为整个特征空间划分得来的J个不重叠的区域(可以参考上页的右图)。
  • equation?tex=%5Ctilde%7By%7D_%7BR_%7Bj%7D%7D :为划分到第 equation?tex=j 个区域 equation?tex=R_j 的样本的平均标签值,用这个值作为该区域的预测值,即如果有一个测试样本在测试时落入到该区域,就将该样本的标签值预测为 equation?tex=%5Ctilde%7By%7D_%7BR_%7Bj%7D%7D

但是这个最小化和探索的过程,计算量是非常非常大的。我们采用「探索式的递归二分」来尝试解决这个问题。

递归二分

回归树采用的是「自顶向下的贪婪式递归方案」。这里的贪婪,指的是每一次的划分,只考虑当前最优,而不回头考虑之前的划分。从数学上定义,即选择切分的维度(特征) equation?tex=x_j 以及切分点 equation?tex=s 使得划分后的树RSS结果最小,公式如下所示:

equation?tex=%5Cbegin%7Baligned%7D%20%26%20R_%7B1%7D%28j%2C%20s%29%3D%5Cleft%5C%7Bx%20%5Cmid%20x_%7Bj%7D%3Cs%5Cright%5C%7D%20%5C%5C%20%26%20R_%7B2%7D%28j%2C%20s%29%3D%5Cleft%5C%7Bx%20%5Cmid%20x_%7Bj%7D%20%5Cgeq%20s%5Cright%5C%7D%20%5C%5C%20%26%20RSS%3D%5Csum%20x_%7Bi%7D%20%5Cin%20R_%7B1%7D%28j%2C%20s%29%5Cleft%28y_%7Bi%7D-%5Ctilde%7By%7D_%7BR%201%7D%5Cright%29%5E%7B2%7D%2B%5Csum%20x_%7Bi%7D%20%5Cin%20R_%7B2%7D%28j%2C%20s%29%5Cleft%28y_%7Bi%7D-%5Ctilde%7By%7D_%7BR_%7B2%7D%7D%5Cright%29%5E%7B2%7D%20%5Cend%7Baligned%7D

回归树模型详解; 回归树构建方法; 切分方式 / 属性选择; 8-9

我们再来看看「递归切分」。下方有两个对比图,其中左图是非递归方式切分得到的,而右图是二分递归的方式切分得到的空间划分结果(下一次划分一定是在之前的划分基础上将某个区域一份为二)。

回归树模型详解; 回归树构建方法; 切分方式 / 属性选择; 8-10

两种方式的差别是:递归切分一定可以找到一个较优的解,非递归切分穷举不了所有情况,算法上无法实现,可能无法得到一个较好的解。

回归树模型详解; 回归树构建方法; 切分方式 / 属性选择; 8-11

回归树总体流程类似于分类树:分枝时穷举每一个特征可能的划分阈值,来寻找最优切分特征和最优切分点阈值,衡量的方法是平方误差最小化。分枝直到达到预设的终止条件(如叶子个数上限)就停止。

但通常在处理具体问题时,单一的回归树模型能力有限且有可能陷入过拟合,我们经常会利用集成学习中的Boosting思想,对回归树进行增强,得到的新模型就是提升树(Boosting Decision Tree),进一步,可以得到梯度提升树(Gradient Boosting Decision Tree,GBDT),再进一步可以升级到XGBoost。通过多棵回归树拟合残差,不断减小预测值与标签值的偏差,从而达到精准预测的目的,ShowMeAI会在后面介绍这些高级算法。

3.过拟合与正则化

1)过拟合问题

决策树模型存在过拟合风险,通常情况下,树的规模太小会导致模型效果不佳,而树的规模太大就会造成过拟合,非常难以控制。

2)过拟合问题处理

对于决策树,我们通常有如下一些策略可以用于环节过拟合:

回归树模型详解; 过拟合与正则化; 过拟合问题处理; 8-12

(1)约束控制树的过度生长

  • 限制树的深度:当达到设置好的最大深度时结束树的生长。
  • 分类误差法:当树继续生长无法得到客观的分类误差减小,就停止生长。
  • 叶子节点最小数据量限制:一个叶子节点的数据量过小,树停止生长。

(2)剪枝

约束树生长的缺点就是提前扼杀了其他可能性,过早地终止了树的生长,我们也可以等待树生长完成以后再进行剪枝,即所谓的后剪枝,而后剪枝算法主要有以下几种:

  • Reduced-Error Pruning(REP,错误率降低剪枝)。
  • Pesimistic-Error Pruning(PEP,悲观错误剪枝)。
  • Cost-Complexity Pruning(CCP,代价复杂度剪枝)。
  • Error-Based Pruning(EBP,基于错误的剪枝)。

3)正则化

对于回归树而言,在剪枝过程中我们会添加正则化项衡量。如下所示,考虑剪枝后得到的子树 equation?tex=%5Cleft%20%5C%7BT_a%20%5Cright%20%5C%7D,其中 equation?tex=%5Calpha 是正则化项的系数。当固定住 equation?tex=%5Calpha 之后,最佳的 equation?tex=T_a 就是使得下列式子值最小的子树。

equation?tex=%5Csum_%7Bm%3D1%7D%5E%7B%7CT%7C%7D%20%5Csum_%7Bx_%7Bi%7D%20%5Cin%20R_%7Bm%7D%7D%5Cleft%28y_%7Bi%7D-%5Ctilde%7By%7D_%7BR_%7B2%7D%7D%5Cright%29%5E%7B2%7D%2B%5Calpha%7CT%7C

  • equation?tex=%7CT%7C 是回归树叶子节点的个数。
  • equation?tex=%5Calpha 可以通过交叉验证去选择。

更多监督学习的算法模型总结可以查看ShowMeAI的文章 AI知识技能速查 | 机器学习-监督学习


视频教程

可以点击 B站 查看视频的【双语字幕】版本

frameLabelStart--frameLabelEnd

https://www.bilibili.com/video/BV1y44y187wN?p=12

机器学习【算法】系列教程

机器学习【实战】系列教程

ShowMeAI 系列教程推荐

ShowMeAI用知识加速每一次技术成长

相关实践学习
使用PAI-EAS一键部署ChatGLM及LangChain应用
本场景中主要介绍如何使用模型在线服务(PAI-EAS)部署ChatGLM的AI-Web应用以及启动WebUI进行模型推理,并通过LangChain集成自己的业务数据。
机器学习概览及常见算法
机器学习(Machine Learning, ML)是人工智能的核心,专门研究计算机怎样模拟或实现人类的学习行为,以获取新的知识或技能,重新组织已有的知识结构使之不断改善自身的性能,它是使计算机具有智能的根本途径,其应用遍及人工智能的各个领域。 本课程将带你入门机器学习,掌握机器学习的概念和常用的算法。
目录
相关文章
|
1月前
|
机器学习/深度学习 数据采集 数据挖掘
实战派教学:掌握Scikit-learn,轻松实现数据分析与机器学习模型优化!
【10月更文挑战第4天】Scikit-learn凭借高效、易用及全面性成为数据科学领域的首选工具,简化了数据预处理、模型训练与评估流程,并提供丰富算法库。本文通过实战教学,详细介绍Scikit-learn的基础入门、数据预处理、模型选择与训练、评估及调优等关键步骤,助你快速掌握并优化数据分析与机器学习模型。从环境搭建到参数调优,每一步都配有示例代码,便于理解和实践。
85 2
|
29天前
|
机器人
1024 云上见 使用 PAI+LLaMA Factory 微调 Qwen2-VL 模型,搭建 “文旅领域知识问答机器人” 领精美计时器
1024 云上见 使用 PAI+LLaMA Factory 微调 Qwen2-VL 模型,搭建 “文旅领域知识问答机器人” 领精美计时器
85 3
|
2月前
|
机器学习/深度学习 人工智能 分布式计算
使用PAI+LLaMA Factory 微调 Qwen2-VL 模型,搭建文旅领域知识问答机器人
本次教程介绍了如何使用 PAI ×LLaMA Factory 框架,基于全参方法微调 Qwen2-VL 模型,使其能够进行文旅领域知识问答,同时通过人工测试验证了微调的效果。
使用PAI+LLaMA Factory 微调 Qwen2-VL 模型,搭建文旅领域知识问答机器人
|
1月前
|
机器学习/深度学习 数据采集 监控
如何使用机器学习模型来自动化评估数据质量?
【10月更文挑战第6天】如何使用机器学习模型来自动化评估数据质量?
|
8天前
|
机器学习/深度学习 数据采集 监控
如何使用机器学习模型来自动化评估数据质量?
如何使用机器学习模型来自动化评估数据质量?
|
5天前
|
机器学习/深度学习 人工智能 算法
【手写数字识别】Python+深度学习+机器学习+人工智能+TensorFlow+算法模型
手写数字识别系统,使用Python作为主要开发语言,基于深度学习TensorFlow框架,搭建卷积神经网络算法。并通过对数据集进行训练,最后得到一个识别精度较高的模型。并基于Flask框架,开发网页端操作平台,实现用户上传一张图片识别其名称。
21 0
【手写数字识别】Python+深度学习+机器学习+人工智能+TensorFlow+算法模型
|
9天前
|
机器学习/深度学习 算法 PyTorch
用Python实现简单机器学习模型:以鸢尾花数据集为例
用Python实现简单机器学习模型:以鸢尾花数据集为例
28 1
|
1月前
|
数据采集 移动开发 数据可视化
模型预测笔记(一):数据清洗分析及可视化、模型搭建、模型训练和预测代码一体化和对应结果展示(可作为baseline)
这篇文章介绍了数据清洗、分析、可视化、模型搭建、训练和预测的全过程,包括缺失值处理、异常值处理、特征选择、数据归一化等关键步骤,并展示了模型融合技术。
53 1
模型预测笔记(一):数据清洗分析及可视化、模型搭建、模型训练和预测代码一体化和对应结果展示(可作为baseline)
|
18天前
|
机器学习/深度学习 数据采集 Python
从零到一:手把手教你完成机器学习项目,从数据预处理到模型部署全攻略
【10月更文挑战第25天】本文通过一个预测房价的案例,详细介绍了从数据预处理到模型部署的完整机器学习项目流程。涵盖数据清洗、特征选择与工程、模型训练与调优、以及使用Flask进行模型部署的步骤,帮助读者掌握机器学习的最佳实践。
56 1
|
21天前
|
机器学习/深度学习 数据采集 监控
如何使用机器学习模型来自动化评估数据质量?
如何使用机器学习模型来自动化评估数据质量?

热门文章

最新文章