Tunability: Importance of Hyperparameters of Machine Learning Algorithms

By Philipp Probst, Anne-Laure Boulesteix, Bernd Bischl. 2018

816 citations recorded by the source corpus.

Modern supervised machine learning algorithms involve hyperparameters that have to be set before running them. Options for setting hyperparameters are default values from the software package, manual configuration by the user or configuring them for optimal predictive performance by a tuning procedure. The goal of this paper is two-fold. Firstly, we formalize the problem of tuning from a statistical point of view, define data-based defaults and suggest general measures quantifying the tunability of hyperparameters of algorithms. Secondly, we conduct a large-scale benchmarking study based on 38 datasets from the OpenML platform and six common machine learning algorithms. We apply our measures to assess the tunability of their parameters. Our results yield default values for hyperparameters and enable users to decide whether it is worth conducting a possibly time consuming tuning strategy, to focus on the most important hyperparameters and to chose adequate hyperparameter spaces for tuning.

Subfields: Artificial Intelligence, Computational Theory and Mathematics

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Han-Jia Ye
  2. De-chuan Zhan
  3. Wei-Lun Chao
  4. Marco F. Huber
  5. Marc-André Zöller
  6. Anne-Laure Boulesteix
  7. Philipp Probst
  8. Bernd Bischl

In Computational Theory and Mathematics

  1. Marco F. Huber
  2. Marc-André Zöller
  3. Han-Jia Ye
  4. Wei-Lun Chao
  5. Anne-Laure Boulesteix
  6. Philipp Probst
  7. Marvin N. Wright
  8. Tobias Pielok

Citations in the corpus