研究者修复Softmax与交叉熵精度缺陷
AI 总结
论文《A Fix for Softmax and Cross-Entropy》报告了PyTorch、JAX、TensorFlow等标准Softmax与交叉熵实现中的两个浮点缺陷。这些问题可能在分类模型高度自信时错误归零正确类别梯度,也可能在学生模型与教师模型完全一致时留下残余梯度。研究提出的修复保留原有数学定义且无需提高浮点精度,并在模运算与知识蒸馏实验中展现出更平稳的训练表现。
深度见解
这些缺陷涉及AI训练的基础代码,并可能在特定数值条件下悄然改变梯度,但现有证据尚不足以证明其普遍严重影响大模型训练。
机器之心 2026-10-25 13:28 北京
Gemini发现Bug、包揽大量实验。
机器之心编辑部
深度学习中最基础、使用最广泛的两个函数,竟然藏着两个隐蔽的 Bug。
#Google DeepMind#PyTorch#JAX#TensorFlow#Optax#StableMax#Gemini Flash 3.7#Gemini Flash 3.8#Gemini 4 Argon#A Fix for Softmax and Cross-Entropy