Score Centering 实现细节:topk(128) 加 ρ 重缩放抵消漂移

nrehiew_ · x · 2026-10-16

nrehiew 继续解读 Score Centering 论文的实现细节:为修正偏差,需要在梯度中减去交叉熵漂移项。

实践中,该 stop-gradient 项只需用词表的 topk(128) 概率来计算,但为此需要引入系数 ρ——根据尾部(非 top-k)概率对 top-k 进行重缩放,使总和仍为 1。这大幅降低了修正项的计算开销。

所属事件:RL 训练漂移被证实等价于向采样模型蒸馏(2 条相关)→

原文链接 →

「研究」频道最新

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