Extremely Large Minibatch SGD: Training ResNet-50 on ImageNet in 15 Minutes
Not on the map. Why?
Abstract (arXiv)
We demonstrate that training ResNet-50 on ImageNet for 90 epochs can be achieved in 15 minutes with 1024 Tesla P100 GPUs. This was made possible by using a large minibatch size of 32k. To maintain accuracy with this large minibatch size, we employed several techniques such as RMSprop warm-up, batch normalization without moving averages, and a slow-start learning rate schedule. This paper also describes the details of the hardware and software of the system used to achieve the above performance.
Key facts. Cited by 22 papers in the map (data through 14 Oct 2026). First posted on arXiv 12 Nov 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 4; 2022: 8; 2023: 6; 2024: 3; 2025: 0; 2026: 1.
22 papers.
Sort:
By area:
- 2022Towards Efficient and Scalable Sharpness-Aware Minimizationcited by 129
- 2023FFCV: Accelerating Training by Removing Data Bottleneckscited by 49
- 2021ResIST: Layer-Wise Decomposition of ResNets for Distributed Trainingcited by 18
- 2022Auto-scaling Vision Transformers without Trainingcited by 15
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compressioncited by 15
- 2022Numerical Optimizations for Weighted Low-rank Estimation on Language Modelcited by 14
- 2023Knowledge Distillation in Vision Transformers: A Critical Reviewcited by 13
- 2021Concurrent Adversarial Learning for Large-Batch Trainingcited by 11
- 2021Combining a Convolutional Neural Network with Autoencoders to Predict the Survival Chance of COVID-19 Patientscited by 5
- 2021PAFNet: An Efficient Anchor-Free Object Detector Guidancecited by 5
- 2024Communication-Efficient Adaptive Batch Size Strategies for Distributed Local Gradient Methodscited by 4
- 2024Adaptive Batch Size Schedules for Distributed Training of Language Models with Data and Model Parallelismcited by 4
10 more
- 2022A New Perspective for Understanding Generalization Gap of Deep Neural Networks Trained with Large Batch Sizescited by 3
- 2022Transfer learning and Local interpretable model agnostic based visual approach in Monkeypox Disease Detection and Classification: A Deep Learning insightscited by 3
- 2024I/O in Machine Learning Applications on HPC Systems: A 360-degree Surveycited by 3
- 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Trainingcited by 2
- 2022Scaling the Wild: Decentralizing Hogwild!-style Shared-memory SGDcited by 1
- 2022Locally Asynchronous Stochastic Gradient Descent for Decentralised Deep Learningcited by 1
- 2023KAPLA: Pragmatic Representation and Fast Solving of Scalable NN Accelerator Dataflowcited by 0
- 2023Defect Analysis of 3D Printed Cylinder Object Using Transfer Learning Approachescited by 0
- 2023Towards Few-Annotation Learning in Computer Vision: Application to Image Classification and Object Detection taskscited by 0
- 2026SaluNet: Enabling Total Plasticity in Normalization-Free Deep Networkscited by 0
- 2026SaluNet: Enabling Total Plasticity in Normalization-Free Deep Networks1 Jun 2026
- 2024Adaptive Batch Size Schedules for Distributed Training of Language Models with Data and Model Parallelism30 Dec 2024
- 2024Communication-Efficient Adaptive Batch Size Strategies for Distributed Local Gradient Methods20 Jun 2024
- 2024I/O in Machine Learning Applications on HPC Systems: A 360-degree Survey16 Apr 2024
- 2023Towards Few-Annotation Learning in Computer Vision: Application to Image Classification and Object Detection tasks8 Nov 2023
- 2023Defect Analysis of 3D Printed Cylinder Object Using Transfer Learning Approaches12 Oct 2023
- 2023FFCV: Accelerating Training by Removing Data Bottlenecks21 Jun 2023
- 2023KAPLA: Pragmatic Representation and Fast Solving of Scalable NN Accelerator Dataflow9 Jun 2023
- 2023Knowledge Distillation in Vision Transformers: A Critical Review4 Feb 2023
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression24 Jan 2023
- 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Training12 Nov 2022
- 2022Numerical Optimizations for Weighted Low-rank Estimation on Language Model2 Nov 2022
10 more
- 2022Transfer learning and Local interpretable model agnostic based visual approach in Monkeypox Disease Detection and Classification: A Deep Learning insights1 Nov 2022
- 2022A New Perspective for Understanding Generalization Gap of Deep Neural Networks Trained with Large Batch Sizes21 Oct 2022
- 2022Locally Asynchronous Stochastic Gradient Descent for Decentralised Deep Learning24 Mar 2022
- 2022Scaling the Wild: Decentralizing Hogwild!-style Shared-memory SGD13 Mar 2022
- 2022Towards Efficient and Scalable Sharpness-Aware Minimization5 Mar 2022
- 2022Auto-scaling Vision Transformers without Training24 Feb 2022
- 2021ResIST: Layer-Wise Decomposition of ResNets for Distributed Training2 Jul 2021
- 2021Concurrent Adversarial Learning for Large-Batch Training1 Jun 2021
- 2021PAFNet: An Efficient Anchor-Free Object Detector Guidance28 Apr 2021
- 2021Combining a Convolutional Neural Network with Autoencoders to Predict the Survival Chance of COVID-19 Patients18 Apr 2021
- 2022Towards Efficient and Scalable Sharpness-Aware Minimizationcited by 129
- 2023FFCV: Accelerating Training by Removing Data Bottleneckscited by 49
- 2021ResIST: Layer-Wise Decomposition of ResNets for Distributed Trainingcited by 18
- 2022Auto-scaling Vision Transformers without Trainingcited by 15
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compressioncited by 15
- 2022Numerical Optimizations for Weighted Low-rank Estimation on Language Modelcited by 14
- 2021Concurrent Adversarial Learning for Large-Batch Trainingcited by 11
- 2021Combining a Convolutional Neural Network with Autoencoders to Predict the Survival Chance of COVID-19 Patientscited by 5
- 2021PAFNet: An Efficient Anchor-Free Object Detector Guidancecited by 5
- 2024Communication-Efficient Adaptive Batch Size Strategies for Distributed Local Gradient Methodscited by 4
- 2024Adaptive Batch Size Schedules for Distributed Training of Language Models with Data and Model Parallelismcited by 4
- 2022A New Perspective for Understanding Generalization Gap of Deep Neural Networks Trained with Large Batch Sizescited by 3
12 most cited of 21 All 21 (JSON)
- 2026SaluNet: Enabling Total Plasticity in Normalization-Free Deep Networks1 Jun 2026
- 2024Adaptive Batch Size Schedules for Distributed Training of Language Models with Data and Model Parallelism30 Dec 2024
- 2024Communication-Efficient Adaptive Batch Size Strategies for Distributed Local Gradient Methods20 Jun 2024
- 2024I/O in Machine Learning Applications on HPC Systems: A 360-degree Survey16 Apr 2024
- 2023Towards Few-Annotation Learning in Computer Vision: Application to Image Classification and Object Detection tasks8 Nov 2023
- 2023Defect Analysis of 3D Printed Cylinder Object Using Transfer Learning Approaches12 Oct 2023
- 2023FFCV: Accelerating Training by Removing Data Bottlenecks21 Jun 2023
- 2023KAPLA: Pragmatic Representation and Fast Solving of Scalable NN Accelerator Dataflow9 Jun 2023
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression24 Jan 2023
- 2022Exploiting the Partly Scratch-off Lottery Ticket for Quantization-Aware Training12 Nov 2022
- 2022Numerical Optimizations for Weighted Low-rank Estimation on Language Model2 Nov 2022
- 2022Transfer learning and Local interpretable model agnostic based visual approach in Monkeypox Disease Detection and Classification: A Deep Learning insights1 Nov 2022
12 newest of 21 All 21 (JSON)