English

Stochastic Hyperparameter Optimization through Hypernetworks

Machine Learning 2018-03-09 v2

Abstract

Machine learning models are often tuned by nesting optimization of model weights inside the optimization of hyperparameters. We give a method to collapse this nested optimization into joint stochastic optimization of weights and hyperparameters. Our process trains a neural network to output approximately optimal weights as a function of hyperparameters. We show that our technique converges to locally optimal weights and hyperparameters for sufficiently large hypernetworks. We compare this method to standard hyperparameter optimization strategies and demonstrate its effectiveness for tuning thousands of hyperparameters.

Keywords

Cite

@article{arxiv.1802.09419,
  title  = {Stochastic Hyperparameter Optimization through Hypernetworks},
  author = {Jonathan Lorraine and David Duvenaud},
  journal= {arXiv preprint arXiv:1802.09419},
  year   = {2018}
}

Comments

9 pages, 6 figures; revised figures