arXiv:2602.05786cs.LGstat.AP2026-02被引 1

对比多种调参方法,发现SMAC在树模型调参中表现最佳。

Selecting Hyperparameters for Tree-Boosting

  • 采用SMAC方法进行超参数优化,效果优于其他主流方法。
  • 需超过100次试验才能获得准确结果,小规模搜索易导致模型偏差。
  • 所有超参数均显著影响性能,早停法比枚举迭代次数更优。

树提升是处理表格数据的常用机器学习技术,但其泛化性能高度依赖多个超参数。本文在59个回归与二分类数据集上,实证比较了随机网格搜索、树结构帕兹恩估计(TPE)、基于高斯过程的贝叶斯优化(GP-BO)、Hyperband、SMAC方法以及确定性全网格搜索等超参数优化方法。结果表明,SMAC在平均表现上显著优于其他方法,在固定调参预算下对各类表格数据集均保持稳定性能,适合无法进行大量手动试错的用户。进一步观察发现:(i) 精确调参通常需要超过100次试验;(ii) 使用默认值或在小网格上全搜常导致模型严重不准确;(iii) 所有考虑的超参数均对树提升精度有显著影响,不存在可忽略的关键少数;(iv) 回归任务中通过早停选择提升迭代次数,比将其纳入搜索空间更有效。

原文摘要 · Abstract (English)

Tree-boosting is a widely used machine learning technique for tabular data. However, its out-of-sample accuracy is critically dependent on multiple hyperparameters. In this article, we empirically compare several popular methods for hyperparameter optimization for tree-boosting including random grid search, the tree-structured Parzen estimator (TPE), Gaussian-process-based Bayesian optimization (GP-BO), Hyperband, the sequential model-based algorithm configuration (SMAC) method, and deterministic full grid search using $59$ regression and binary classification data sets. We find that the SMAC method clearly outperforms all the other considered methods on average, and it gives stable performance across a diverse collection of tabular data sets under a fixed tuning budget, which is relevant for users who cannot afford extensive manual trial-and-error tuning. We further observe that (i) a relatively large number of trials larger than $100$ is typically required for accurate tuning, (ii) using default values for hyperparameters or a full search over a small grid often yields very inaccurate models, (iii) all considered hyperparameters can have a material effect on the accuracy of tree-boosting, i.e., there is no small set of hyperparameters that is more important than others, and (iv) choosing the number of boosting iterations using early stopping yields more accurate results compared to including it in the search space for regression tasks.

超参数优化树模型SMAC调参

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。