Adam bias correction compensates for the initial zero values in Adam's exponentially weighted gradient histories. The basic algorithm usesThese factors normalize the coefficient sums of the two running histories.[ref]
Adam: A Method for Stochastic Optimization by Diederik P. Kingma and Jimmy Lei Ba. Algorithm1, printed p2; section3, Initialization Bias Correction, printed p3
For a scalar coordinate, suppose andExpanding the recurrence relation givesIf every has the same expectation , linearity of expectation and the geometric series yieldThus is an unbiased estimator of under these assumptions. Correlation between different times is allowed by this argument. Applying the same calculation to squared gradients with constant raw second moment gives and .
For a constant supplied gradient of 2, choose and . The first uncorrected histories are and , while their corrected values are 2 and 4. The normalization removes the missing coefficient mass caused by initialization.
It does not make a changing history equal to the latest gradient. For an original deterministic example, let , and . Then , sowhich differs from . More generally, changing moment expectations produce a weighted average of past expectations. State the moment assumptions before interpreting “bias-corrected” as unbiased for a present distribution. The nonlinear parameter update built from the corrected histories is a separate object from either moment estimator.
Articles by others on the same topic
There are currently no matching articles.