arXiv · 2211.03970
On the Algorithmic Stability and Generalization of Adaptive Optimization Methods
Abstract
Despite their popularity in deep learning and machine learning in general, the theoretical properties of adaptive optimizers such as Adagrad, RMSProp, Adam or AdamW are not yet fully understood. In this paper, we develop a novel framework to study the stability and generalization of these optimization methods. Based on this framework, we show provable guarantees about such properties that depend heavily on a single parameter $\beta_2$. Our empirical experiments support our claims and provide practical insights into the stability and generalization properties of adaptive optimization methods.
Explore related subjects
Keep this discovery
Han Nguyen, Hai Pham, Sashank J. Reddi, Barnabás Póczos. 2022-11-08. On the Algorithmic Stability and Generalization of Adaptive Optimization Methods. https://arxiv.org/abs/2211.03970
Cite the original work for its findings. Save a collection to share your selection of sources.