跳到正文
新智元· ASI启示录·· 1 天前精选AI 评分70

DeepMind 论文指出 Softmax 与交叉熵标准实现存在舍入 Bug,Gemini 参与发现与修复

AI教科书全错!底层公式惊天Bug曝光,Gemini率先自我修复

AI 导读

谷歌 DeepMind 的 Jonathan T. Barron 团队在 arXiv 提交论文《Softmax和交叉熵的修复》,指出 PyTorch、JAX 及教科书中的标准实现会把极小数字舍入为零,导致梯度消失,称为 Softmax崩溃与 Softmax抵消。

推荐理由

原文解释了浮点舍入如何让梯度归零并给出修复思路,读者可理解低精度训练的一个具体风险来源。

来源:新智元 · mp.weixin.qq.com