On the Convergence of A Class of Adam-Type Algorithms for Non-Convex Optimization
Not on the map. Why?
Abstract (arXiv)
This paper studies a class of adaptive gradient based momentum algorithms that update the search directions and learning rates simultaneously using past gradients. This class, which we refer to as the "Adam-type", includes the popular algorithms such as the Adam, AMSGrad and AdaGrad. Despite their popularity in training deep neural networks, the convergence of these algorithms for solving nonconvex problems remains an open question. This paper provides a set of mild sufficient conditions that guarantee the convergence for the Adam-type methods. We prove that under our derived conditions, these methods can achieve the convergence rate of order O(logT/√T) for nonconvex stochastic optimization. We show the conditions are essential in the sense that violating them may make the algorithm diverge. Moreover, we propose and analyze a class of (deterministic) incremental adaptive gradient algorithms, which has the same O(logT/√T) convergence rate. Our study could also be extended to a broader class of adaptive gradient methods in machine learning and optimization.
Key facts. Cited by 174 papers in the map (data through 14 Oct 2026). First posted on arXiv 8 Aug 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 38; 2022: 33; 2023: 23; 2024: 31; 2025: 25; 2026: 24.
174 papers.
Sort:
By area:
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 174
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 124
- 2022Adam Can Converge Without Any Modification On Update Rulescited by 114
- 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 101
- 2023Convergence of Adam Under Relaxed Assumptionscited by 99
- 2022Robustness to Unbounded Smoothness of Generalized SignSGDcited by 81
- 2022The Power of Adaptivity in SGD: Self-Tuning Step Sizes with Unbounded Gradients and Affine Variancecited by 61
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2023A Theory on Adam Instability in Large-Scale Machine Learningcited by 56
- 2022Communication-Efficient Adaptive Federated Learningcited by 55
- 2023Toward Understanding Why Adam Converges Faster Than SGD for Transformerscited by 53
38 more
- 2021A Novel Convergence Analysis for Algorithms of the Adam Familycited by 50
- 2022Understanding AdamW through Proximal Methods and Scale-Freenesscited by 46
- 2025Muon Outperforms Adam in Tail-End Associative Memory Learningcited by 46
- 2021SUPER-ADAM: Faster and Universal Framework of Adaptive Gradientscited by 40
- 2022High Probability Bounds for a Class of Nonconvex Algorithms with AdaGrad Stepsizecited by 40
- 2023Closing the Gap Between the Upper Bound and the Lower Bound of Adam's Iteration Complexitycited by 38
- 2024LDAdam: Adaptive Optimization from Low-Dimensional Gradient Statisticscited by 38
- 2024The Implicit Bias of Adam on Separable Datacited by 36
- 2021BiAdam: Fast Adaptive Bilevel Optimization Methodscited by 33
- 2025On the Convergence of Muon and Beyondcited by 31
- 2022Provable Adaptivity of Adam under Non-uniform Smoothnesscited by 30
- 2024Understanding Adam Optimizer via Online Learning of Updates: Adam is FTRL in Disguisecited by 26
- 2021Towards Practical Adam: Non-Convexity, Convergence Theory, and Mini-Batch Accelerationcited by 25
- 2021Toward Communication Efficient Adaptive Gradient Methodcited by 25
- 2023Two Sides of One Coin: the Limits of Untuned SGD and the Power of Adaptive Methodscited by 23
- 2024Adam with model exponential moving average is effective for nonconvex optimizationcited by 23
- 2024Adam Exploits ℓ_∞-geometry of Loss Landscape via Coordinate-wise Adaptivitycited by 23
- 2021Does Momentum Change the Implicit Regularization on Separable Data?cited by 22
- 2023AdaSAM: Boosting Sharpness-Aware Minimization with Adaptive Learning Rate and Momentum for Training Deep Neural Networkscited by 21
- 2024On Convergence of Adam for Stochastic Optimization under Relaxed Assumptionscited by 21
- 2022On Distributed Adaptive Optimization with Gradient Compressioncited by 19
- 2024Towards Quantifying the Preconditioning Effect of Adamcited by 19
- 2024ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Ratecited by 18
- 2021On the Convergence of Decentralized Adaptive Gradient Methodscited by 17
- 2025World Models That Know When They Don't Know - Controllable Video Generation with Calibrated Uncertaintycited by 17
- 2022Nest Your Adaptive Algorithm for Parameter-Agnostic Nonconvex Minimax Optimizationcited by 16
- 2024Provable Complexity Improvement of AdaGrad over SGD: Upper and Lower Bounds in Stochastic Non-Convex Optimizationcited by 16
- 2022Maximizing Communication Efficiency for Large-scale Training via 0/1 Adamcited by 15
- 2023FedRC: Tackling Diverse Distribution Shifts Challenge in Federated Learning by Robust Clusteringcited by 15
- 2024Convergence rates for the Adam optimizercited by 15
- 2023Convergence of Adam for Non-convex Objectives: Relaxed Hyperparameters and Non-ergodic Casecited by 14
- 2024Convergence Guarantees for RMSProp and Adam in Generalized-smooth Non-convex Optimization with Affine Noise Variancecited by 14
- 2024AdaFisher: Adaptive Second Order Optimization via Fisher Informationcited by 14
- 2021Unified Convergence Analysis for Adaptive Optimization with Moving Average Estimatorcited by 13
- 2021AdaGDA: Faster Adaptive Gradient Descent Ascent Methods for Minimax Optimizationcited by 13
- 2025Adaptive Preconditioners Trigger Loss Spikes in Adamcited by 13
- 2025Cautious Weight Decaycited by 13
- 2021Communication-Compressed Adaptive Gradient Method for Distributed Nonconvex Optimizationcited by 12
50 most cited of 174 All 174 (JSON)
- 2026Adam under Generalized Smoothness with Second-Moment-Type Stochastic Gradients29 Sep 2026
- 2026Convergence rates for the RMSprop optimizer with full control of the hyperparameters31 Aug 2026
- 2026Second-Moment Memory in Coordinatewise Adam16 Aug 2026
- 2026Unified convergence analysis for gradient descent optimization methods in the training of deep neural networks5 Jul 2026
- 2026Adam Converges in Nonsmooth Nonconvex Optimization21 Jun 2026
- 2026MGUP: A Momentum-Gradient Alignment Update Policy for Stochastic Optimization16 Jun 2026
- 2026Schattor: Schatten-family methods for deep learning optimization14 Jun 2026
- 2026Muon Learns More Robust and Transferable Features than Adam8 Jun 2026
- 2026Why Muon Outperforms Adam: A Curvature Perspective3 Jun 2026
- 2026Understanding Dynamics of Adam in Zero-Sum Games: An ODE Approach19 May 2026
- 2026Adam-SHANG: A Convergent Adam-Type Method for Stochastic Smooth Convex Optimization13 May 2026
- 2026PowerStep: Memory-Efficient Adaptive Optimization via ℓp-Norm Steepest Descent11 May 2026
38 more
- 2026Gradient Extrapolation-Based Policy Optimization7 May 2026
- 2026Accelerating LMO-Based Optimization via Implicit Gradient Transport7 May 2026
- 2026Anon: Extrapolating Adaptivity Beyond SGD and Adam4 May 2026
- 2026AdamO: A Collapse-Suppressed Optimizer for Offline RL3 May 2026
- 2026Cost-Aware Learning30 Apr 2026
- 2026LoDAdaC: a unified local training-based decentralized framework with adaptive gradients and compressed communication11 Apr 2026
- 2026Deterministic Adam-Inspired Methods with Accelerated Convergence Rate9 Apr 2026
- 2026OptEMA: Adaptive Exponential Moving Average for Stochastic Optimization with Zero-Noise Optimality10 Mar 2026
- 2026Towards Understanding Adam Convergence on Highly Degenerate Polynomials10 Mar 2026
- 2026Dynamic Momentum Recalibration in Online Gradient Learning6 Mar 2026
- 2026HomeAdam: Adam and AdamW Algorithms Sometimes Go Home to Obtain Better Provable Generalization3 Mar 2026
- 2026AROpt: An Optimization Method for Autoregressive Time Series Forecasting2 Feb 2026
- 2025World Models That Know When They Don't Know - Controllable Video Generation with Calibrated Uncertainty5 Dec 2025
- 2025HVAdam: A Full-Dimension Adaptive Optimizer25 Nov 2025
- 2025Stragglers Can Contribute More: Uncertainty-Aware Distillation for Asynchronous Federated Learning25 Nov 2025
- 2025Exploring Landscapes for Better Minima along Valleys31 Oct 2025
- 2025Cautious Weight Decay14 Oct 2025
- 2025Muon Outperforms Adam in Tail-End Associative Memory Learning30 Sep 2025
- 2025End-to-End Deep Learning for Predicting Metric Space-Valued Outputs28 Sep 2025
- 2025On the Convergence of Muon and Beyond19 Sep 2025
- 2025Globally aware optimization with resurgence1 Sep 2025
- 2025Compressed Decentralized Momentum Stochastic Gradient Methods for Nonconvex Optimization7 Aug 2025
- 2025Neighbor-Sampling Based Momentum Stochastic Methods for Training Graph Neural Networks1 Aug 2025
- 2025SoftSignSGD(S3): An Enhanced Optimizer for Practical DNN Training and Loss Spikes Minimization Beyond Adam9 Jul 2025
- 2025Simple Convergence Proof of Adam From a Sign-like Descent Perspective8 Jul 2025
- 2025FedDAA: Dynamic Client Clustering for Concept Drift Adaptation in Federated Learning26 Jun 2025
- 2025An Adaptive Method Stabilizing Activations for Enhanced Generalization10 Jun 2025
- 2025Adaptive Preconditioners Trigger Loss Spikes in Adam5 Jun 2025
- 2025Unified Scaling Laws for Compressed Representations2 Jun 2025
- 2025LightSAM: Parameter-Agnostic Sharpness-Aware Minimization30 May 2025
- 2025On the O(frac√dK1/4) Convergence Rate of AdamW Measured by ℓ1 Norm17 May 2025
- 2025Sharp higher order convergence rates for the Adam optimizer28 Apr 2025
- 2025A Langevin sampling algorithm inspired by the Adam optimizer26 Apr 2025
- 2025Adaptive Moment Estimation Optimization Algorithm Using Projection Gradient for Deep Learning13 Mar 2025
- 2025Adaptive Lipschitz-Free Proximal Gradient Methods for Stochastic Composite Nonconvex Optimization28 Feb 2025
- 2025Chemical knowledge-informed framework for privacy-aware retrosynthesis learning26 Feb 2025
- 2025Elucidating Subspace Perturbation in Zeroth-Order Optimization: Theory and Practice at Scale31 Jan 2025
- 2024Scaled Conjugate Gradient Method for Nonconvex Optimization in Deep Neural Networks16 Dec 2024
50 newest of 174 All 174 (JSON)
- 2026Gradient Extrapolation-Based Policy Optimizationcited by 0
- 2026Gradient Extrapolation-Based Policy Optimization7 May 2026
- 2026AdamO: A Collapse-Suppressed Optimizer for Offline RLcited by 1
- 2026AdamO: A Collapse-Suppressed Optimizer for Offline RL3 May 2026
- 2021Gradient Descent on Neural Networks Typically Occurs at the Edge of Stabilitycited by 427
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 174
- 2024Why Transformers Need Adam: A Hessian Perspectivecited by 124
- 2022Adam Can Converge Without Any Modification On Update Rulescited by 114
- 2022Adaptive Gradient Methods at the Edge of Stabilitycited by 101
- 2023Convergence of Adam Under Relaxed Assumptionscited by 99
- 2022Robustness to Unbounded Smoothness of Generalized SignSGDcited by 81
- 2022The Power of Adaptivity in SGD: Self-Tuning Step Sizes with Unbounded Gradients and Affine Variancecited by 61
- 2024Implicit Bias of AdamW: ℓ_∞ Norm Constrained Optimizationcited by 58
- 2023A Theory on Adam Instability in Large-Scale Machine Learningcited by 56
- 2022Communication-Efficient Adaptive Federated Learningcited by 55
- 2023Toward Understanding Why Adam Converges Faster Than SGD for Transformerscited by 53
12 most cited of 168 All 168 (JSON)
- 2026Adam under Generalized Smoothness with Second-Moment-Type Stochastic Gradients29 Sep 2026
- 2026Convergence rates for the RMSprop optimizer with full control of the hyperparameters31 Aug 2026
- 2026Second-Moment Memory in Coordinatewise Adam16 Aug 2026
- 2026Unified convergence analysis for gradient descent optimization methods in the training of deep neural networks5 Jul 2026
- 2026Adam Converges in Nonsmooth Nonconvex Optimization21 Jun 2026
- 2026MGUP: A Momentum-Gradient Alignment Update Policy for Stochastic Optimization16 Jun 2026
- 2026Schattor: Schatten-family methods for deep learning optimization14 Jun 2026
- 2026Muon Learns More Robust and Transferable Features than Adam8 Jun 2026
- 2026Why Muon Outperforms Adam: A Curvature Perspective3 Jun 2026
- 2026Understanding Dynamics of Adam in Zero-Sum Games: An ODE Approach19 May 2026
- 2026Adam-SHANG: A Convergent Adam-Type Method for Stochastic Smooth Convex Optimization13 May 2026
- 2026PowerStep: Memory-Efficient Adaptive Optimization via ℓp-Norm Steepest Descent11 May 2026
12 newest of 168 All 168 (JSON)