Generalized Gradient Norm Clipping & Non-Euclidean $(L_0,L_1)$-Smoothness
Pethick, Thomas, Xie, Wanyun, Erdogan, Mete, Antonakopoulos, Kimon, Silveti-Falls, Tony, Cevher, Volkan
This work introduces a hybrid non-Euclidean optimization method which generalizes gradient norm clipping by combining steepest descent and conditional gradient approaches. The method achieves the best of both worlds by establishing a descent property under a generalized notion of ($L_0$,$L_1$)-smoothness. Weight decay is incorporated in a principled manner by identifying a connection to the Frank-Wolfe short step. In the stochastic case, we show an order optimal $O(n^{-1/4})$ convergence rate by leveraging a momentum based gradient estimator. We discuss how to instantiate the algorithms for deep learning and demonstrate their properties on image classification and language modeling.
Jun-3-2025
- Country:
- North America > United States
- Colorado (0.04)
- Asia > Middle East
- Jordan (0.04)
- North America > United States
- Genre:
- Research Report (0.63)
- Industry:
- Government (0.67)
- Technology: