机器学习模型超参数调整技巧

[MLM] Tips for Tuning Hyperparameters in Machine Learning Models

图片作者 | Canva 创建

如果您熟悉机器学习,您就知道训练过程可以使模型学习其参数(或模型系数)的最优值。但是,机器学习模型也有一组超参数,您在训练模型时应该指定它们的值。那么,如何找到这些超参数的最优值呢?

您可以使用超参数调整来找到超参数的最佳值。通过系统地调整超参数,您可以优化模型以获得最佳结果。

本教程提供了有效的超参数调整的实用技巧——从构建基线模型到使用贝叶斯优化等高级技术。无论您是超参数调整的新手,还是希望改进您的方法,这些技巧都将帮助您构建更好的机器学习模型。让我们开始吧。

1. 从简单开始:训练一个未经任何调整的基线模型

在开始超参数调整过程时,最好从简单开始,训练一个未经任何调整的基线模型。这个初始模型作为衡量后续调整工作效果的参考点。

为什么此步骤至关重要以及如何有效执行它:

  • 基线模型提供了一个基准,可以与调整后的模型进行比较。这有助于量化通过超参数调整实现的改进。
  • 选择一个默认模型:选择一个适合当前问题的模型。例如,分类问题使用决策树,回归问题使用线性回归。
  • 使用默认超参数:使用库提供的默认超参数训练模型。例如,如果使用 scikit-learn,则在不指定任何参数的情况下实例化模型。

使用适当的指标评估基线模型的性能。此步骤涉及将数据分割为训练集和测试集,训练模型,进行预测,并评估结果。

记录基线模型的性能指标。这将在您继续进行超参数调整时用于比较。

2. 使用带交叉验证的超参数搜索

建立基线模型后,下一步是通过超参数调整来优化模型的性能。利用带交叉验证的超参数搜索技术是找到最佳超参数集的稳健方法。

为什么使用带交叉验证的超参数搜索?

  • 交叉验证通过对多个折叠的结果进行平均,提供了更可靠的模型性能估计,降低了对特定训练-测试分割过拟合的风险。
  • 网格搜索和随机搜索等超参数搜索方法允许对超参数空间进行系统性探索,确保对潜在配置进行全面评估。
  • 此方法有助于选择能够很好地泛化到未见数据的超参数,从而在生产环境中获得更好的模型性能。

选择搜索技术:选择一种超参数搜索方法。最常见的两种策略是

  • 网格搜索,涉及对参数网格进行穷举搜索
  • 随机搜索,涉及从指定分布中随机抽样参数

定义超参数网格:指定要搜索的超参数及其各自的范围或分布。

使用交叉验证:您可以使用 cross_val_score 来评估模型在指定交叉验证方案下的性能,而不是单独定义交叉验证策略。

以这种方式使用带交叉验证的超参数调整,可以确保更可靠的性能估计和更好的模型泛化能力。

3. 使用随机搜索进行初步探索

在开始超参数调整时,通常最好使用随机搜索进行初步探索。与网格搜索相比,随机搜索提供了一种更有效的方法来探索广泛的超参数,尤其是在处理高维超参数空间时。

定义超参数分布:指定超参数及其要从中抽样的各自分布。

使用带交叉验证的随机搜索进行设置:使用带交叉验证的随机搜索来探索超参数空间。

评估模型:使用最佳超参数训练模型,并在测试集上评估其性能。

因此,随机搜索更适合高维超参数空间和计算成本高昂的模型。

4. 使用验证曲线监控过拟合

验证曲线有助于可视化超参数对训练和验证性能的影响,从而帮助您识别过拟合或欠拟合。

以下是一个示例。此代码片段使用验证曲线评估随机森林分类器性能随 n_estimators 超参数值变化的情况。它通过计算 5 折交叉验证下,一系列 n_estimators 值(10、100、200、400、800、1000)的训练和交叉验证分数来实现这一点。

然后,它绘制了训练集和交叉验证集的平均准确率得分及其标准差。生成的图有助于可视化在不同的 n_estimators 值下模型是过拟合还是欠拟合。

5. 使用贝叶斯优化进行高效搜索

使用贝叶斯优化进行超参数调整是一种非常高效有效的方法。它使用概率模型来探索超参数空间——需要更少的评估和计算资源。

您需要像 scikit-optimizehyperopt 这样的库来执行贝叶斯优化。在这里,我们将使用 scikit-optimize

定义超参数空间:指定要搜索的超参数及其各自的范围。

使用带交叉验证的贝叶斯优化进行设置:使用带交叉验证的贝叶斯优化来探索超参数空间。

评估模型:使用贝叶斯优化找到的最佳超参数训练最终模型,并在测试集上评估其性能。

总结

有效的超参数调整可以极大地改善机器学习模型的性能。

通过从一个简单的基线模型开始,并逐步使用搜索技术,您可以系统地探索并找到最佳超参数。从随机搜索的初步探索到贝叶斯优化的有效微调,我们概述了优化模型超参数的实用技巧。

祝您超参数调整愉快!

对《机器学习模型超参数调整技巧》的 2 条回复

  1. Gadisa 2024年8月8日 下午5:58 #

    哇!非常感谢!!

    请回答以下问题
    如果我想训练两个以上的(许多)机器学习模型,应该选择哪个模型作为基线模型?

    谢谢!
    Gadisa

    • James Carmichael 2024年8月9日 上午7:32 #

      你好 Gadisa……不客气!请详细说明您模型的目标,以便我们更好地指导您。

留下回复

Machine Learning Mastery 是 Guiding Tech Media 的一部分,Guiding Tech Media 是一家领先的数字媒体出版商,专注于帮助人们了解技术。访问我们的公司网站以了解更多关于我们的使命和团队的信息。