DeepMind 的 Jon Barron:所有 softmax 与交叉熵实现都有两个微妙 bug

jon_barron · x · 2026-10-29

Google DeepMind 研究员 Jon Barron 发布新结果:包括 JAX、PyTorch 以及教科书在内的所有现有 softmax 和交叉熵实现都存在两个微妙的 bug。他给出的简单改写可以同时修复这两个问题,并消除 grokking 过程中出现的 "slingshots" 现象。DeepMind 研究主管 Raia Hadsell 转发称这是"质疑一切"的好提醒。

原文链接 →

「研究」频道最新

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