Learning an Adaptive Learning Rate Schedule
Not on the map. Why?
Abstract (arXiv)
The learning rate is one of the most important hyper-parameters for model training and generalization. However, current hand-designed parametric learning rate schedules offer limited flexibility and the predefined schedule may not match the training dynamics of high dimensional and non-convex optimization problems. In this paper, we propose a reinforcement learning based framework that can automatically learn an adaptive learning rate schedule by leveraging the information from past training histories. The learning rate dynamically changes based on the current training dynamics. To validate this framework, we conduct experiments with different neural network architectures on the Fashion MINIST and CIFAR10 datasets. Experimental results show that the auto-learned learning rate controller can achieve better test results. In addition, the trained controller network is generalizable -- able to be trained on one data set and transferred to new problems.
Key facts. Cited by 29 papers in the map (data through 7 Oct 2026). First posted on arXiv 20 Sep 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 5; 2022: 7; 2023: 4; 2024: 4; 2025: 5; 2026: 4.
29 papers.
Sort:
By area:
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2022AutoBalance: Optimized Loss Functions for Imbalanced Datacited by 49
- 2023Optimal Linear Decay Learning Rate Schedules and Further Refinementscited by 41
- 2022Automated Dynamic Algorithm Configurationcited by 37
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2022A Closer Look at Learned Optimization: Stability, Robustness, and Inductive Biasescited by 30
- 2021A Generalizable Approach to Learning Optimizerscited by 23
- 2025A Multi-Power Law for Loss Curve Prediction Across Learning Rate Schedulescited by 22
- 2021Reinforced Attention for Few-Shot Learning and Beyondcited by 18
- 2024Dynamic Layer Tying for Parameter-Efficient Transformerscited by 10
17 more
- 2022A Simple Guard for Learned Optimizerscited by 9
- 2024Narrowing the Focus: Learned Optimizers for Pretrained Modelscited by 3
- 2025Controlling changes to attention logitscited by 3
- 2023Studying K-FAC Heuristics by Viewing Adam through a Second-Order Lenscited by 2
- 2021Efficient Meta Subspace Optimizationcited by 1
- 2023Generalisable Agents for Neural Network Optimisationcited by 1
- 2025Interactive Training: Feedback-Driven Neural Network Optimizationcited by 1
- 2026Reinforcement Learning for Micro-Level Claims Reservingcited by 1
- 2021To Raise or Not To Raise: The Autonomous Learning Rate Questioncited by 0
- 2023Intelligent gradient amplification for deep neural networkscited by 0
- 2024Neural Optimizer Equation, Decay Function, and Learning Rate Schedule Joint Evolutioncited by 0
- 2024Curriculum Is More Influential Than Haptic Information During Reinforcement Learning of Object Manipulation Against Gravitycited by 0
- 2025SparseDoctor: Towards Efficient Chat Doctor with Mixture of Experts Enhanced Large Language Modelscited by 0
- 2025Optimization as a Dynamical System: Generative Schedules from Latent ODEscited by 0
- 2026Meta-Cognitive Reinforcement Learning with Self-Doubt and Recoverycited by 0
- 2026POP: Prior-Fitted First-Order Optimization Policiescited by 0
- 2026SOLAR: A State-Driven Online Learning Rate Scheduler for LLM Pretrainingcited by 0
- 2026SOLAR: A State-Driven Online Learning Rate Scheduler for LLM Pretraining28 Sep 2026
- 2026POP: Prior-Fitted First-Order Optimization Policies17 Feb 2026
- 2026Meta-Cognitive Reinforcement Learning with Self-Doubt and Recovery28 Jan 2026
- 2026Reinforcement Learning for Micro-Level Claims Reserving12 Jan 2026
- 2025Controlling changes to attention logits26 Nov 2025
- 2025Interactive Training: Feedback-Driven Neural Network Optimization2 Oct 2025
- 2025Optimization as a Dynamical System: Generative Schedules from Latent ODEs27 Sep 2025
- 2025SparseDoctor: Towards Efficient Chat Doctor with Mixture of Experts Enhanced Large Language Models15 Sep 2025
- 2025A Multi-Power Law for Loss Curve Prediction Across Learning Rate Schedules17 Mar 2025
- 2024Narrowing the Focus: Learned Optimizers for Pretrained Models17 Aug 2024
- 2024Curriculum Is More Influential Than Haptic Information During Reinforcement Learning of Object Manipulation Against Gravity13 Jul 2024
- 2024Neural Optimizer Equation, Decay Function, and Learning Rate Schedule Joint Evolution10 Apr 2024
17 more
- 2024Dynamic Layer Tying for Parameter-Efficient Transformers23 Jan 2024
- 2023Generalisable Agents for Neural Network Optimisation30 Nov 2023
- 2023Studying K-FAC Heuristics by Viewing Adam through a Second-Order Lens23 Oct 2023
- 2023Optimal Linear Decay Learning Rate Schedules and Further Refinements11 Oct 2023
- 2023Intelligent gradient amplification for deep neural networks29 May 2023
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Up17 Nov 2022
- 2022Learning to Learn with Generative Models of Neural Network Checkpoints26 Sep 2022
- 2022A Closer Look at Learned Optimization: Stability, Robustness, and Inductive Biases22 Sep 2022
- 2022Automated Dynamic Algorithm Configuration27 May 2022
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizers22 Mar 2022
- 2022A Simple Guard for Learned Optimizers28 Jan 2022
- 2022AutoBalance: Optimized Loss Functions for Imbalanced Data4 Jan 2022
- 2021Efficient Meta Subspace Optimization28 Oct 2021
- 2021To Raise or Not To Raise: The Autonomous Learning Rate Question16 Jun 2021
- 2021A Generalizable Approach to Learning Optimizers2 Jun 2021
- 2021Reinforced Attention for Few-Shot Learning and Beyond9 Apr 2021
- 2021Learning to Optimize: A Primer and A Benchmark23 Mar 2021
- 2021Learning to Optimize: A Primer and A Benchmarkcited by 206
- 2022Learning to Learn with Generative Models of Neural Network Checkpointscited by 85
- 2022VeLO: Training Versatile Learned Optimizers by Scaling Upcited by 57
- 2022AutoBalance: Optimized Loss Functions for Imbalanced Datacited by 49
- 2023Optimal Linear Decay Learning Rate Schedules and Further Refinementscited by 41
- 2022Automated Dynamic Algorithm Configurationcited by 37
- 2022Practical tradeoffs between memory, compute, and performance in learned optimizerscited by 32
- 2022A Closer Look at Learned Optimization: Stability, Robustness, and Inductive Biasescited by 30
- 2021A Generalizable Approach to Learning Optimizerscited by 23
- 2025A Multi-Power Law for Loss Curve Prediction Across Learning Rate Schedulescited by 22
- 2021Reinforced Attention for Few-Shot Learning and Beyondcited by 18
- 2024Dynamic Layer Tying for Parameter-Efficient Transformerscited by 10
12 most cited of 28 All 28 (JSON)
- 2026SOLAR: A State-Driven Online Learning Rate Scheduler for LLM Pretraining28 Sep 2026
- 2026POP: Prior-Fitted First-Order Optimization Policies17 Feb 2026
- 2026Meta-Cognitive Reinforcement Learning with Self-Doubt and Recovery28 Jan 2026
- 2026Reinforcement Learning for Micro-Level Claims Reserving12 Jan 2026
- 2025Controlling changes to attention logits26 Nov 2025
- 2025Interactive Training: Feedback-Driven Neural Network Optimization2 Oct 2025
- 2025Optimization as a Dynamical System: Generative Schedules from Latent ODEs27 Sep 2025
- 2025A Multi-Power Law for Loss Curve Prediction Across Learning Rate Schedules17 Mar 2025
- 2024Narrowing the Focus: Learned Optimizers for Pretrained Models17 Aug 2024
- 2024Curriculum Is More Influential Than Haptic Information During Reinforcement Learning of Object Manipulation Against Gravity13 Jul 2024
- 2024Neural Optimizer Equation, Decay Function, and Learning Rate Schedule Joint Evolution10 Apr 2024
- 2024Dynamic Layer Tying for Parameter-Efficient Transformers23 Jan 2024
12 newest of 28 All 28 (JSON)