Adam优化器的Initialization Bias Correction

时间：2024-04-15 11:55:36 浏览：次

Adam

如上图所示，算法中涉及两个滑动平均 $m_t$ 和 $v_t$ 。 $m_t$ 是梯度的滑动平均， $v_t$ 是梯度的平方的的滑动平均。问题出在这两个变量的初始化，它们都被初始化为0，那么在迭代之初和decay非常弱的时候（ $\beta \sim 1$ ）,会出现 0/0，导致溢出。

因此，Adam使用了一个简单的初始化校正。为了纠正滑动平均 $v_t$ 和我们真正想要的 $E[g_t^2]$ 之间的差距，我们来算一下 $E[v_t]$ 。

$v_t$

discrepancy

如果 $E(g_i^2)$ 是平稳的，即每个时刻的均值一样，那么我们有 $\zeta=0$ 。那么唯一的差别就仅有 $1 - \beta_2^t$ 。所以除掉它就好啦！