使用 bfloat16 (bf16) 训练大型语言模型会导致残差流中出现严重的精度损失,使得小更新被舍入掉。这是因为 bf16 数字之间可表示的间隔会随着数值增大而增大,而残差流会累积许多小更新。模拟显示,bf16 残差流损失了 fp32 全精度流中漂移的很大一部分。建议的解决方案包括将残差路径保留在 fp32 中,同时使用 bf16 进行矩阵乘法,或采用随机舍入来减轻偏差。 AI
影响 强调了 bf16 训练中一个关键的精度问题,该问题可能导致模型行为偏差,并需要特定的缓解策略。
排序理由 技术论文,详细介绍了 LLM 训练精度中的一个特定问题。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →