Anytime Pretraining: Horizon-Free Learning-Rate Schedules with Weight Averaging
Not on the map. Why?
Abstract (arXiv)
Large language models are increasingly trained in continual or open-ended settings, where the total training horizon is not known in advance. Despite this, most existing pretraining recipes are not anytime: they rely on horizon-dependent learning rate schedules and extensive tuning under a fixed compute budget. In this work, we provide a theoretical analysis demonstrating the existence of anytime learning schedules for overparameterized linear regression, and we highlight the central role of weight averaging - also known as model merging - in achieving the minimax convergence rates of stochastic gradient descent. We show that these anytime schedules polynomially decay with time, with the decay rate determined by the source and capacity conditions of the problem. Empirically, we evaluate 150M and 300M parameter language models trained for up to 32x Chinchilla scale, comparing constant and 1/√t schedules with weight averaging against a well-tuned cosine schedule. Across the full training range, the anytime schedules achieve comparable final loss to cosine decay. Taken together, our results suggest that weight averaging combined with simple, horizon-free step sizes offers a practical and effective anytime alternative to cosine learning rate schedules for large language model pretraining.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 66 of its 82 references that are in the map. First posted on arXiv 3 Feb 2026.
Builds on
66 of 82 references.
References · 66
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2021Scaling Vision Transformerscited by 993
- 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 860
- 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
- 2024OLMo: Accelerating the Science of Language Modelscited by 599
- 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 396
38 more
- 2019Online Continual Learning with Maximally Interfered Retrievalcited by 376
- 20242 OLMo 2 Furiouscited by 302
- 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
- 2024The Road Less Scheduledcited by 158
- 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 151
- 2021Ensemble of Averages: Improving Model Selection and Boosting Performance in Domain Generalizationcited by 144
- 2023Continual Pre-Training of Large Language Models: How to (re)warm your model?cited by 135
- 2021An Empirical Investigation of the Role of Pre-training in Lifelong Learningcited by 134
- 2024Simple and Scalable Strategies to Continually Pre-train Large Language Modelscited by 122
- 2013Non-strongly-convex smooth stochastic approximation with convergence rate O(1/n)cited by 110
- 2024A Dynamical Model of Neural Scaling Lawscited by 105
- 2018Tight Analyses for Non-Smooth Stochastic Gradient Descentcited by 94
- 2020Efficient Continual Learning with Modular Networks and Task-Driven Priorscited by 87
- 2022Continual Pre-Training Mitigates Forgetting in Language and Visioncited by 82
- 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 75
- 2018The Unusual Effectiveness of Averaging in GAN Trainingcited by 73
- 2021Benign Overfitting of Constant-Stepsize SGD for Linear Regressioncited by 71
- 2024How Does Critical Batch Size Scale in Pre-training?cited by 68
- 2019The Step Decay Schedule: A Near Optimal, Geometrically Decaying Learning Rate Procedure For Least Squarescited by 67
- 20244+3 Phases of Compute-Optimal Neural Scaling Lawscited by 66
- 2024Exponential Moving Average of Weights in Deep Learning: Dynamics and Benefitscited by 65
- 2022Stop Wasting My Time! Saving Days of ImageNet and BERT Training with Latest Weight Averagingcited by 54
- 2024Scaling Laws in Linear Regression: Compute, Parameters, and Datacited by 54
- 2023Benchmarking Neural Network Training Algorithmscited by 46
- 2024Scaling and renormalization in high-dimensional regressioncited by 45
- 2024Why Warmup the Learning Rate? Underlying Mechanisms and Improvementscited by 45
- 2023Optimal Linear Decay Learning Rate Schedules and Further Refinementscited by 41
- 2021Last Iterate Risk Bounds of SGD with Decaying Stepsize for Overparameterized Linear Regressioncited by 37
- 2023Early Weight Averaging meets High Learning Rates for LLM Pre-trainingcited by 36
- 2025Model Merging in Pre-training of Large Language Modelscited by 36
- 2024Scaling Optimal LR Across Token Horizonscited by 35
- 2025Straight to Zero: Why Linearly Decaying the Learning Rate to Zero Works Best for LLMscited by 34
- 2025The Surprising Agreement Between Convex Optimization Theory and Learning-Rate Scheduling for Large Model Trainingcited by 31
- 2025Accelerating Neural Network Training: An Analysis of the AlgoPerf Competitioncited by 30
- 2024Scaling Law with Learning Rate Annealingcited by 29
- 2017A Markov Chain Theory Approach to Characterizing the Minimax Optimality of Stochastic Gradient Descent (for Least Squares)cited by 25
- 2022The Power and Limitation of Pretraining-Finetuning for Linear Regression under Covariate Shiftcited by 25
- 2025A Multi-Power Law for Loss Curve Prediction Across Learning Rate Schedulescited by 22
50 most cited of 66 All 66 (JSON)
Built on by
Citing papers per year: 2026: 8.
8 papers.
- 2026A Defense of the Quadratic Modelcited by 6
- 2026Anytime Training with Schedule-Free Spectral Optimizationcited by 3
- 2026Statistical Properties of Training & Generalizationcited by 2
- 2026WSqD: A Horizon-Free Learning Rate Schedule for Large Model Trainingcited by 0
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretrainingcited by 0
- 2026Evaluating Tiny Recursive Models Across Training for Code Generationcited by 0
- 2026Terminal Shrinkage Averaging Reveals a Schedule-Estimator Interaction in LLM Pretrainingcited by 0
- 2026QAM: Quadratic-Accurate Checkpoint Merging via Sequential Consistencycited by 0
- 2026QAM: Quadratic-Accurate Checkpoint Merging via Sequential Consistency28 Sep 2026
- 2026Terminal Shrinkage Averaging Reveals a Schedule-Estimator Interaction in LLM Pretraining21 Sep 2026
- 2026Evaluating Tiny Recursive Models Across Training for Code Generation29 Aug 2026
- 2026Curvature-Conditioned Multiscale Momentum with Sphere Constraints for LLM Pretraining28 Aug 2026
- 2026A Defense of the Quadratic Model23 Jul 2026
- 2026WSqD: A Horizon-Free Learning Rate Schedule for Large Model Training12 Jul 2026
- 2026Statistical Properties of Training & Generalization18 Jun 2026
- 2026Anytime Training with Schedule-Free Spectral Optimization21 May 2026