mlsanityData to 14 Oct 2026

Adaptive Strategies in Non-convex Optimization

arXiv 2306.10278 ↗First posted 17 Jun 2023cs.LG math.OC stat.MLcited by 0

Not on the map. Why?

Abstract (arXiv)

An algorithm is said to be adaptive to a certain parameter (of the problem) if it does not need a priori knowledge of such a parameter but performs competitively to those that know it. This dissertation presents our work on adaptive algorithms in following scenarios: 1. In the stochastic optimization setting, we only receive stochastic gradients and the level of noise in evaluating them greatly affects the convergence rate. Tuning is typically required when without prior knowledge of the noise scale in order to achieve the optimal rate. Considering this, we designed and analyzed noise-adaptive algorithms that can automatically ensure (near)-optimal rates under different noise scales without knowing it. 2. In training deep neural networks, the scales of gradient magnitudes in each coordinate can scatter across a very wide range unless normalization techniques, like BatchNorm, are employed. In such situations, algorithms not addressing this problem of gradient scales can behave very poorly. To mitigate this, we formally established the advantage of scale-free algorithms that adapt to the gradient scales and presented its real benefits in empirical experiments. 3. Traditional analyses in non-convex optimization typically rely on the smoothness assumption. Yet, this condition does not capture the properties of some deep learning objective functions, including the ones involving Long Short-Term Memory networks and Transformers. Instead, they satisfy a much more relaxed condition, with potentially unbounded smoothness. Under this condition, we show that a generalized SignSGD algorithm can theoretically match the best-known convergence rates obtained by SGD with gradient clipping but does not need explicit clipping at all, and it can empirically match the performance of Adam and beat others. Moreover, it can also be made to automatically adapt to the unknown relaxed smoothness.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 42 of its 150 references that are in the map. First posted on arXiv 17 Jun 2023.

Builds on

42 of 150 references.

References · 42

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  4. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  5. 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
  6. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  7. 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,442
  8. 2016Layer Normalizationcited by 5,669
  9. 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
  10. 2016Pointer Sentinel Mixture Modelscited by 3,096
  11. 2017Decoupled Weight Decay Regularizationcited by 1,831
  12. 2018DARTS: Differentiable Architecture Searchcited by 1,798
30 more
  1. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  2. 2020Adaptive Federated Optimizationcited by 1,046
  3. 2019On the Convergence of Adam and Beyondcited by 841
  4. 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
  5. 2014Deeply-Supervised Netscited by 367
  6. 2019Why gradient clipping accelerates training: A theoretical justification for adaptivitycited by 341
  7. 2020Better Theory for SGD in the Nonconvex Worldcited by 158
  8. 2018On the Convergence of Stochastic Gradient Descent with Adaptive Stepsizescited by 156
  9. 2017Dissecting Adam: The Sign, Magnitude and Variance of Stochastic Gradientscited by 139
  10. 2019Painless Stochastic Gradient: Interpolation, Line-Search, and Convergence Ratescited by 119
  11. 2018An Alternative View: When Does SGD Escape Local Minima?cited by 115
  12. 2019Adversarial AutoAugmentcited by 111
  13. 2020Stochastic Optimization with Heavy-Tailed Noise via Accelerated Gradient Clippingcited by 104
  14. 2018Closing the Generalization Gap of Adaptive Gradient Methods in Training Deep Neural Networkscited by 90
  15. 2020Improved Analysis of Clipping Algorithms for Non-convex Optimizationcited by 89
  16. 2022Robustness to Unbounded Smoothness of Generalized SignSGDcited by 81
  17. 2021High-probability Bounds for Non-Convex Stochastic Optimization with Heavy Tailscited by 80
  18. 2020A High Probability Analysis of Adaptive SGD with Momentumcited by 61
  19. 2021Understanding the Generalization of Adam in Learning Neural Networks with Proper Regularizationcited by 49
  20. 2022Understanding AdamW through Proximal Methods and Scale-Freenesscited by 46
  21. 2020Disentangling Adaptive Gradient Methods from Learning Ratescited by 42
  22. 2021Stability and Convergence of Stochastic Gradient Clipping: Beyond Lipschitz Continuity and Smoothnesscited by 37
  23. 2019Why are Adaptive Methods Good for Attention Models?cited by 35
  24. 2020A Second look at Exponential and Cosine Step Sizes: Simplicity, Adaptivity, and Performancecited by 26
  25. 2022A Communication-Efficient Distributed Gradient Clipping Algorithm for Training Deep Neural Networkscited by 20
  26. 2017Comparison of Batch Normalization and Weight Normalization Algorithms for the Large-scale Image Classificationcited by 18
  27. 2019Stochastic algorithms with geometric step decay converge linearly on sharp functionscited by 14
  28. 2019A Generic Acceleration Framework for Stochastic Composite Optimizationcited by 9
  29. 2020Batch Normalization Biases Residual Blocks Towards the Identity Function in Deep Networkscited by 6
  30. 2021Rethinking Adam: A Twofold Exponential Moving Average Approachcited by 6

Built on by

Not cited yet.