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

MANGO:一个用于并行超参数调优的Python库

MANGO: A Python Library for Parallel Hyperparameter Tuning

Arm(安谋科技)

Sandeep Singh Sandha, Mohit Aggarwal, Igor Fedorov, Mani Srivastava

摘要:

调整机器学习算法的超参数是一项繁琐的任务,通常是手动完成的。为了实现自动化的超参数调整,最近的工作已经开始使用基于贝叶斯优化的技术。然而,为了实际上实现大规模机器学习训练管道的自动调整,现有库中仍然存在显著的差距,包括缺乏抽象、容错和灵活性以支持任何分布式计算框架上的调度。为了解决这些挑战,我们提出了Mango,一个用于并行超参数调优的Python库。Mango支持使用任何分布式调度框架,实现智能并行搜索策略,并提供丰富的抽象来定义与scikit-learn兼容的复杂超参数搜索空间。Mango在性能上与另一个广泛使用的库Hyperopt相当。Mango是开源的,目前在Arm Research的生产中使用,以提供最先进的超参数调整功能。

英文摘要:

Tuning hyperparameters for machine learning algorithms is a tedious task, one that is typically done manually. To enable automated hyperparameter tuning, recent works have started to use techniques based on Bayesian optimization. However, to practically enable automated tuning for large scale machine learning training pipelines, significant gaps remain in existing libraries, including lack of abstractions, fault tolerance, and flexibility to support scheduling on any distributed computing framework. To address these challenges, we present Mango, a Python library for parallel hyperparameter tuning. Mango enables the use of any distributed scheduling framework, implements intelligent parallel search strategies, and provides rich abstractions for defining complex hyperparameter search spaces that are compatible with scikit-learn. Mango is comparable in performance to Hyperopt, another widely used library. Mango is available open-source and is currently used in production at Arm Research to provide state-of-art hyperparameter tuning capabilities.

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

↑