SGD momentum optimizer with step estimation by online parabola model
In stochastic gradient descent, especially for neural network training, there are currently dominating first order methods: not modeling local distance to minimum. This information required for optimal step size is provided by second order methods, however, they have many difficulties, starting with full Hessian having square of dimension number of coefficients. This article proposes a minimal step from successful first order momentum method toward second order: online parabola modelling in just a single direction: normalized v̂ from momentum method. It is done by estimating linear trend of gradients g⃗=∇ F(θ⃗) in v̂ direction: such that g(θ⃗_+θv̂)≈λ (θ -p) for θ = θ⃗·v̂, g= g⃗·v̂, θ⃗_=θ⃗-θv̂. Using linear regression, λ, p are MSE estimated by just updating four averages (of g, θ, gθ, θ^2) in the considered direction. Exponential moving averages allow here for inexpensive online estimation, weakening contribution of the old gradients. Controlling sign of curvature λ, we can repel from saddles in contrast to attraction in standard Newton method. In the remaining directions: not considered in second order model, we can simultaneously perform e.g. gradient descent.
READ FULL TEXT