arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.06834cs.LG
收藏

Hyper-Tune:大规模高效超参数调整

Hyper-Tune: Towards Efficient Hyper-parameter Tuning at Scale

Peking University(北京大学);ETH Zürich(苏黎世联邦理工学院);Kuaishou Technology(快手科技)

Yang Li, Yu Shen, Huaijun Jiang, Wentao Zhang, Jixiang Li, Ji Liu, Ce Zhang, Bin Cui

摘要:

机器学习不断增长的需求和复杂性给超参数调优系统带来了压力:虽然模型的评估成本不断增加,但最先进技术的可扩展性开始成为一个关键瓶颈。在本文中,我们的经验时,部署超参数调优在现实世界中的应用程序在生产和现有系统的局限性的启发,我们提出了Hyper-Tune,一个有效的和强大的分布式超参数调优框架。与现有系统相比,Hyper-Tune突出了多个系统优化,包括(1)自动资源分配,(2)异步调度,(3)多保真度优化器。我们对基准数据集和生产中的大规模真实数据集进行了广泛的评估。从经验上看,在这些优化的帮助下,Hyper-Tune在各种场景下的性能都优于竞争对手的超参数调优系统,包括XGBoost、CNN、RNN和神经网络的一些架构超参数。与最先进的BOHB和A-BOHB相比,Hyper-Tune分别实现了高达11.2倍和5.1倍的加速比。

英文摘要:

The ever-growing demand and complexity of machine learning are putting pressure on hyper-parameter tuning systems: while the evaluation cost of models continues to increase, the scalability of state-of-the-arts starts to become a crucial bottleneck. In this paper, inspired by our experience when deploying hyper-parameter tuning in a real-world application in production and the limitations of existing systems, we propose Hyper-Tune, an efficient and robust distributed hyper-parameter tuning framework. Compared with existing systems, Hyper-Tune highlights multiple system optimizations, including (1) automatic resource allocation, (2) asynchronous scheduling, and (3) multi-fidelity optimizer. We conduct extensive evaluations on benchmark datasets and a large-scale real-world dataset in production. Empirically, with the aid of these optimizations, Hyper-Tune outperforms competitive hyper-parameter tuning systems on a wide range of scenarios, including XGBoost, CNN, RNN, and some architectural hyper-parameters for neural networks. Compared with the state-of-the-art BOHB and A-BOHB, Hyper-Tune achieves up to 11.2x and 5.1x speedups, respectively.

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

↑