Google 研究员 Jon Barron:所有 softmax/交叉熵实现都有两个隐蔽 bug

jon_barron · x · 2026-10-25

Google 研究员 Jon Barron 发布新结果:包括 JAX、PyTorch 乃至教科书在内的所有现有 softmax 与交叉熵实现都存在两个微妙的 bug。一次简单的重写即可同时修复,并消除 grokking 过程中的「slingshot」(弹射)现象。

转发者 @ritwikak 实测反馈:这个修复刚刚阻止了她一次模型训练崩溃,称这一简单修复的影响之大「怎么强调都不为过」。

这意味着一个嵌入所有主流框架、影响无数训练 run 的底层数值问题被定位并给出了解决方案,对训练稳定性研究有直接意义。

所属事件:Barron 揭 softmax 与交叉熵实现两大数值 bug(21 条相关)→

原文链接 →

「研究」频道最新

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