Cyclical Learning Rates for Training Neural Networks

June 03, 2015 · Entered Twilight · 🏛 IEEE Workshop/Winter Conference on Applications of Computer Vision

"Last commit was 5.0 years ago (≥5 year threshold)"

Evidence collected by the PWNC Scanner

Repo contents: LICENSE, README.md, __init__.py, clr_callback.py, clr_callback_tests.ipynb, images

Authors Leslie N. Smith arXiv ID 1506.01186 Category cs.CV: Computer Vision Cross-listed cs.LG, cs.NE Citations 2.8K Venue IEEE Workshop/Winter Conference on Applications of Computer Vision Repository https://github.com/bckenstler/CLR ⭐ 1210 Last Checked 1 month ago

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.