Towards Understanding Adam Convergence on Highly Degenerate Polynomials
Abstract
Lay Summary
Training deep learning models is essentially like navigating a complex, hilly landscape to find the lowest valley. While the Adam optimizer is the most popular "vehicle" for this journey, researchers still don't fully understand exactly where and why it naturally outperforms older methods. We answer this by studying Adam's behavior when it encounters highly "degenerated" valleys—areas of the landscape so pathologically flat that traditional algorithms lose all sense of direction and stall. Our paper presents the exciting result that Adam possesses a built-in "auto-acceleration" mechanism when navigating these extremely flat, degenerate regions. Traditionally, engineers thought algorithms needed manual speed controllers to escape such areas—akin to constantly shifting gears to get out of a flat, muddy plain. Instead, we show Adam naturally separates its memory of past progress from immediate obstacles, speeding up exactly when older methods crawl. We also map out how Adam's starting settings dictate its behavior to avoid erratic bouncing. Our findings reveal that these degenerate, flat directions are actually a crucial key to understanding Adam's acceleration over traditional methods like Stochastic Gradient Descent (SGD). Ultimately, this provides a strong foundation for designing even better optimizers in the future and deepens our understanding of how network architectures interact with their complex training landscapes.