PubMed · 10953243
Gradient-based optimization of hyperparameters.
Abstract
Many machine learning algorithms can be formulated as the minimization of a training criterion that involves a hyperparameter. This hyperparameter is usually chosen by trial and error with a model selection criterion. In this article we present a methodology to optimize several hyperparameters, based on the computation of the gradient of a model selection criterion with respect to the hyperparameters. In the case of a quadratic training criterion, the gradient of the selection criterion with respect to the hyperparameters is efficiently computed by backpropagating through a Cholesky decomposition. In the more general case, we show that the implicit function theorem can be used to derive a formula for the hyperparameter gradient involving second derivatives of the training criterion.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Y Bengio. 2000. Gradient-based optimization of hyperparameters.. https://doi.org/10.1162/089976600300015187
Cite the original work for its findings. Save a collection to share your selection of sources.