Mechanic: A Learning Rate Tuner
Ashok Cutkosky, Aaron Defazio, Harsh Mehta
Abstract
We introduce a technique for tuning the learning rate scale factor of any base optimization algorithm and schedule automatically, which we call mechanic. Our method provides a practical realization of recent theoretical reductions for accomplishing a similar goal in online convex optimization. We rigorously evaluate mechanic on a range of large scale deep learning tasks with varying batch sizes, schedules, and base optimization algorithms. These experiments demonstrate that depending on the problem, mechanic either comes very close to, matches or even improves upon manual tuning of learning rates.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 5cfea8d0-7c67-4945-9594-bb42cb8411deCited by top-tier papers6
- Fast TRAC: A Parameter-Free Optimizer for Lifelong Reinforcement LearningAneesh Muppidi, Zhiyu Zhang, Heng YangNeurIPS 2024 · 19 citations
- Fully Unconstrained Online LearningAshok Cutkosky, Zakaria MhammediNeurIPS 2024 · 13 citations
- Tuning-Free Stochastic OptimizationAhmed Khaled, Chi JinICML 2024 · 13 citations
- Stepping on the Edge: Curvature Aware Learning Rate TunersVincent Roulet, Atish Agarwala, Jean-Bastien Grill, Grzegorz Swirszcz et al.NeurIPS 2024 · 9 citations
- State-free Reinforcement LearningMingyu Chen, Aldo Pacchiano, Xuezhou ZhangNeurIPS 2024
Builds on6
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- Symbolic Discovery of Optimization AlgorithmsXiangning Chen, Chen Liang, Da Huang, Esteban Real et al.NeurIPS 2023 · 734 citations
- Learning-Rate-Free Learning by D-AdaptationAaron Defazio, Konstantin MishchenkoICML 2023 · 117 citations
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size ScheduleMaor Ivgi, Oliver Hinder, Yair CarmonICML 2023 · 98 citations
- Gradient Descent: The Ultimate OptimizerKartik Chandra, Audrey Xie, Jonathan Ragan-Kelley, Erik MeijerNeurIPS 2022 · 66 citations
Related papers
- Convex Dominance in Deep Learning I: A Scaling Law of Loss and Learning RateZhiqi Bu, Shiyun Xu, Jialin MaoICLR 2026 · 4 citations
- The Road Less ScheduledAaron Defazio, Xingyu Yang, Ahmed Khaled, Konstantin Mishchenko et al.NeurIPS 2024 · 208 citations
- AdaScale SGD: A User-Friendly Algorithm for Distributed TrainingTyler B. Johnson, Pulkit Agrawal, Haijie Gu, Carlos GuestrinICML 2020 · 41 citations
- The Surprising Agreement Between Convex Optimization Theory and Learning-Rate Scheduling for Large Model TrainingFabian Schaipp, Alexander Hägele, Adrien B. Taylor, Umut Simsekli et al.ICML 2025
- QLABGrad: A Hyperparameter-Free and Convergence-Guaranteed Scheme for Deep LearningMinghan Fu, Fang-Xiang WuAAAI 2024 · 12 citations
