mlsanityData to 7 Oct 2026

Anytime Pretraining: Horizon-Free Learning-Rate Schedules with Weight Averaging

arXiv 2602.03702 ↗First posted 3 Feb 2026cs.LG cs.AI math.OCcited by 8

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

  1. 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
  2. 2024The Llama 3 Herd of Modelscited by 14,192
  3. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  4. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
  5. 2022Training Compute-Optimal Large Language Modelscited by 2,927
  6. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  7. 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
  8. 2021Scaling Vision Transformerscited by 993
  9. 2023A Comprehensive Survey of Continual Learning: Theory, Method and Applicationcited by 860
  10. 2024MiniCPM: Unveiling the Potential of Small Language Models with Scalable Training Strategiescited by 694
  11. 2024OLMo: Accelerating the Science of Language Modelscited by 599
  12. 2023Analyzing and Improving the Training Dynamics of Diffusion Modelscited by 396
38 more
  1. 2019Online Continual Learning with Maximally Interfered Retrievalcited by 376
  2. 20242 OLMo 2 Furiouscited by 302
  3. 2024Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durationscited by 160
  4. 2024The Road Less Scheduledcited by 158
  5. 2022Self-Stabilization: The Implicit Bias of Gradient Descent at the Edge of Stabilitycited by 151
  6. 2021Ensemble of Averages: Improving Model Selection and Boosting Performance in Domain Generalizationcited by 144
  7. 2023Continual Pre-Training of Large Language Models: How to (re)warm your model?cited by 135
  8. 2021An Empirical Investigation of the Role of Pre-training in Lifelong Learningcited by 134
  9. 2024Simple and Scalable Strategies to Continually Pre-train Large Language Modelscited by 122
  10. 2013Non-strongly-convex smooth stochastic approximation with convergence rate O(1/n)cited by 110
  11. 2024A Dynamical Model of Neural Scaling Lawscited by 105
  12. 2018Tight Analyses for Non-Smooth Stochastic Gradient Descentcited by 94
  13. 2020Efficient Continual Learning with Modular Networks and Task-Driven Priorscited by 87
  14. 2022Continual Pre-Training Mitigates Forgetting in Language and Visioncited by 82
  15. 2024Understanding Warmup-Stable-Decay Learning Rates: A River Valley Loss Landscape Perspectivecited by 75
  16. 2018The Unusual Effectiveness of Averaging in GAN Trainingcited by 73
  17. 2021Benign Overfitting of Constant-Stepsize SGD for Linear Regressioncited by 71
  18. 2024How Does Critical Batch Size Scale in Pre-training?cited by 68
  19. 2019The Step Decay Schedule: A Near Optimal, Geometrically Decaying Learning Rate Procedure For Least Squarescited by 67
  20. 20244+3 Phases of Compute-Optimal Neural Scaling Lawscited by 66
  21. 2024Exponential Moving Average of Weights in Deep Learning: Dynamics and Benefitscited by 65
  22. 2022Stop Wasting My Time! Saving Days of ImageNet and BERT Training with Latest Weight Averagingcited by 54
  23. 2024Scaling Laws in Linear Regression: Compute, Parameters, and Datacited by 54
  24. 2023Benchmarking Neural Network Training Algorithmscited by 46
  25. 2024Scaling and renormalization in high-dimensional regressioncited by 45
  26. 2024Why Warmup the Learning Rate? Underlying Mechanisms and Improvementscited by 45
  27. 2023Optimal Linear Decay Learning Rate Schedules and Further Refinementscited by 41
  28. 2021Last Iterate Risk Bounds of SGD with Decaying Stepsize for Overparameterized Linear Regressioncited by 37
  29. 2023Early Weight Averaging meets High Learning Rates for LLM Pre-trainingcited by 36
  30. 2025Model Merging in Pre-training of Large Language Modelscited by 36
  31. 2024Scaling Optimal LR Across Token Horizonscited by 35
  32. 2025Straight to Zero: Why Linearly Decaying the Learning Rate to Zero Works Best for LLMscited by 34
  33. 2025The Surprising Agreement Between Convex Optimization Theory and Learning-Rate Scheduling for Large Model Trainingcited by 31
  34. 2025Accelerating Neural Network Training: An Analysis of the AlgoPerf Competitioncited by 30
  35. 2024Scaling Law with Learning Rate Annealingcited by 29
  36. 2017A Markov Chain Theory Approach to Characterizing the Minimax Optimality of Stochastic Gradient Descent (for Least Squares)cited by 25
  37. 2022The Power and Limitation of Pretraining-Finetuning for Linear Regression under Covariate Shiftcited by 25
  38. 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

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 8.

8 papers.