研究人员开发了一种名为“曲率条件化多尺度动量与球面约束”(Curvature-Conditioned Multiscale Momentum with Sphere Constraints)的新方法,以提高大型语言模型(LLM)预训练的效率。该技术解决了由噪声梯度和病态损失景观带来的挑战,这些问题会减缓在损失景观关键平坦方向上的进展。通过采用一种专门针对这些平坦方向的多尺度动量,该方法增强了降噪和曲率适应能力,同时球面约束可防止参数膨胀。实验表明,在使用AdamW和Muon等优化器时,在各种架构和模型大小的LLM预训练中均实现了显著加速。 AI
影响 可能降低LLM预训练所需的计算成本和时间,使模型开发更加便捷。
排序理由 详细介绍LLM预训练新方法的学术论文。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →