Robustness to Unbounded Smoothness of Generalized SignSGD
Not on the map. Why?
Abstract (arXiv)
Traditional analyses in non-convex optimization typically rely on the smoothness assumption, namely requiring the gradients to be Lipschitz. However, recent evidence shows that this smoothness condition does not capture the properties of some deep learning objective functions, including the ones involving Recurrent Neural Networks and LSTMs. Instead, they satisfy a much more relaxed condition, with potentially unbounded smoothness. Under this relaxed assumption, it has been theoretically and empirically shown that the gradient-clipped SGD has an advantage over the vanilla one. In this paper, we show that clipping is not indispensable for Adam-type algorithms in tackling such scenarios: we theoretically prove that a generalized SignSGD algorithm can obtain similar convergence rates as SGD with clipping but does not need explicit clipping at all. This family of algorithms on one end recovers SignSGD and on the other end closely resembles the popular Adam algorithm. Our analysis underlines the critical role that momentum plays in analyzing SignSGD-type and Adam-type algorithms: it not only reduces the effects of noise, thus removing the need for large mini-batch in previous analyses of SignSGD-type algorithms, but it also substantially reduces the effects of unbounded smoothness and gradient norms. We also compare these algorithms with popular optimizers on a set of deep learning tasks, observing that we can match the performance of Adam while beating the others.
Key facts. Cited by 81 papers in the map (data through 14 Oct 2026). Builds on 39 of its 63 references that are in the map. First posted on arXiv 23 Aug 2022.
Builds on
39 of 63 references.
References · 39
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2016Pointer Sentinel Mixture Modelscited by 3,096
- 2021Convex optimizationcited by 2,515
- 2020Adaptive Federated Optimizationcited by 1,046
- 2019On the Convergence of Adam and Beyondcited by 841
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 571
- 2014Deeply-Supervised Netscited by 367
27 more
- 2019Why gradient clipping accelerates training: A theoretical justification for adaptivitycited by 341
- 2019Momentum-Based Variance Reduction in Non-Convex SGDcited by 306
- 2017Regularizing and Optimizing LSTM Language Modelscited by 205
- 2018AdaGrad stepsizes: Sharp convergence over nonconvex landscapescited by 182
- 2018On the Convergence of A Class of Adam-Type Algorithms for Non-Convex Optimizationcited by 174
- 2020A Simple Convergence Proof of Adam and Adagradcited by 171
- 2020Better Theory for SGD in the Nonconvex Worldcited by 158
- 2018On the Convergence of Stochastic Gradient Descent with Adaptive Stepsizescited by 156
- 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 139
- 2020Momentum Improves Normalized SGDcited by 139
- 2017Non-convex Optimization for Machine Learningcited by 133
- 2018A Sufficient Condition for Convergences of Adam and RMSPropcited by 126
- 2020Stochastic Optimization with Heavy-Tailed Noise via Accelerated Gradient Clippingcited by 104
- 2018Tight Analyses for Non-Smooth Stochastic Gradient Descentcited by 97
- 2020Improved Analysis of Clipping Algorithms for Non-convex Optimizationcited by 89
- 2016Scale-Free Online Learningcited by 85
- 2021High-probability Bounds for Non-Convex Stochastic Optimization with Heavy Tailscited by 80
- 2015Beyond Convexity: Stochastic Quasi-Convex Optimizationcited by 76
- 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 49
- 2022Understanding AdamW through Proximal Methods and Scale-Freenesscited by 46
- 2021Non-convex Distributionally Robust Optimization: Non-asymptotic Analysiscited by 39
- 2021Stability and Convergence of Stochastic Gradient Clipping: Beyond Lipschitz Continuity and Smoothnesscited by 37
- 2015Scale-Free Algorithms for Online Linear Optimizationcited by 30
- 2018Efficient Decentralized Deep Learning by Dynamic Model Averagingcited by 30
- 2020LaProp: Separating Momentum and Adaptivity in Adamcited by 20
- 2018Universal Stagewise Learning for Non-Convex Problems with Convergence on Averaged Solutionscited by 9
- 2021Rethinking Adam: A Twofold Exponential Moving Average Approachcited by 6
Built on by
Citing papers per year: 2022: 1; 2023: 15; 2024: 31; 2025: 17; 2026: 17.
81 papers.
Sort:
By area:
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 265
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 124
- 2023Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Becited by 114
- 2023Convergence of Adam Under Relaxed Assumptionscited by 99
- 2023Linear attention is (maybe) all you need (to understand transformer optimization)cited by 98
- 2024Heavy-Tailed Class Imbalance and Why Adam Outperforms Gradient Descent on Language Modelscited by 80
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 72
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2023Convergence of AdaGrad for Non-convex Objectives: Simple Proofs and Relaxed Assumptionscited by 57
- 2025ASGO: Adaptive Structured Gradient Optimizationcited by 57
- 2023Lion Secretly Solves Constrained Optimization: As Lyapunov Predictscited by 51
- 2024From Gradient Clipping to Normalization for Heavy Tailed SGDcited by 49
38 more
- 2023Convex and Non-convex Optimization Under Generalized Smoothnesscited by 44
- 2024Nonconvex Stochastic Optimization under Heavy-Tailed Noises: Optimal Convergence without Gradient Clippingcited by 42
- 2023Beyond Uniform Smoothness: A Stopped Analysis of Adaptive SGDcited by 41
- 2025Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noisecited by 41
- 2023Closing the Gap Between the Upper Bound and the Lower Bound of Adam's Iteration Complexitycited by 38
- 2024Does SGD really happen in tiny subspaces?cited by 36
- 2023Parameter-Agnostic Optimization under Relaxed Smoothnesscited by 29
- 2024Understanding Adam Optimizer via Online Learning of Updates: Adam is FTRL in Disguisecited by 26
- 2024Adam with model exponential moving average is effective for nonconvex optimizationcited by 23
- 2023Variance-reduced Clipping for Non-convex Optimizationcited by 22
- 2024AdaGrad under Anisotropic Smoothnesscited by 22
- 2024On Convergence of Adam for Stochastic Optimization under Relaxed Assumptionscited by 21
- 2024SWAN: SGD with Normalization and Whitening Enables Stateless LLM Trainingcited by 20
- 2024Towards Quantifying the Preconditioning Effect of Adamcited by 19
- 2025SUMO: Subspace-Aware Moment-Orthogonalization for Accelerating Memory-Efficient LLM Trainingcited by 19
- 2024Communication Efficient Distributed Training with Distributed Lioncited by 18
- 2025Generalized Gradient Norm Clipping & Non-Euclidean (L0,L1)-Smoothnesscited by 17
- 2024Provable Complexity Improvement of AdaGrad over SGD: Upper and Lower Bounds in Stochastic Non-Convex Optimizationcited by 16
- 2024Bilevel Optimization under Unbounded Smoothness: A New Algorithm and Convergence Analysiscited by 15
- 2024Revisiting Convergence of AdaGrad with Relaxed Assumptionscited by 15
- 2024On the Convergence of Adam under Non-uniform Smoothness: Separability from SGDM and Beyondcited by 14
- 2024Convergence Guarantees for RMSProp and Adam in Generalized-smooth Non-convex Optimization with Affine Noise Variancecited by 14
- 2025Sign Operator for Coping with Heavy-Tailed Noise in Non-Convex Optimization: High Probability Bounds Under (L0, L1)-Smoothnesscited by 13
- 2024Gradient-Variation Online Learning under Generalized Smoothnesscited by 11
- 2024Understanding Adam Requires Better Rotation Dependent Assumptionscited by 11
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 9
- 2025On the O(frac√dK1/4) Convergence Rate of AdamW Measured by ℓ1 Normcited by 8
- 2023High-probability Convergence Bounds for Nonlinear Stochastic Gradient Descent Under Heavy-tailed Noisecited by 7
- 2024MGDA Converges under Generalized Smoothness, Provablycited by 7
- 2024Large Deviation Upper Bounds and Improved MSE Rates of Nonlinear SGD: Heavy-tailed Noise and Power of Symmetrycited by 7
- 2024A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothnesscited by 7
- 2025Unbiased Gradient Low-Rank Projectioncited by 7
- 2026On Surprising Effectiveness of Masking Updates in Adaptive Optimizerscited by 7
- 2026MUON+: Towards More Effective Muon via One Additional Normalization Step for LLM Pre-trainingcited by 6
- 2024R-SFLLM: Jamming Resilient Framework for Split Federated Learning with Large Language Modelscited by 5
- 2026StoSignSGD: Unbiased Structural Stochasticity Fixes SignSGD for Training Large Language Modelscited by 5
- 2026Optimizer-Model Consistency: Full Finetuning with the Same Optimizer as Pretraining Forgets Lesscited by 5
- 2023Rethinking SIGN Training: Provable Nonconvex Acceleration without First- and Second-Order Gradient Lipschitzcited by 4
50 most cited of 81 All 81 (JSON)
- 2026Adam under Generalized Smoothness with Second-Moment-Type Stochastic Gradients29 Sep 2026
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Training28 Sep 2026
- 2026Optimizers for Diffusion Models: A Controlled Benchmark19 Sep 2026
- 2026Why Clipping Matters in AdaGrad? Toward a High-Probability Theory under Generalized Smoothness12 Aug 2026
- 2026Unified convergence analysis for gradient descent optimization methods in the training of deep neural networks5 Jul 2026
- 2026Free Heavy-Tailed Lunch for Muon: A Theoretical Justification of Empirical Success12 Jun 2026
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Training19 May 2026
- 2026Ringmaster LMO: Asynchronous Linear Minimization Oracle Momentum Method18 May 2026
- 2026Stochastic Non-Smooth Convex Optimization with Unbounded Gradients15 May 2026
- 2026Optimizer-Model Consistency: Full Finetuning with the Same Optimizer as Pretraining Forgets Less7 May 2026
- 2026When and Why SignSGD Outperforms SGD: A Theoretical Study Based on ℓ1-norm Lower Bounds7 May 2026
- 2026StoSignSGD: Unbiased Structural Stochasticity Fixes SignSGD for Training Large Language Models16 Apr 2026
38 more
- 2026Byzantine-Robust Optimization under (L0, L1)-Smoothness12 Mar 2026
- 2026Never Saddle for Reparameterized Steepest Descent as Mirror Flow2 Mar 2026
- 2026MUON+: Towards More Effective Muon via One Additional Normalization Step for LLM Pre-training25 Feb 2026
- 2026On Surprising Effectiveness of Masking Updates in Adaptive Optimizers17 Feb 2026
- 2026Provably Convergent Decentralized Optimization over Directed Graphs under Generalized Smoothness7 Jan 2026
- 2025Alada: Alternating Adaptation of Momentum Method for Memory-Efficient Matrix Optimization15 Dec 2025
- 2025Unbiased Gradient Low-Rank Projection20 Oct 2025
- 2025Decentralized Stochastic Nonconvex Optimization under the (L0,L1)-Smoothness10 Sep 2025
- 2025Theoretical Analysis on how Learning Rate Warmup Accelerates Convergence9 Sep 2025
- 2025Near-Optimal Convergence of Accelerated Gradient Methods under Generalized and (L0, L1)-Smoothness9 Aug 2025
- 2025Revisiting Convergence: Shuffling Complexity Beyond Lipschitz Smoothness11 Jul 2025
- 2025Simple Convergence Proof of Adam From a Sign-like Descent Perspective8 Jul 2025
- 2025Lions and Muons: Optimization via Stochastic Frank-Wolfe under Heavy-Tailed Noise4 Jun 2025
- 2025Generalized Gradient Norm Clipping & Non-Euclidean (L0,L1)-Smoothness2 Jun 2025
- 2025On the Interaction of Batch Noise, Adaptivity, and Compression, under (L0,L1)-Smoothness: An SDE Approach30 May 2025
- 2025SUMO: Subspace-Aware Moment-Orthogonalization for Accelerating Memory-Efficient LLM Training30 May 2025
- 2025LightSAM: Parameter-Agnostic Sharpness-Aware Minimization30 May 2025
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)19 May 2025
- 2025On the O(frac√dK1/4) Convergence Rate of AdamW Measured by ℓ1 Norm17 May 2025
- 2025ASGO: Adaptive Structured Gradient Optimization26 Mar 2025
- 2025On the Convergence of Adam-Type Algorithm for Bilevel Optimization under Unbounded Smoothness5 Mar 2025
- 2025Sign Operator for Coping with Heavy-Tailed Noise in Non-Convex Optimization: High Probability Bounds Under (L0, L1)-Smoothness11 Feb 2025
- 2024A Nearly Optimal Single Loop Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Dec 2024
- 2024Nonconvex Stochastic Optimization under Heavy-Tailed Noises: Optimal Convergence without Gradient Clipping27 Dec 2024
- 2024SWAN: SGD with Normalization and Whitening Enables Stateless LLM Training17 Dec 2024
- 2024Understanding Adam Requires Better Rotation Dependent Assumptions25 Oct 2024
- 2024Error Feedback under (L0,L1)-Smoothness: Normalization and Momentum22 Oct 2024
- 2024Large Deviation Upper Bounds and Improved MSE Rates of Nonlinear SGD: Heavy-tailed Noise and Power of Symmetry21 Oct 2024
- 2024Adaptive Gradient Normalization and Independent Sampling for (Stochastic) Generalized-Smooth Optimization17 Oct 2024
- 2024Nonlinear Stochastic Gradient Descent and Heavy-tailed Noise: A Unified Framework and High-probability Guarantees17 Oct 2024
- 2024From Gradient Clipping to Normalization for Heavy Tailed SGD17 Oct 2024
- 2024Extended convexity and smoothness and their applications in deep learning8 Oct 2024
- 2024An Accelerated Algorithm for Stochastic Bilevel Optimization under Unbounded Smoothness28 Sep 2024
- 2024Gradient-Variation Online Learning under Generalized Smoothness17 Aug 2024
- 2024R-SFLLM: Jamming Resilient Framework for Split Federated Learning with Large Language Models16 Jul 2024
- 2024AdaGrad under Anisotropic Smoothness21 Jun 2024
- 2024Provable Complexity Improvement of AdaGrad over SGD: Upper and Lower Bounds in Stochastic Non-Convex Optimization7 Jun 2024
- 2024SPAM: Stochastic Proximal Point Method with Momentum Variance Reduction for Non-convex Cross-Device Federated Learning30 May 2024
50 newest of 81 All 81 (JSON)
- 2026Optimizers for Diffusion Models: A Controlled Benchmarkcited by 0
- 2026Optimizers for Diffusion Models: A Controlled Benchmark19 Sep 2026
- 2023Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-trainingcited by 265
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 124
- 2023Noise Is Not the Main Factor Behind the Gap Between SGD and Adam on Transformers, but Sign Descent Might Becited by 114
- 2023Convergence of Adam Under Relaxed Assumptionscited by 99
- 2023Linear attention is (maybe) all you need (to understand transformer optimization)cited by 98
- 2024Heavy-Tailed Class Imbalance and Why Adam Outperforms Gradient Descent on Language Modelscited by 80
- 2025Gluon: Making Muon & Scion Great Again! (Bridging Theory and Practice of LMO-based Optimizers for LLMs)cited by 72
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2023Convergence of AdaGrad for Non-convex Objectives: Simple Proofs and Relaxed Assumptionscited by 57
- 2025ASGO: Adaptive Structured Gradient Optimizationcited by 57
- 2023Lion Secretly Solves Constrained Optimization: As Lyapunov Predictscited by 51
- 2024From Gradient Clipping to Normalization for Heavy Tailed SGDcited by 49
12 most cited of 80 All 80 (JSON)
- 2026Adam under Generalized Smoothness with Second-Moment-Type Stochastic Gradients29 Sep 2026
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Training28 Sep 2026
- 2026Why Clipping Matters in AdaGrad? Toward a High-Probability Theory under Generalized Smoothness12 Aug 2026
- 2026Unified convergence analysis for gradient descent optimization methods in the training of deep neural networks5 Jul 2026
- 2026Free Heavy-Tailed Lunch for Muon: A Theoretical Justification of Empirical Success12 Jun 2026
- 2026LionMuon: Alternating Spectral and Sign Descent for Efficient Training19 May 2026
- 2026Ringmaster LMO: Asynchronous Linear Minimization Oracle Momentum Method18 May 2026
- 2026Stochastic Non-Smooth Convex Optimization with Unbounded Gradients15 May 2026
- 2026Optimizer-Model Consistency: Full Finetuning with the Same Optimizer as Pretraining Forgets Less7 May 2026
- 2026When and Why SignSGD Outperforms SGD: A Theoretical Study Based on ℓ1-norm Lower Bounds7 May 2026
- 2026StoSignSGD: Unbiased Structural Stochasticity Fixes SignSGD for Training Large Language Models16 Apr 2026
- 2026Byzantine-Robust Optimization under (L0, L1)-Smoothness12 Mar 2026
12 newest of 80 All 80 (JSON)