针对高成本超参优化,动态调整训练与停止时机以提升性价比。
Cost-Sensitive Freeze-thaw Bayesian Optimization for Efficient Hyperparameter Tuning
- 引入效用函数衡量成本与性能的权衡,基于用户偏好数据估算。
- 新采集函数和停止准则使算法在最大效用处自动终止。
- 结合迁移学习提升采样效率,适合资源受限场景的超参调优。
本文针对成本敏感的超参数优化(HPO)问题,基于冻融贝叶斯优化(freeze-thaw BO)框架提出改进方法。假设用户希望在预期性能提升无法覆盖额外计算成本时提前终止优化过程。为此,我们在冻融框架中引入效用函数,描述成本与性能之间的权衡关系,该函数可由用户偏好数据估计。结合新的采集函数与停止准则,算法可动态选择未来最有望提升效用的配置继续训练,并在接近最大效用时自动停止。此外,通过迁移学习构建专用于成本敏感HPO的代理模型,显著提升了样本效率。在多个多保真度HPO基准上验证,本方法优于所有对比的冻融BO与迁移-BO基线,同时实现了更优的成本-性能平衡。代码已公开于https://github.com/db-Lee/CFBO。
原文摘要 · Abstract (English)
In this paper, we address the problem of \emph{cost-sensitive} hyperparameter optimization (HPO) built upon freeze-thaw Bayesian optimization (BO). Specifically, we assume a scenario where users want to early-stop the HPO process when the expected performance improvement is not satisfactory with respect to the additional computational cost. Motivated by this scenario, we introduce \emph{utility} in the freeze-thaw framework, a function describing the trade-off between the cost and performance that can be estimated from the user's preference data. This utility function, combined with our novel acquisition function and stopping criterion, allows us to dynamically continue training the configuration that we expect to maximally improve the utility in the future, and also automatically stop the HPO process around the maximum utility. Further, we improve the sample efficiency of existing freeze-thaw methods with transfer learning to develop a specialized surrogate model for the cost-sensitive HPO problem. We validate our algorithm on established multi-fidelity HPO benchmarks and show that it outperforms all the previous freeze-thaw BO and transfer-BO baselines we consider, while achieving a significantly better trade-off between the cost and performance. Our code is publicly available at https://github.com/db-Lee/CFBO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。