研究定位 LM 预训练后期 loss spike:输出层舍入误差所致

micahgoldblum · x · 2026-11-06

Chen 等人的新工作追踪了 LM 预训练中后期出现的 loss spike,将其根源定位到输出层的舍入误差(quantization effect):高学习率下输出 logit 范数持续增长,最终引发损失突刺。论文还解释了 z-loss、μ-centering 和 momentum SGD 三种修复为何以相同机制起作用——都是约束/平滑输出层行为。作者包括 Tom Goldstein、John Kirchenbauer、Micah Goldblum 等。

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →