Entropy-SGD: Biasing Gradient Descent Into Wide Valleys
  This paper proposes a new optimization algorithm called Entropy-SGD for
training deep neural networks that is motivated by the local geometry of the
energy landscape. Local extrema with low generalization error have a large
proportion of almost-zero eigenvalues in the Hessian with very few positive or
negative eigenvalues. We leverage upon this observation to construct a
local-entropy-based objective function that favors well-generalizable solutions
lying in large flat regions of the energy landscape, while avoiding
poorly-generalizable solutions located in the sharp valleys. Conceptually, our
algorithm resembles two nested loops of SGD where we use Langevin dynamics in
the inner loop to compute the gradient of the local entropy before each update
of the weights. We show that the new objective has a smoother energy landscape
and show improved generalization over SGD using uniform stability, under
certain assumptions. Our experiments on convolutional and recurrent networks
demonstrate that Entropy-SGD compares favorably to state-of-the-art techniques
in terms of generalization error and training time.
