Adaptive Inertia: Disentangling the Effects of Adaptive Learning Rate
  and Momentum
  Adaptive Moment Estimation (Adam), which combines Adaptive Learning Rate and
Momentum, would be the most popular stochastic optimizer for accelerating the
training of deep neural networks. However, it is empirically known that Adam
often generalizes worse than Stochastic Gradient Descent (SGD). The purpose of
this paper is to unveil the mystery of this behavior in the diffusion
theoretical framework. Specifically, we disentangle the effects of Adaptive
Learning Rate and Momentum of the Adam dynamics on saddle-point escaping and
flat minima selection. We prove that Adaptive Learning Rate can escape saddle
points efficiently, but cannot select flat minima as SGD does. In contrast,
Momentum provides a drift effect to help the training process pass through
saddle points, and almost does not affect flat minima selection. This partly
explains why SGD (with Momentum) generalizes better, while Adam generalizes
worse but converges faster. Furthermore, motivated by the analysis, we design a
novel adaptive optimization framework named Adaptive Inertia, which uses
parameter-wise adaptive inertia to accelerate the training and provably favors
flat minima as well as SGD. Our extensive experiments demonstrate that the
proposed adaptive inertia method can generalize significantly better than SGD
and conventional adaptive gradient methods.
