On the diffusion approximation of nonconvex stochastic gradient descent
On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima