Computer Science > Machine Learning
[Submitted on 5 Feb 2024 (v1), last revised 30 Aug 2024 (this version, v9)]
Title:Can We Remove the Square-Root in Adaptive Gradient Methods? A Second-Order Perspective
View PDF HTML (experimental)Abstract:Adaptive gradient optimizers like Adam(W) are the default training algorithms for many deep learning architectures, such as transformers. Their diagonal preconditioner is based on the gradient outer product which is incorporated into the parameter update via a square root. While these methods are often motivated as approximate second-order methods, the square root represents a fundamental difference. In this work, we investigate how the behavior of adaptive methods changes when we remove the root, i.e., strengthen their second-order motivation. Surprisingly, we find that such square-root-free adaptive methods close the generalization gap to SGD on convolutional architectures, while maintaining their root-based counterpart's performance on transformers. The second-order perspective also has practical benefits for developing non-diagonal methods that can incorporate arbitrary curvature approximations through the concept of preconditioner invariance. In contrast to root-based methods like Shampoo, root-free counterparts work well and fast with half-precision since they do not require numerically unstable matrix root decompositions and inversions. Overall, our findings provide new insights into the development of adaptive methods and raise important questions regarding the overlooked role of adaptivity in their success. (experiment code: this https URL optimizer code: this https URL)
Submission history
From: Wu Lin [view email][v1] Mon, 5 Feb 2024 20:15:19 UTC (651 KB)
[v2] Tue, 13 Feb 2024 14:33:15 UTC (646 KB)
[v3] Mon, 4 Mar 2024 03:21:47 UTC (647 KB)
[v4] Tue, 4 Jun 2024 17:47:41 UTC (1,118 KB)
[v5] Mon, 17 Jun 2024 06:23:51 UTC (1,119 KB)
[v6] Thu, 20 Jun 2024 05:29:07 UTC (1,120 KB)
[v7] Mon, 15 Jul 2024 05:23:41 UTC (1,120 KB)
[v8] Sat, 27 Jul 2024 09:32:03 UTC (1,120 KB)
[v9] Fri, 30 Aug 2024 16:45:58 UTC (1,123 KB)
Current browse context:
math
References & Citations
Bibliographic and Citation Tools
Bibliographic Explorer (What is the Explorer?)
Litmaps (What is Litmaps?)
scite Smart Citations (What are Smart Citations?)
Code, Data and Media Associated with this Article
CatalyzeX Code Finder for Papers (What is CatalyzeX?)
DagsHub (What is DagsHub?)
Gotit.pub (What is GotitPub?)
Papers with Code (What is Papers with Code?)
ScienceCast (What is ScienceCast?)
Demos
Recommenders and Search Tools
Influence Flower (What are Influence Flowers?)
Connected Papers (What is Connected Papers?)
CORE Recommender (What is CORE?)
IArxiv Recommender
(What is IArxiv?)
arXivLabs: experimental projects with community collaborators
arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.
Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.
Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.