Jon Barron:softmax 与交叉熵所有主流实现都有两处隐藏 bug

jon_barron · x · 2026-10-25

Google 研究员 Jon Barron 分享新结果:JAX、PyTorch 乃至教科书中 softmax 与交叉熵的所有现有实现都存在两个微妙的 bug,简单重写即可修复,并能消除 grokking 过程中的 slingshot 现象。此前 AggieInCA 等人的论文曾观察到训练 loss 尖峰(slingshot effect),当时推测与数值精度有关但未能定位触发条件,Barron 的新工作找到了根因并给出修复方案。原观察者确认该论文修复了他们当年未能解决的问题。

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

原文链接 →

「研究」频道最新

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