导读:近期更新了《Softmax数值稳定性》的相关内容,包括《Attention里的Softmax总算溢出怎么办?Softmax数值稳定性优化怎么做》。如果 Softmax数值稳定性 对你有帮助,请转发和分享本内容。知识因分享而拥有更大能量,感谢您成为这传播链条中的重要一环。
Attention里的Softmax总算溢出怎么办?Softmax数值稳定性优化怎么做 训练大模型时,Attention矩阵里的Softmax经常因为指数运算爆出无穷大,导致梯度变成NaN。直接拿原始分数做exp很容易在数值上失控。常用做法是先减去每行最大值再算指数,配合log_sum_exp技巧避免除零和溢出。本文从原理到实现讲清楚为什么减去最大值不影响结果,以及如何在自注... 栏目:语言推理 时间:08-11 Softmax数值稳定性 Attention计算溢出 log_sum_exp