Google Inc.

Praneeth Karimireddy

Sanjiv Kumar

Seungyeon Kim

Jingzhao Zhang

Sashank Reddi

Andreas Veit

Generally, the present disclosure is directed to systems and methods that perform adaptive optimization with improved convergence properties. The adaptive optimization techniques described herein are useful in various optimization scenarios, including, for example, training a machine-learned model such as, for example, a neural network. In particular, according to one aspect of the present disclosure, a system implementing the adaptive optimization technique can, over a plurality of iterations, employ an adaptive per coordinate clipping threshold to clip a current first moment of the coordinate to obtain a current update value that enables faster convergence for the machine-learned model when the noise in the stochastic gradients is heavy tailed.

Growing community of inventors

Renens, Switzerland

Praneeth Karimireddy