Frugal Optimization for Cost-related Hyperparameters

Authors

  • Qingyun Wu Microsoft Research
  • Chi Wang Microsoft Research
  • Silu Huang Microsoft Research

DOI:

https://doi.org/10.1609/aaai.v35i12.17239

Keywords:

Hyperparameter Tuning / Algorithm Configuration

Abstract

The increasing demand for democratizing machine learning algorithms calls for hyperparameter optimization (HPO) solutions at low cost. Many machine learning algorithms have hyperparameters which can cause a large variation in the training cost. But this effect is largely ignored in existing HPO methods, which are incapable to properly control cost during the optimization process. To address this problem, we develop a new cost-frugal HPO solution. The core of our solution is a simple but new randomized direct-search method, for which we provide theoretical guarantees on the convergence rate and the total cost incurred to achieve convergence. We provide strong empirical results in comparison with state-of-the-art HPO methods on large AutoML benchmarks.

Downloads

Published

2021-05-18

How to Cite

Wu, Q., Wang, C., & Huang, S. (2021). Frugal Optimization for Cost-related Hyperparameters. Proceedings of the AAAI Conference on Artificial Intelligence, 35(12), 10347-10354. https://doi.org/10.1609/aaai.v35i12.17239

Issue

Section

AAAI Technical Track on Machine Learning V