导读:本期,我们将一同探索由小伙伴原创的《log_sum_exp》。这不仅是一份知识的分享,更凝结了创作者的思考与热情。接下来的内容,将为您清晰梳理其核心脉络与独特价值。如果您从《log_sum_exp》中获得了一丝启发或帮助,您的每一次点赞与转发,都将化为对创作者最直接的认可与支持,让有价值的思想传播得更远。知识因分享而拥有更大能量,感谢您成为这传播链条中的重要一环。
Attention里的Softmax总算溢出怎么办?Softmax数值稳定性优化怎么做 训练大模型时,Attention矩阵里的Softmax经常因为指数运算爆出无穷大,导致梯度变成NaN。直接拿原始分数做exp很容易在数值上失控。常用做法是先减去每行最大值再算指数,配合log_sum_exp技巧避免除零和溢出。本文从原理到实现讲清楚为什么减去最大值不影响结果,以及如何在自注... 栏目:语言推理 时间:08-11 Softmax数值稳定性 Attention计算溢出 log_sum_exp