Cyclical Learning Rates for Training Neural Networks

Leslie N. Smith
6/3/2015
cs.CVcs.LGcs.NE

Official code: bckenstler/CLR (1,204 stars, MIT, last commit 2020-06-05).

Abstract

It is known that the learning rate is the most important hyper-parameter to tune for training deep neural networks. This paper describes a new method for setting the learning rate, named cyclical learning rates, which practically eliminates the need to experimentally find the best values and schedule for the global learning rates. Instead of monotonically decreasing the learning rate, this method lets the learning rate cyclically vary between reasonable boundary values. Training with cyclical learning rates instead of fixed values achieves improved classification accuracy without a need to tune and often in fewer iterations. This paper also describes a simple way to estimate "reasonable bounds" -- linearly increasing the learning rate of the network for a few epochs. In addition, cyclical learning rates are demonstrated on the CIFAR-10 and CIFAR-100 datasets with ResNets, Stochastic Depth networks, and DenseNets, and the ImageNet dataset with the AlexNet and GoogLeNet architectures. These are practical tools for everyone who trains neural networks.

View on arXivView PDF

Code Implementations(1)

bckenstler/CLROfficial100%
MIT
1,204240Python, Jupyter NotebookMar 21, 20176 years agoMIT

Similar Papers

Cite this paper

@article{smith2015cyclical,
  title  = {Cyclical Learning Rates for Training Neural Networks},
  author = {Leslie N. Smith},
  year   = {2015},
  eprint = {1506.01186},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/1506.01186}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/1506.01186.svg)](https://codeofpaper.com/paper/1506.01186)

Discussion