English

Anytime Training with Schedule-Free Spectral Optimization

Machine Learning 2026-05-25 v1 Artificial Intelligence Optimization and Control Machine Learning

Abstract

Standard neural network training relies on learning-rate schedules tied to a fixed horizon, leading to strong path dependence and costly re-tuning as data availability changes. Schedule-Free (SF) methods address this by removing explicit schedules, yet SF-AdamW, the current state-of-the-art anytime optimizer, consistently underperforms well-tuned AdamW baselines. We propose SF-NorMuon, a schedule-free spectral optimizer that closes this gap: with a single hyperparameter configuration, SF-NorMuon matches or exceeds tuned AdamW on 125M and 772M parameter language models across 11--8×8\times Chinchilla horizons. On the theoretical side, we prove a stationarity guarantee for schedule-free spectral dynamics and identify weight decay at the fast iterate as essential for long-horizon stability. SF-NorMuon enables practitioners to obtain high-quality checkpoints at any point during training without committing to a horizon in advance. By closing the performance gap with tuned baselines, SF-NorMuon makes horizon-free optimization more practical, taking a step towards truly open-ended, continual learning.

Cite

@article{arxiv.2605.23061,
  title  = {Anytime Training with Schedule-Free Spectral Optimization},
  author = {Anuj Apte and Pranav Deshpande and Niraj Kumar and Shouvanik Chakrabarti and Junhyung Lyle Kim},
  journal= {arXiv preprint arXiv:2605.23061},
  year   = {2026}
}