flâneur

[2310.07831] Optimal Linear Decay Learning Rate Schedules and Further Refinements

arxiv.org · 731 words · saved by 1 readers

Abstract:Learning rate schedules used in practice bear little resemblance to those recommended by theory. We close much of this theory/practice gap, and as a consequence are able to derive new problem-adaptive learning rate schedules. Our main technical contribution is a refined analysis of learning rate schedules for a wide class of optimization algorithms (including SGD). When considering only worst-case analysis, our theory predicts that the optimal choice is the linear decay schedule where the step-size is set proportional to 1 - t/T, where t is the current iteration and T is the total number of steps. To go beyond this worst-case analysis, we use the observed gradient norms to derive schedules refined for any particular task. These refined schedules exhibit learning rate warm-up and rapid learning rate annealing near the end of training. Ours is the first systematic approach to automatically yield both of these properties. We perform the most comprehensive evaluation of learning rate schedules to date, evaluating across 10 diverse deep learning problems, a series of LLMs, and a suite of logistic regression problems. We validate that overall, the linear-decay schedule outperforms all commonly used default schedules including cosine annealing. Our adaptive schedule refinement method gives further improvements.

[2310.07831] Optimal Linear Decay Learning Rate Schedules and Further Refinements --> Computer Science > Machine Learning arXiv:2310.07831 (cs) [Submitted on 11 Oct 2023 ( v1 ), last revised 29 Oct 2024 (this version, v2)] Title: Optimal Linear Decay Learning Rate Schedules and Further Refinements Authors: Aaron Defazio , Ashok Cutkosky , Harsh Mehta , Konstantin Mishchenko View a PDF of the paper titled Optimal Linear Decay Learning Rate Schedules and Further Refinements, by Aaron Defazio and Ashok Cutkosky and Harsh Mehta and Konstantin Mishchenko View PDF HTML (experimental) Abstract: Learn

saved by

related reading