ScheduleFree+: Scaling Learning-Rate-Free & Schedule-Free Learning to Large Language Models

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autor principal: Defazio, Aaron
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909055139708928
author Defazio, Aaron
author_facet Defazio, Aaron
contents Schedule-Free Learning has shown promise as a practical anytime training method for machine learning, showing success across dozens of standard benchmark problems. However, strong performance for LLM training has only been demonstrated at small scales. We identify a number of fixes necessary to scale up Schedule-Free Learning to larger batch sizes and model sizes, and present a learning-rate-free and schedule-free method (ScheduleFree+) for training large language models which greatly outperforms Warmup-Stable-Decay (WSD) schedules. We also demonstrate that Schedule-Free Learning is most effective for long duration training, and at 1000 tokens per parameter, it outperforms SOTA schedules by 31%. Schedule-Free Learning provides a theoretical foundation for the use of model averaging and checkpoint merging during pretraining.
format Preprint
id arxiv_https___arxiv_org_abs_2605_19095
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle ScheduleFree+: Scaling Learning-Rate-Free & Schedule-Free Learning to Large Language Models
Defazio, Aaron
Machine Learning
Artificial Intelligence
Schedule-Free Learning has shown promise as a practical anytime training method for machine learning, showing success across dozens of standard benchmark problems. However, strong performance for LLM training has only been demonstrated at small scales. We identify a number of fixes necessary to scale up Schedule-Free Learning to larger batch sizes and model sizes, and present a learning-rate-free and schedule-free method (ScheduleFree+) for training large language models which greatly outperforms Warmup-Stable-Decay (WSD) schedules. We also demonstrate that Schedule-Free Learning is most effective for long duration training, and at 1000 tokens per parameter, it outperforms SOTA schedules by 31%. Schedule-Free Learning provides a theoretical foundation for the use of model averaging and checkpoint merging during pretraining.
title ScheduleFree+: Scaling Learning-Rate-Free & Schedule-Free Learning to Large Language Models
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2605.19095