A new research paper explores the effectiveness of the learning-rate cooldown phase in large-model pretraining, a common component of Warmup-Stable-Decay (WSD) schedules. The study reveals that the benefit of this cooldown is contingent on the interplay between the structure of gradient noise and whether the optimizer employs normalization. Without normalization, stochastic gradient descent (SGD) naturally contracts and benefits little from cooldown. However, normalized methods and SignSGD can get stuck in a noise floor, where cooldown might offer improvements depending on the specific noise characteristics. AI
IMPACT Provides theoretical insights into optimizing large model training, potentially leading to more efficient pretraining methods.
RANK_REASON Research paper published on arXiv detailing theoretical findings about optimization schedules.
- alphaXiv
- arXiv
- CatalyzeX Code Finder for Papers
- cs.LG
- DagsHub
- Gotit.pub
- Hugging Face
- IArxiv
- Influence Flower
- Normalized SGD
- ScienceCast
- SGD
- SignSGD
- stochastic gradient descent
- Warmup-Stable-Decay (WSD)
AI-generated summary · Google Gemini · from 2 sources. How we write summaries →