arXiv · 1910.04522
Probabilistic Rollouts for Learning Curve Extrapolation Across Hyperparameter Settings
Abstract
We propose probabilistic models that can extrapolate learning curves of iterative machine learning algorithms, such as stochastic gradient descent for training deep networks, based on training data with variable-length learning curves. We study instantiations of this framework based on random forests and Bayesian recurrent neural networks. Our experiments show that these models yield better predictions than state-of-the-art models from the hyperparameter optimization literature when extrapolating the performance of neural networks trained with different hyperparameter settings.
Explore related subjects
Keep this discovery
Matilde Gargiani, Aaron Klein, Stefan Falkner, Frank Hutter. 2019-10-10. Probabilistic Rollouts for Learning Curve Extrapolation Across Hyperparameter Settings. https://arxiv.org/abs/1910.04522
Cite the original work for its findings. Save a collection to share your selection of sources.