防训练崩溃的简单修法:只在LM头把AdamW换成动量SGD

micahgoldblum · x · 2026-11-03

Goldblum 线程第 10 条:给出修复方案——只在 LM 头用动量 SGD,其余仍用 AdamW。SGD-M 保留了梯度的零和结构,避免 Adam 引起的平均输出嵌入漂移。引用上下文指出 FP32 下输出 logit 仍增长但 loss 不发散,而 BF16 中 logit 与预测分布的量化误差会在失败前增长数个数量级。

原文链接 →

「研究」频道最新

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