LM head 反向传播丢失 95-99% 梯度,提高 head rank 加速收敛 16 倍

yoavartzi · x · 2026-11-13

引用的论文分析指出:反向传播时 95–99% 的梯度范数在 LM head 处丢失。作者证明 softmax 瓶颈(隐藏维度 D 远小于词表 V)损害的不只是表达能力,还有优化效果——将 head rank 从 32 提到 4096 后,2B 模型预训练收敛速度提升 16 倍。评论者补充可用 CPLM 在一定程度上缓解这一问题。

原文链接 →

「研究」频道最新

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