机器学习-交叉验证

交叉验证 (Cross-Validation) 是一种评估模型性能和选择模型参数的统计学方法,特别是在数据量有限的情况下。它比简单地将数据分成训练集和测试集更加可靠,因为它利用了所有的数据进行训练和测试。

什么是交叉验证?

交叉验证的基本思想是将数据集分成 K 个大小相似的子集(称为"折叠"或"组",folds)。然后,依次将每个子集作为测试集,其余 K-1 个子集合并作为训练集,训练并测试模型。这个过程重复 K 次,每个子集都会被用作一次测试集。最后,将 K 次测试的结果(例如准确率、误差等)取平均值作为模型的最终性能评估。

最常见的交叉验证类型是 K 折交叉验证 (K-Fold Cross-Validation)。

K 折交叉验证的步骤:

数据分割: 将数据集随机分成 K 个大小相似的子集(通常 K 取 5 或 10)。

迭代训练和测试:

对于每个子集 i (i = 1, 2, ..., K):

将子集 i 作为测试集。

将其余 K-1 个子集作为训练集。

使用训练集训练模型。

使用测试集评估模型性能,并记录结果(例如准确率)。

性能评估: 计算 K 次测试结果的平均值,作为模型的最终性能评估指标。

其他类型的交叉验证:

留一交叉验证 (Leave-One-Out Cross-Validation, LOOCV): K 折交叉验证的特例,其中 K 等于样本数量。每次只留下一个样本作为测试集,其余样本作为训练集。计算成本较高,但在样本量较小时比较有用。

分层 K 折交叉验证 (Stratified K-Fold Cross-Validation): 确保每个子集中的类别比例与整个数据集中的类别比例相同。这对于分类问题,特别是类别不平衡的情况非常重要。

如何通过交叉验证选择正则化参数?

交叉验证可以用来选择最佳的正则化参数(例如 L1 或 L2 正则化中的 λ)。以下是具体步骤:

确定参数范围: 为正则化参数 λ 选择一个候选值范围(例如 0.001, 0.01, 0.1, 1, 10)。

对每个 λ 值执行 K 折交叉验证:

使用选定的 λ 值训练模型。

执行 K 折交叉验证,并记录每个 λ 值对应的平均性能指标(例如平均准确率或平均误差)。

选择最佳 λ 值: 选择在交叉验证中表现最好的 λ 值(例如,具有最高平均准确率或最低平均误差的 λ 值)作为最终模型的正则化参数。

示例:

假设我们使用 L2 正则化 (Ridge Regression) 训练一个线性回归模型,并使用 5 折交叉验证来选择最佳的 λ 值。

我们选择 λ 的候选值范围为 0.001, 0.01, 0.1, 1, 10。

对于每个 λ 值(例如 λ = 0.001):

我们将数据分成 5 份。

我们进行 5 次迭代,每次使用 4 份数据进行训练,1 份数据进行测试,并记录测试误差。

我们计算这 5 次测试误差的平均值,作为 λ = 0.001 时的交叉验证误差。

我们对每个 λ 值重复步骤 2,得到每个 λ 值对应的交叉验证误差。

我们选择交叉验证误差最低的 λ 值作为最终模型的正则化参数。

总结:

交叉验证是一种评估模型性能和选择模型参数的强大技术。通过在不同的数据子集上训练和测试模型,交叉验证可以提供对模型泛化能力的更可靠的估计,并帮助我们选择最佳的正则化参数,从而构建更准确、更鲁棒的机器学习模型。 使用交叉验证选择正则化参数可以有效地防止过拟合并提高模型的泛化能力。

相关推荐
飞猫的边缘AI26 分钟前
英特尔量子生意
人工智能·量子计算·边缘ai
深蓝AI27 分钟前
桌面 Agent 缺的不是视觉:Vercel 开源 agent-desktop,用无障碍树 ref 替代截图坐标
人工智能·程序员
镜子AI35 分钟前
镜子AI 技术复盘:三条自动化链路上的“停止条件“设计
运维·人工智能·自动化
向哆哆1 小时前
羊行为目标检测数据集:4类别、2,000张图像 | 目标检测
人工智能·yolo·目标检测·目标跟踪
如若1231 小时前
【论文复现】CoFailMed:从零复现医学图像分割的共享失败审计,含完整代码流程与科研绘图
人工智能·python
赛博守夜人2 小时前
AI安全攻防与治理之十六:对抗样本、恶意文件与多模态攻击,AI 内容安全检测的真实盲区
人工智能·安全
罗西的思考2 小时前
【轻量化复刻大模型 / KIMI K3】KDA(Kimi Delta Attention):并行
人工智能·算法·机器学习
yh弓长3 小时前
Agent 的 ReAct、Plan、Execute、Reflection 模式
人工智能
GlobalInfo3 小时前
2026年全球机器人用超平执行器市场研究报告:市场规模达7.51亿美元,人形机器人商业化与轻量化运动控制需求推动行业增长
大数据·人工智能·数码相机·microsoft·机器人
冬奇Lab3 小时前
一天一个开源项目(第252篇):Lightpanda —— 用 Zig 从零写的 AI 专用无头浏览器,内存省 16 倍、速度快 9 倍
人工智能·开源·资讯