Large-Scale Deep Learning Optimizations: A Comprehensive Survey
Not on the map. Why?
Abstract (arXiv)
Deep learning have achieved promising results on a wide spectrum of AI applications. Larger datasets and models consistently yield better performance. However, we generally spend longer training time on more computation and communication. In this survey, we aim to provide a clear sketch about the optimizations for large-scale deep learning with regard to the model accuracy and model efficiency. We investigate algorithms that are most commonly used for optimizing, elaborate the debatable topic of generalization gap arises in large-batch training, and review the SOTA strategies in addressing the communication overhead and reducing the memory footprints.
Key facts. Cited by 10 papers in the map (data through 7 Oct 2026). Builds on 87 of its 125 references that are in the map. First posted on arXiv 1 Nov 2021.
Builds on
87 of 125 references.
References · 87
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,617
- 2017Mask R-CNNcited by 7,259
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2014Fully Convolutional Networks for Semantic Segmentationcited by 4,845
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
38 more
- 2014Going Deeper with Convolutionscited by 4,231
- 2015Fast R-CNNcited by 2,930
- 2019ALBERT: A Lite BERT for Self-supervised Learning of Language Representationscited by 2,772
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,479
- 2016Google's Neural Machine Translation System: Bridging the Gap between Human and Machine Translationcited by 1,471
- 2020Sharpness-Aware Minimization for Efficiently Improving Generalizationcited by 1,404
- 2019SpecAugment: A Simple Data Augmentation Method for Automatic Speech Recognitioncited by 1,355
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,337
- 2016Optimization Methods for Large-Scale Machine Learningcited by 1,140
- 2017Mixed Precision Trainingcited by 1,135
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,126
- 2017Neural Collaborative Filteringcited by 1,108
- 2016An overview of gradient descent optimization algorithmscited by 1,021
- 2020On Layer Normalization in the Transformer Architecturecited by 906
- 2018Adafactor: Adaptive Learning Rates with Sublinear Memory Costcited by 896
- 2019On the Convergence of Adam and Beyondcited by 834
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
- 2019On the Variance of the Adaptive Learning Rate and Beyondcited by 763
- 2019Large Batch Optimization for Deep Learning: Training BERT in 76 minutescited by 671
- 2018Local SGD Converges Fast and Communicates Littlecited by 595
- 2018signSGD: Compressed Optimisation for Non-Convex Problemscited by 561
- 2017Large Batch Training of Convolutional Networkscited by 529
- 2017Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Trainingcited by 471
- 2017Sharp Minima Can Generalize For Deep Netscited by 462
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2018Shampoo: Preconditioned Stochastic Tensor Optimizationcited by 413
- 2018Sparsified SGD with Memorycited by 373
- 2016Entropy-SGD: Biasing Gradient Descent Into Wide Valleyscited by 357
- 2017Don't Decay the Learning Rate, Increase the Batch Sizecited by 313
- 2019Error Feedback Fixes SignSGD and other Gradient Compression Schemescited by 306
- 2017Train longer, generalize better: closing the generalization gap in large batch training of neural networkscited by 278
- 2017Three Factors Influencing Minima in SGDcited by 270
- 2014One weird trick for parallelizing convolutional neural networkscited by 264
- 2017TernGrad: Ternary Gradients to Reduce Communication in Distributed Deep Learningcited by 253
- 2017The Marginal Value of Adaptive Gradient Methods in Machine Learningcited by 248
- 2017Sparse Communication for Distributed Gradient Descentcited by 243
- 2015Escaping From Saddle Points --- Online Stochastic Gradient for Tensor Decompositioncited by 241
- 2018The Convergence of Sparsified Gradient Methodscited by 228
50 most cited of 87 All 87 (JSON)
Built on by
Citing papers per year: 2022: 4; 2023: 3; 2024: 1; 2025: 1; 2026: 1.
10 papers.
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Modelscited by 172
- 2023CAME: Confidence-guided Adaptive Memory Efficient Optimizationcited by 34
- 2023On Efficient Training of Large-Scale Deep Learning Models: A Literature Reviewcited by 26
- 2022CowClip: Reducing CTR Prediction Model Training Time from 12 hours to 10 minutes on 1 GPUcited by 4
- 2022Refiner: Data Refining against Gradient Leakage Attacks in Federated Learningcited by 2
- 2025Synthetic Human Action Video Data Generation with Pose Transfercited by 2
- 2022Benchmark Assessment for DeepSpeed Optimization Librarycited by 1
- 2024Towards Communication-efficient Federated Learning via Sparse and Aligned Adaptive Optimizationcited by 1
- 2023Accurate and Fast Fischer-Tropsch Reaction Microkinetics using PINNscited by 0
- 2026Relative Generalization Invariance of LLM Pretrainingcited by 0
- 2026Relative Generalization Invariance of LLM Pretraining26 Sep 2026
- 2025Synthetic Human Action Video Data Generation with Pose Transfer11 Jun 2025
- 2024Towards Communication-efficient Federated Learning via Sparse and Aligned Adaptive Optimization28 May 2024
- 2023Accurate and Fast Fischer-Tropsch Reaction Microkinetics using PINNs17 Nov 2023
- 2023CAME: Confidence-guided Adaptive Memory Efficient Optimization5 Jul 2023
- 2023On Efficient Training of Large-Scale Deep Learning Models: A Literature Review7 Apr 2023
- 2022Refiner: Data Refining against Gradient Leakage Attacks in Federated Learning5 Dec 2022
- 2022Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models13 Aug 2022
- 2022CowClip: Reducing CTR Prediction Model Training Time from 12 hours to 10 minutes on 1 GPU13 Apr 2022
- 2022Benchmark Assessment for DeepSpeed Optimization Library12 Feb 2022