跳到正文
热点事件持续更新

Softmax减去最大值防数值溢出

1 篇报道1 个报道来源1 小时前更新

先了解这件事

AI 综述

cneuralnetwork 演示了数值稳定的 PyTorch Softmax 写法:在计算指数前,从全部 logit 中减去其中的最大值,避免数值很大时 e^x 溢出为无穷大并产生 NaN。 这一做法不改变 Softmax 结果,因为分子和分母会约去相同的 e^(-c) 因子;减去最大值后,最大指数为 e^0=1,其余指数均不超过1。

AI 根据报道生成 · 52 分钟前更新

报道时间线

沿着报道,了解事件的不同侧面。

10月29日
  1. AGI Hunt
    两行 PyTorch 代码实现数值稳定的 softmax:减去最大值防溢出的数学原理

    两行 PyTorch 代码通过从所有 logit 中减去最大值,实现数值稳定的 softmax,避免 x 很大时 e^x 溢出为无穷大并产生 NaN。对所有 logit 减去同一常数不会改变结果,因为 e^(-c) 因子会在分子与分母中约掉;取 c = max(x) 后,最大指数变为 e^0 = 1,其余指数均 ≤1,从而保证训练稳定。

本事件热度走势

还没有足够的连续观测数据,暂不绘制趋势。