Evolution of Optimization Methods: Algorithms, Scenarios, and Evaluations
Organizations: Zhejiang University, APRIL Lab · Fudan University · Shanghai Jiaotong University · National University of Singapore
Abstract
Balancing convergence speed, generalization capability, and computational efficiency remains a core challenge in deep learning optimization. First-order gradient descent methods, epitomized by stochastic gradient descent (SGD) and Adam, serve as the cornerstone of modern training pipelines. However, large-scale model training, stringent differential privacy requirements, and distributed learning paradigms expose critical limitations in these conventional approaches regarding privacy protection and memory efficiency. To mitigate these bottlenecks, researchers explore second-order optimization techniques to surpass first-order performance ceilings, while zeroth-order methods reemerge to alleviate memory constraints inherent to large-scale training. Despite this proliferation of methodologies, the field lacks a cohesive framework that unifies underlying principles and delineates application scenarios for these disparate approaches. In this work, we retrospectively analyze the evolutionary trajectory of deep learning optimization algorithms and present a comprehensive empirical evaluation of mainstream optimizers across diverse model architectures and training scenarios. We distill key emerging trends and fundamental design trade-offs, pinpointing promising directions for future research. By synthesizing theoretical insights with extensive empirical evidence, we provide actionable guidance for designing next-generation highly efficient, robust, and trustworthy optimization methods. The code is available at https://github.com/APRIL-AIGC/Awesome-Optimizer.
Figures & tables
| Symbol | Description |
| Model parameters (weights and biases) evaluated at iteration . | |
| Learning rate, determining the step size of parameter updates . | |
| Exact gradient of the objective function with respect to parameters at step . | |
| Stochastic gradient (mini-batch gradient) of the objective function at step . | |
| Biased first moment estimate (exponential moving average of past gradients) . | |
| Biased second raw moment estimate (exponential moving average of past squared gradients) . |
| Optimizer | Venue | Type | Limitation | Highlight |
| First-Order Methods ( Sec. 3.2 ) | ||||
| SGDM [ 208 ] | Nature’1986 | Accelerating Convergence Rate | Complex hyperparameter tuning. | First proposed to introduce a momentum mechanism to accelerate convergence. |
| Adam [ 123 ] | ICLR’15 | Strategies for Adaptive Step-Size Estimation | High memory usage, worse generalization. | First proposed a method combining adaptive step-size estimation with momentum. |
| AdaBelief [ 344 ] | NeurIPS’20 | Strategies for Adaptive Step-Size Estimation | requires task-specific tuning. | Compared to adaptive methods like Adam, it achieves better generalization. |
| Shampoo [ 83 ] | ICML’18 | Approximating Curvature Information | Computationally expensive. | First proposed to exploit tensor structures to maintain preconditioners |
| 4-bit Shampoo [ 281 ] | NeurIPS’24 | Approximating Curvature Information | Computationally expensive. | Compared to standard Shampoo, it significantly reduces memory usage by quantizing optimizer states to 4-bit. |
| Procedure | DeiT | A2 | A3 |
| Dataset | IN-1K | IN-1K | IN-1K |
| Epochs | 300 | 300 | 100 |
| Batch size | 1024 | 1024 | 1024 |
| Optimizer | AdamW | AdamW | AdamW |
| Learning rate | |||
| Optimizer Momentum | 0.9, 0.999 | 0.9, 0.999 | 0.9, 0.999 |
| Optimizer | 0.1 LR | 0.2 LR | 1.0 LR | 5.0 LR | 10.0 LR | Best LR |
| SGD [ 205 ] | 48.056 | 55.918 | 64.360 | 60.590 | 59.262 | 0.1 |
| RMSprop [ 256 ] | 54.254 | 43.416 | 20.512 | 6.1140 | - | 1e-4 |
| Adam [ 123 ] | 66.234 | 70.772 | 72.240 | - | - | 1e-3 |
| AdaBelief [ 344 ] | 29.248 | 40.210 | 58.562 | 66.528 | 68.642 | 1e-2 |
| Adafactor [ 223 ] | 66.294 | 70.834 | 74.630 | 73.344 | - | 1e-3 |
| RAdam [ 148 ] | 66.220 | 70.826 | 71.190 | - | - | 1e-3 |
| Optimizer | ViT-S | ResNet-50 | Llama-60m | ||
| Acc (%) | Acc (%) | Base LR | 5.0 LR | 0.2 LR | |
| SGD [ 205 ] | 64.360 | 75.926 | - | - | - |
| RMSprop [ 256 ] | 54.254 | 71.536 | 522.57 | 269.27 | 1178.34 |
| Adam [ 123 ] | 72.240 | 74.480 | 614.93 | 357.95 | 964.46 |
| AdaBelief [ 344 ] | 68.642 | 74.232 | 43.27 | 41.89 | 72.43 |
| Adafactor [ 223 ] | 74.630 | 76.808 | 13.78 | 16.55 | 16.48 |
| Model | Optimizer | 100 Epoch Acc (%) | 300 Epoch Acc (%) | Improvement ( ) |
| ViT-S | SGD [ 205 ] | 64.360 | 73.774 | + 9.414 |
| RMSprop [ 256 ] | 54.254 | 57.090 | + 2.836 | |
| Adam [ 123 ] | 72.240 | 77.274 | + 5.034 | |
| AdaBelief [ 344 ] | 68.642 | 74.640 | + 5.998 | |
| Adafactor [ 223 ] | 74.630 | 80.108 | + 5.478 | |
| RAdam [ 148 ] | 71.190 | 77.032 | + 5.842 |
| Optimizer | Convergence | Accuracy | Memory | Generalization | Hyperparameter |
| Speed | Efficiency | Robustness | |||
| SGD [ 205 ] | ★ | ★★ | ★★★★★ | ★★ | ★ |
| RMSprop [ 256 ] | ★ | ★ | ★★★★ | ★★ | ★★ |
| Adam [ 123 ] | ★★★ | ★★★ | ★★★ | ★★ | ★★★ |
| AdaBelief [ 344 ] | ★★★★ | ★★ | ★★★ | ★★ | ★★ |
| Adafactor [ 223 ] | ★★ | ★★★ | ★★★★ | ★★★ | ★★★ |
| Benchmark Task | Hardware Type | Hyperparameter Search (5 LR Scales 23 Optimizers) | Final Evaluation (Best LR 23 Optimizers) | Total GPU Hours |
| ResNet-50 [ 87 ] (ImageNet-1K [ 103 ] ) | NVIDIA A100 | - | 288 Hours (100/300 epochs) | 288 Hours |
| ViT-S [ 49 ] (ImageNet-1K [ 103 ] ) | NVIDIA A100 | 544 Hours (100 epochs) | 213 Hours (300 epochs) | 757 Hours |
| LlaMA-60M [ 153 ] (WikiText-103 [ 172 ] ) | NVIDIA A100 | 28 Hours (Chinchilla scaling) | - | 28 Hours |
| Total Aggregated Cost | Equivalent to 45 Days on a single NVIDIA A100 GPU | 1073 Hours | ||
Deep Learning Optimization
Deep learning optimization develops algorithms and training strategies that help neural networks converge, generalize, and use compute efficiently.
See all 204 papers