研究人员发现,大型语言模型预训练期间不稳定的一个重要原因是:损失尖峰。这些尖峰通常在训练后期出现,被称为“定时炸弹式尖峰”,与模型最终语言建模头中输出对数(logits)的增长和发散有关。研究发现,该头中的量化误差会引发这些尖峰,数值精度降低会加速尖峰的出现,而提高精度则能增强稳定性。提出的缓解方法包括将LM头中的AdamW替换为带动量的SGD,这可以抑制对数增长,并在延长训练期间提高稳定性。现有的缓解技术似乎也能减少量化误差,这表明它们的效果存在一个共同的数值机制。 AI
影响 识别出LLM预训练中的一个关键数值不稳定性,可能提高训练效率和稳定性。
排序理由 学术论文,详细介绍了LLM训练不稳定的一个具体技术发现和提出的缓解方法。[lever_c_demoted from research: ic=1 ai=1.0]
- AdamW
- alphaXiv
- arXiv
- CatalyzeX
- Connected Papers
- CORE Recommender
- DagsHub
- Gotit.pub
- Hugging Face
- IArxiv Recommender
- Litmaps
- ScienceCast
- scite Smart Citations
- SGD
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →