Beyond adaptive gradient: Fast-Controlled Minibatch Algorithm for large-scale optimization
Not on the map. Why?
Abstract (arXiv)
Adaptive gradient methods have been increasingly adopted by deep learning community due to their fast convergence and reduced sensitivity to hyper-parameters. However, these methods come with limitations, such as increased memory requirements for elements like moving averages and a poorly understood convergence theory. To overcome these challenges, we introduce F-CMA, a Fast-Controlled Mini-batch Algorithm with a random reshuffling method featuring a sufficient decrease condition and a line-search procedure to ensure loss reduction per epoch, along with its deterministic proof of global convergence to a stationary point. To evaluate the F-CMA, we integrate it into conventional training protocols for classification tasks involving both convolutional neural networks and vision transformer models, allowing for a direct comparison with popular optimizers. Computational tests show significant improvements, including a decrease in the overall training time by up to 68%, an increase in per-epoch efficiency by up to 20%, and in model accuracy by up to 5%.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 57 of its 95 references that are in the map. First posted on arXiv 24 Nov 2024.
Builds on
57 of 95 references.
References · 57
- 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
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,795
- 2016Wide Residual Networkscited by 3,070
- 2017Decoupled Weight Decay Regularizationcited by 1,831
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,150
- 2016An overview of gradient descent optimization algorithmscited by 1,033
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 905
- 2018A Convergence Theory for Deep Learning via Over-Parameterizationcited by 787
- 2019On the Variance of the Adaptive Learning Rate and Beyondcited by 772
38 more
- 2013Stochastic First- and Zeroth-order Methods for Nonconvex Stochastic Programmingcited by 724
- 2021A Survey of Transformerscited by 401
- 2016Deep Learning without Poor Local Minimacited by 234
- 2019SGD: General Analysis and Improved Ratescited by 207
- 2018AdaGrad stepsizes: Sharp convergence over nonconvex landscapescited by 182
- 2020A Simple Convergence Proof of Adam and Adagradcited by 171
- 2018On the Convergence of Stochastic Gradient Descent with Adaptive Stepsizescited by 156
- 2020Understanding Gradient Clipping in Private SGD: A Geometric Perspectivecited by 142
- 2017Convergence Analysis of Two-layer Neural Networks with ReLU Activationcited by 137
- 2020Stochastic Polyak Step-size for SGD: An Adaptive Learning Rate for Fast Convergencecited by 137
- 2019Painless Stochastic Gradient: Interpolation, Line-Search, and Convergence Ratescited by 119
- 2020Random Reshuffling: Simple Analysis with Vast Improvementscited by 115
- 2022Adam Can Converge Without Any Modification On Update Rulescited by 114
- 2023Convergence of Adam Under Relaxed Assumptionscited by 99
- 2018SGD and Hogwild! Convergence Without the Bounded Gradients Assumptioncited by 95
- 2016Variance Reduction for Faster Non-Convex Optimizationcited by 72
- 2015From error bounds to the complexity of first-order descent methods for convex functionscited by 43
- 2020Optimal and Practical Algorithms for Smooth and Strongly Convex Decentralized Optimizationcited by 43
- 2023A survey on efficient vision transformers: algorithms, techniques, and performance benchmarkingcited by 41
- 2020Provable Benefits of Overparameterization in Model Compression: From Double Descent to Pruning Neural Networkscited by 40
- 2022On Almost Sure Convergence Rates of Stochastic Gradient Methodscited by 35
- 2020A Qualitative Study of the Dynamic Behavior for Adaptive Gradient Algorithmscited by 33
- 2021Proximal and Federated Random Reshufflingcited by 33
- 2015Probabilistic Line Searches for Stochastic Optimizationcited by 29
- 2019Linear Convergence of Adaptive Stochastic Gradient Descentcited by 29
- 2015Convergence Rate of Incremental Gradient and Newton Methodscited by 27
- 2013Accelerated, Parallel and Proximal Coordinate Descentcited by 25
- 2013Distributed Coordinate Descent Method for Learning with Big Datacited by 23
- 2023Two Sides of One Coin: the Limits of Untuned SGD and the Power of Adaptive Methodscited by 23
- 2023Adaptive SGD with Polyak stepsize and Line-search: Robust Convergence and Variance Reductioncited by 23
- 2015On the Convergence Rate of Incremental Aggregated Gradient Algorithmscited by 21
- 2018Accelerated Bregman Proximal Gradient Methods for Relatively Smooth Convex Optimizationcited by 20
- 2021Lower Bounds and Optimal Algorithms for Smooth and Strongly Convex Decentralized Optimization Over Time-Varying Networkscited by 19
- 2021Learning with Gradient Descent and Weakly Convex Lossescited by 16
- 2021Convergence of Random Reshuffling Under The Kurdyka-Lojasiewicz Inequalitycited by 16
- 2019Parabolic Approximation Line Search for DNNscited by 14
- 2022On the Convergence of AdaGrad(Norm) on Rd: Beyond Convexity, Non-Asymptotic Rate and Accelerationcited by 13
- 2016Surpassing Gradient Descent Provably: A Cyclic Incremental Method with Linear Convergence Ratecited by 12
50 most cited of 57 All 57 (JSON)
Built on by
Not cited yet.