Kavli Affiliate: Yi Zhou | First 5 Authors: Qi Zhang, Yi Zhou, Shaofeng Zou, , | Summary: This paper provides the first tight convergence analyses for RMSProp and Adam in non-convex optimization under the most relaxed assumptions of coordinate-wise generalized smoothness and affine noise variance. We first analyze RMSProp, which is a special case of […]
Continue.. Convergence Guarantees for RMSProp and Adam in Generalized-smooth Non-convex Optimization with Affine Noise Variance