Automated Machine Learning on Big Data using Stochastic Algorithm Tuning

By T. Nickson, Michael A. Osborne, Steven Reece, Stephen J. Roberts. 2014

25 citations recorded by the source corpus.

We introduce a means of automating machine learning (ML) for big data tasks, by performing scalable stochastic Bayesian optimisation of ML algorithm parameters and hyper-parameters. More often than not, the critical tuning of ML algorithm parameters has relied on domain expertise from experts, along with laborious hand-tuning, brute search or lengthy sampling runs. Against this background, Bayesian optimisation is finding increasing use in automating parameter tuning, making ML algorithms accessible even to non-experts. However, the state of the art in Bayesian optimisation is incapable of scaling to the large number of evaluations of algorithm performance required to fit realistic models to complex, big data. We here describe a stochastic, sparse, Bayesian optimisation strategy to solve this problem, using many thousands of noisy evaluations of algorithm performance on subsets of data in order to effectively train algorithms for big data. We provide a comprehensive benchmarking of possible sparsification strategies for Bayesian optimisation, concluding that a Nystrom approximation offers the best scaling and performance for real tasks. Our proposed algorithm demonstrates substantial improvement over the state of the art in tuning the parameters of a Gaussian Process time series prediction task on real, big data.

Subfields: Artificial Intelligence, Management Science and Operations Research, Optimization & Training, Neural Architecture Search & AutoML

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. Frank Hutter
  2. Aaron Klein
  3. Stefan Falkner
  4. Qiang Yang
  5. Yee Whye Teh
  6. Quoc V. Le
  7. Wei-Wei Tu
  8. Balaji Lakshminarayanan

In Management Science and Operations Research

  1. Frank Hutter
  2. Aaron Klein
  3. Stefan Falkner
  4. Yang Yu
  5. Wei-Wei Tu
  6. Qiang Yang
  7. Yi-Qi Hu
  8. Isabelle M Guyon

In Optimization & Training

  1. Frank Hutter
  2. Stefan Falkner
  3. Aaron Klein
  4. Philipp Hennig
  5. Yang Yu
  6. Yi-Qi Hu
  7. Isabelle M Guyon
  8. Wei-Wei Tu

In Neural Architecture Search & AutoML

  1. Frank Hutter
  2. Aaron Klein
  3. Stefan Falkner
  4. Balaji Lakshminarayanan
  5. Philipp Hennig
  6. Yee Whye Teh
  7. Isabelle M Guyon
  8. Hugo Jair Escalante

Citations in the corpus