跳到正文
机器之心·· 5 小时前精选AI 评分73

Google DeepMind 研究揭示 Softmax 和交叉熵的隐藏数值 Bug

PyTorch、JAX都被点名!Softmax曝出隐藏Bug,Gemini立功

AI 导读

Google DeepMind 团队提出主流深度学习框架中 Softmax 和交叉熵存在数值精度 Bug,可能导致梯度计算错误。论文提供了保留原有数学定义的修复方案,在模运算等特定任务中提升了训练稳定性,研究过程有 Gemini 模型参与。

推荐理由

揭示了主流框架中Softmax和交叉熵的隐藏数值Bug,解释了训练异常现象,并给出了低成本修复方案。

来源:机器之心 · mp.weixin.qq.com