Gradient-based Hyperparameter Optimization through Reversible Learning
Abstract
Tuning hyperparameters of learning algorithms is hard because gradients are usually unavailable. We compute exact gradients of cross-validation performance with respect to all hyperparameters by chaining derivatives backwards through the entire training procedure. These gradients allow us to optimize thousands of hyperparameters, including step-size and momentum schedules, weight initialization distributions, richly parameterized regularization schemes, and neural network architectures. We compute hyperparameter gradients by exactly reversing the dynamics of stochastic gradient descent with momentum.
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
The claims and statistics of this paper haven’t been checked yet.
Run the check — free, no account needed. It reads the open-access text and checks the sources it cites.
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
Examples of hyper-parameter selection methods include architecture search guided by hyper-gradients and differential hyper-parameters, as well as genetic algorithm or particle swarm style optimization.
“Examples include architecture search guided by hyper-gradients and differential hyper-parameters [59] as well as genetic algorithm or particle swarm style optimization [60].”
From An Introduction to Deep Learning for the Physical Layer (2017) · cites this paper as [59]
Self-check
Percentages, sums and reported differences re-computed from the paper’s own numbers.
The self-check needs the paper’s full text.
Cites
Adam: A Method for Stochastic Optimization
Kingma, D. P., & Ba, J. (2014) · arXiv · doi:10.48550/arxiv.1412.6980 · Cited by 98,946 · see citing works
Practical Bayesian Optimization of Machine Learning Algorithms
Snoek, J., Larochelle, H., & Adams, R. P. (2012) · arXiv · doi:10.48550/arxiv.1206.2944 · Cited by 6,691 · see citing works
Self-citation
Flagged first, then by citations
Research works printed but not linked: 35
- Bastien, Frédéric, Lamblin, Pascal, Pascanu, Razvan, Bergstra, James, Goodfellow, Ian J., Bergeron, Arnaud, Bouchard, Nicolas, and Bengio, Yoshua. Theano: new features and speed improvements. Deep Learning and Unsupervised Feature Learning NIPS 2012 Workshop, 2012.
- Baydin, A. G. and Pearlmutter, B. A. Automatic differenti- ation of algorithms for machine learning. In Proceedings of the AutoML Workshop at the International Conference on Machine Learning (ICML), 2014.
- Bengio, Yoshua. Gradient-based optimization of hyperpa- rameters. Neural computation, 12(8):1889-1900, 2000.
- Bengio, Yoshua, Simard, Patrice, and Frasconi, Paolo. Learning long-term dependencies with gradient descent is difficult. Neural Networks, IEEE Transactions on, 5 (2):157-166, 1994.
- Bergstra, James, Breuleux, Olivier, Bastien, Frédéric, Lamblin, Pascal, Pascanu, Razvan, Desjardins, Guil- laume, Turian, Joseph, Warde-Farley, David, and Ben- gio, Yoshua. Theano: a CPU and GPU math expression compiler. In Proceedings of the Python for Scientific Computing Conference (SciPy), June 2010. Oral Pre- sentation.
- Bergstra, James, Bardenet, Rémi, Bengio, Yoshua, Kégl, Balázs, et al. Algorithms for hyper-parameter optimiza- tion. In Advances in Neural Information Processing Sys- tems, 2011.
- Bergstra, James, Yamins, Daniel, and Cox, David. Making a science of model search: Hyperparameter optimiza- tion in hundreds of dimensions for vision architectures. In International Conference on Machine Learning, pp. 115-123, 2013.
- Chapelle, Olivier, Vapnik, Vladimir, Bousquet, Olivier, and Mukherjee, Sayan. Choosing multiple parameters for support vector machines. Machine learning, 46(1-3): 131-159, 2002.
- Chen, Dingding and Hagan, Martin T. Optimal use of reg- ularization and cross-validation in neural network mod- eling. In International Joint Conference on Neural Net- works, volume 2, pp. 1275-1280. IEEE, 1999.
- Courbariaux, Matthieu, Bengio, Yoshua, and David, Jean- Pierre. Low precision arithmetic for deep learning. arXiv preprint arXiv:1412.7024, 2014. · arXiv
- Dahl, George E, Jaitly, Navdeep, and Salakhutdinov, Rus- lan. Multi-task neural networks for QSAR predictions. arXiv preprint arXiv:1406.1231, 2014. · arXiv
- Domke, Justin. Generic methods for optimization-based modeling. In International Conference on Artificial In- telligence and Statistics, pp. 318-326, 2012.
- Eigenmann, Robert and Nossek, Josef A. Gradient based adaptive regularization. In Proceedings of the 1999 IEEE Signal Processing Society Workshop on Neural Networks, pp. 87-94. IEEE, 1999.
- Erhan, Dumitru, Bengio, Yoshua, Courville, Aaron, Man- zagol, Pierre-Antoine, Vincent, Pascal, and Bengio, Samy. Why does unsupervised pre-training help deep learning? The Journal of Machine Learning Research, 11:625-660, 2010.
- Foo, Chuan-sheng, Do, Chuong B., and Ng, Andrew Y. Ef- ficient multiple hyperparameter learning for log-linear models. In Advances in neural information processing systems, pp. 377-384, 2008.
- Graves, Alex, Wayne, Greg, and Danihelka, Ivo. Neural turing machines. arXiv preprint arXiv:1410.5401, 2014. · arXiv
- Hensman, James and Lawrence, Neil D. Nested variational compression in deep Gaussian processes. arXiv preprint arXiv:1412.1370, 2014. · arXiv
- Hochreiter, Sepp and Schmidhuber, Jürgen. Long short- term memory. Neural computation, 9(8):1735-1780, 1997.
- Hut, P., Makino, J., and McMillan, S. Building a better leapfrog. Astrophysical Journal, Part 2 -Letters, 443: L93-L96, April 1995. doi: 10.1086/187844. · DOI
- Hutter, Frank, Hoos, Holger H, and Leyton-Brown, Kevin. Sequential model-based optimization for general algo- rithm configuration. In Proceedings of LION-5, volume 6683, pp. 507-523. Springer, 2011.
- Lake, Brenden M. Towards more human-like concept learning in machines: Compositionality, causality, and learning-to-learn. PhD thesis, Massachusetts Institute of Technology, 2014.
- Larsen, Jan, Svarer, Claus, Andersen, Lars Nonboe, and Hansen, Lars Kai. Adaptive regularization in neural net- work modeling. In Neural Networks: Tricks of the Trade, pp. 113-132. Springer, 1998.
- MacKay, David J.C. Information theory, inference, and learning algorithms. Cambridge University Press, 2003.
- MacKay, David J.C. and Neal, Radford M. Automatic rel- evance determination for neural networks. In Technical Report. Cambridge University, 1994.
- Martens, James and Sutskever, Ilya. Training deep and recurrent networks with hessian-free optimization. In Neural Networks: Tricks of the Trade, pp. 479-535. Springer, 2012.
- Oliphant, Travis E. Python for scientific computing. Com- puting in Science & Engineering, 9(3):10-20, 2007.
- Pascanu, Razvan, Mikolov, Tomas, and Bengio, Yoshua. Understanding the exploding gradient problem. arXiv preprint arXiv:1211.5063, 2012. · arXiv
- Pearlmutter, Barak. An investigation of the gradient de- scent process in neural networks. PhD thesis, Carnegie Mellon University, 1996.
- Pearlmutter, Barak A. Fast exact multiplication by the Hes- sian. Neural computation, 6(1):147-160, 1994.
- Pearlmutter, Barak A. and Siskind, Jeffrey Mark. Reverse- mode AD in a functional framework: Lambda the ulti- mate backpropagator. ACM Transactions on Program- ming Languages and Systems (TOPLAS), 30(2):7, 2008.
- Rasmussen, Carl E. and Williams, Christopher K.I. Gaus- sian Processes for Machine Learning. The MIT Press, Cambridge, MA, USA, 2006.
- Salimans, Tim, Kingma, Diederik P., and Welling, Max. Markov chain Monte Carlo and variational inference: Bridging the gap. arXiv preprint arXiv:1410.6460, 2014. · arXiv
- Solak, E., Murray Smith, R., Leithead, W.E., Leith, D., and Rasmussen, Carl E. Derivative observations in Gaussian process models of dynamic systems. Advances in Neural Information Processing Systems, pp. 1057-1064, 2003.
- Sutskever, Ilya, Martens, James, Dahl, George, and Hin- ton, Geoffrey. On the importance of initialization and momentum in deep learning. In Proceedings of the 30th International Conference on Machine Learning (ICML- 13), pp. 1139-1147, 2013.
- Sutskever, Ilya, Vinyals, Oriol, and Le, Quoc V. V. Se- quence to sequence learning with neural networks. In Advances in Neural Information Processing Systems 27, pp. 3104-3112.
Cited by
Sorted by citations
An Introduction to Deep Learning for the Physical Layer
O'Shea, T., & Hoydis, J. (2017) · IEEE Transactions on Cognitive Communications and Networking · doi:10.1109/tccn.2017.2758370 · Cited by 2,980 · see citing works
Toward an integration of deep learning and neuroscience
Marblestone, A. H., Wayne, G., & Kording, K. P. (2016) · Frontiers in Computational Neuroscience · doi:10.3389/fncom.2016.00094 · Cited by 119 · see citing works
Learning Unsupervised Metaformer for Anomaly Detection
Wu, J.-C., Chen, D.-J., Fuh, C.-S., & Liu, T.-L. (2021) · 2021 IEEE/CVF International Conference on Computer Vision (ICCV) · doi:10.1109/iccv48922.2021.00433 · Cited by 91 · see citing works
Learning to (Learn at Test Time): RNNs with Expressive Hidden States
Sun, Y., Li, X., Dalal, K., Xu, J., Vikram, A., Zhang, G., Dubois, Y., Chen, X., Wang, X., Koyejo, S., et al. (2024) · arXiv · doi:10.48550/arxiv.2407.04620 · Cited by 18 · see citing works
“Conceptually, calling backward on ∇ ℓ means taking gradients of gradients – a well explored technique in meta-learning [54].”
End-to-End Test-Time Training for Long Context
Tandon, A., Dalal, K., Li, X., Koceja, D., Rød, M., Buchanan, S., Wang, X., Leskovec, J., Koyejo, S., Hashimoto, T., et al. (2025) · arXiv
“Then we average the loss after TTT over many independent training sequences, and optimize this average w.r.t. the model’s initialization for TTT through gradients of gradients in the outer loop [71, 3, 27].”
- Authors
- Maclaurin, D., Duvenaud, D., & Adams, R. P.
- Venue
- arXiv
- Publisher
- arXiv
- Published
- 2015
- arXiv
- 1502.03492
- References
- 2
- Type
- Preprint
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review