arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.05786cs.LGstat.APstat.ML
收藏

选择树提升的超参数

Selecting Hyperparameters for Tree-Boosting

Seminar for Statistics, ETH Zurich(苏黎世联邦理工学院统计研究所)

Floris Jan Koster, Fabio Sigrist

AI总结:

本文通过59个数据集比较了多种超参数优化方法,发现SMAC方法显著优于其他方法,并揭示了超参数调优的关键因素。

摘要:

树提升是一种广泛用于表格数据的机器学习技术。然而,其样本外准确性严重依赖于多个超参数。在本文中,我们使用59个回归和分类数据集,实证比较了几种流行的树提升超参数优化方法,包括随机网格搜索、树结构Parzen估计器(TPE)、基于高斯过程的贝叶斯优化(GP-BO)、Hyperband、基于序列模型的算法配置(SMAC)方法以及确定性全网格搜索。我们发现SMAC方法明显优于所有其他考虑的方法。我们进一步观察到:(i)需要相对较大的试验次数(大于100)才能进行准确的调优,(ii)使用超参数的默认值会产生非常不准确的模型,(iii)所有考虑的超参数都可能对树提升的准确性产生实质性影响,即不存在一组比其他超参数更重要的超参数,以及(iv)对于回归任务,使用早停法选择提升迭代次数比将其包含在搜索空间中能产生更准确的结果。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑