Deep learning with Elastic Averaging SGD
Not on the map. Why?
Abstract (arXiv)
We study the problem of stochastic optimization for deep learning in the parallel computing environment under communication constraints. A new algorithm is proposed in this setting where the communication and coordination of work among concurrent processes (local workers), is based on an elastic force which links the parameters they compute with a center variable stored by the parameter server (master). The algorithm enables the local workers to perform more exploration, i.e. the algorithm allows the local variables to fluctuate further from the center variable by reducing the amount of communication between local workers and the master. We empirically demonstrate that in the deep learning setting, due to the existence of many local optima, allowing more exploration can lead to the improved performance. We propose synchronous and asynchronous variants of the new algorithm. We provide the stability analysis of the asynchronous variant in the round-robin scheme and compare it with the more common parallelized method ADMM. We show that the stability of EASGD is guaranteed when a simple stability condition is satisfied, which is not the case for ADMM. We additionally propose the momentum-based version of our algorithm that can be applied in both synchronous and asynchronous settings. Asynchronous variant of the algorithm is applied to train convolutional neural networks for image classification on the CIFAR and ImageNet datasets. Experiments demonstrate that the new algorithm accelerates the training of deep architectures compared to DOWNPOUR and other common baseline approaches and furthermore is very communication efficient.
Key facts. Cited by 90 papers in the map (data through 14 Oct 2026). First posted on arXiv 20 Dec 2014.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 29; 2022: 13; 2023: 16; 2024: 14; 2025: 11; 2026: 6.
90 papers.
Sort:
By area:
- 2024Branch-Train-MiX: Mixing Expert LLMs into a Mixture-of-Experts LLMcited by 109
- 2021Software-Hardware Co-design for Fast and Scalable Training of Deep Learning Recommendation Modelscited by 78
- 2022Federated Learning via Inexact ADMMcited by 44
- 2021Anarchic Federated Learningcited by 42
- 2021A Bayesian Federated Learning Framework with Online Laplace Approximationcited by 41
- 2021Communication Efficiency in Federated Learning: Achievements and Challengescited by 40
- 2023Eliminating Domain Bias for Federated Learning in Representation Spacecited by 38
- 2022On Privacy and Personalization in Cross-Silo Federated Learningcited by 36
- 2021Communication-efficient SGD: From Local SGD to One-Shot Averagingcited by 19
- 2021ResIST: Layer-Wise Decomposition of ResNets for Distributed Trainingcited by 18
- 2021Distributed Learning Systems with First-order Methodscited by 17
- 2024An Upload-Efficient Scheme for Transferring Knowledge From a Server-Side Pre-trained Generator to Clients in Heterogeneous Federated Learningcited by 17
38 more
- 2023On the Role of Server Momentum in Federated Learningcited by 16
- 2022Towards Communication-efficient Vertical Federated Learning Training via Cache-enabled Local Updatescited by 15
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compressioncited by 15
- 2024Smart-Infinity: Fast Large Language Model Training using Near-Storage Processing on a Real Systemcited by 15
- 2026Demystifying Data-Driven Probabilistic Medium-Range Weather Forecastingcited by 15
- 2021Asynchronous Federated Learning for Sensor Data with Concept Driftcited by 14
- 2021Communication-Efficient ADMM-based Federated Learningcited by 14
- 2025Federated Continual Learning: Concepts, Challenges, and Solutionscited by 13
- 2021Semi-Synchronous Federated Learning for Energy-Efficient Training and Accelerated Convergence in Cross-Silo Settingscited by 11
- 2021Federated Few-Shot Learning with Adversarial Learningcited by 11
- 2021GIST: Distributed Training for Large-Scale Graph Convolutional Networkscited by 10
- 2021JUWELS Booster -- A Supercomputer for Large-Scale AI Researchcited by 10
- 2021Large-scale Building Height Retrieval from Single SAR Imagery based on Bounding Box Regression Networkscited by 10
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Surveycited by 10
- 2025FeDaL: Federated Dataset Learning for General Time Series Foundation Modelscited by 10
- 2021Distributed Deep Learning Using Volunteer Computing-Like Paradigmcited by 9
- 2021Efficient and Reliable Overlay Networks for Decentralized Federated Learningcited by 9
- 2023Weighted Averaged Stochastic Gradient Descent: Asymptotic Normality and Optimalitycited by 8
- 2023Learning Rate Schedules in the Presence of Distribution Shiftcited by 7
- 2023A2CiD2: Accelerating Asynchronous Communication in Decentralized Deep Learningcited by 7
- 2023Accelerating Distributed ML Training via Selective Synchronizationcited by 7
- 2023An effective theory of collective deep learningcited by 7
- 2025HtFLlib: A Comprehensive Heterogeneous Federated Learning Library and Benchmarkcited by 7
- 2021Decoupled Greedy Learning of CNNs for Synchronous and Asynchronous Distributed Learningcited by 6
- 2022Personalized Federated Learning with Communication Compressioncited by 6
- 2024ACCO: Accumulate While You Communicate for Communication-Overlapped Sharded LLM Trainingcited by 6
- 2021The Gradient Convergence Bound of Federated Multi-Agent Reinforcement Learning with Efficient Communicationcited by 5
- 2024No Need to Talk: Asynchronous Mixture of Language Modelscited by 5
- 2025OmniLearn: A Framework for Distributed Deep Learning over Heterogeneous Clusterscited by 5
- 2021Partitioning sparse deep neural networks for scalable training and inferencecited by 4
- 2021Better Training using Weight-Constrained Stochastic Dynamicscited by 4
- 2022Enabling All In-Edge Deep Learning: A Literature Reviewcited by 4
- 2023DropCompute: simple and more robust distributed synchronous training via compute variance reductioncited by 4
- 2023PipeOptim: Ensuring Effective 1F1B Schedule with Optimizer-Dependent Weight Predictioncited by 4
- 2024GRAWA: Gradient-based Weighted Averaging for Distributed Training of Deep Learning Modelscited by 4
- 2024New Insight in Cervical Cancer Diagnosis Using Convolution Neural Network Architecturecited by 4
- 2020Elastic Consistency: A General Consistency Model for Distributed Stochastic Gradient Descentcited by 3
- 2021Distributed stochastic optimization with large delayscited by 3
50 most cited of 90 All 90 (JSON)
- 2026Training for the Model You Return: Improving Optimization for Iterate-Averaged Language Models23 Jun 2026
- 2026Unifying Local Communications and Local Updates for LLM Pretraining9 Jun 2026
- 2026A Hierarchical Sampling Framework for bounding the Generalization Error of Federated Learning5 May 2026
- 2026Scalable Hyperparameter-Divergent Ensemble Training with Automatic Learning Rate Exploration for Large Models27 Apr 2026
- 2026LoDAdaC: a unified local training-based decentralized framework with adaptive gradients and compressed communication11 Apr 2026
- 2026Demystifying Data-Driven Probabilistic Medium-Range Weather Forecasting26 Jan 2026
- 2025LSAM: Asynchronous Distributed Training with Landscape-Smoothed Sharpness-Aware Minimization3 Sep 2025
- 2025Cooperative SGD with Dynamic Mixing Matrices20 Aug 2025
- 2025FeDaL: Federated Dataset Learning for General Time Series Foundation Models6 Aug 2025
- 2025Communication-Efficient Distributed Training for Collaborative Flat Optima Recovery in Deep Learning27 Jul 2025
- 2025Federated Learning for Commercial Image Sources17 Jul 2025
- 2025HtFLlib: A Comprehensive Heterogeneous Federated Learning Library and Benchmark4 Jun 2025
38 more
- 2025PADAM: Parallel averaged Adam reduces the error for stochastic optimization in scientific machine learning28 May 2025
- 2025Pseudo-Asynchronous Local SGD: Robust and Efficient Data-Parallel Training25 Apr 2025
- 2025OmniLearn: A Framework for Distributed Deep Learning over Heterogeneous Clusters21 Mar 2025
- 2025GC-Fed: Gradient Centralized Federated Learning with Partial Client Participation17 Mar 2025
- 2025Federated Continual Learning: Concepts, Challenges, and Solutions10 Feb 2025
- 2024New Insight in Cervical Cancer Diagnosis Using Convolution Neural Network Architecture23 Oct 2024
- 2024No Need to Talk: Asynchronous Mixture of Language Models4 Oct 2024
- 2024A Dynamic Weighting Strategy to Mitigate Worker Node Failure in Distributed Deep Learning14 Sep 2024
- 2024Weight Scope Alignment: A Frustratingly Easy Method for Model Merging22 Aug 2024
- 2024Training Through Failure: Effects of Data Consistency in Parallel Machine Learning Training8 Jun 2024
- 2024ACCO: Accumulate While You Communicate for Communication-Overlapped Sharded LLM Training3 Jun 2024
- 2024Adjacent Leader Decentralized Stochastic Gradient Descent18 May 2024
- 2024FCNCP: A Coupled Nonnegative CANDECOMP/PARAFAC Decomposition Based on Federated Learning18 Apr 2024
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Survey9 Apr 2024
- 2024SignSGD with Federated Voting25 Mar 2024
- 2024An Upload-Efficient Scheme for Transferring Knowledge From a Server-Side Pre-trained Generator to Clients in Heterogeneous Federated Learning23 Mar 2024
- 2024Branch-Train-MiX: Mixing Expert LLMs into a Mixture-of-Experts LLM12 Mar 2024
- 2024Smart-Infinity: Fast Large Language Model Training using Near-Storage Processing on a Real System11 Mar 2024
- 2024GRAWA: Gradient-based Weighted Averaging for Distributed Training of Deep Learning Models7 Mar 2024
- 2023Parallel Trust-Region Approaches in Neural Network Training: Beyond Traditional Methods21 Dec 2023
- 2023On the Role of Server Momentum in Federated Learning19 Dec 2023
- 2023Exploring the Robustness of Decentralized Training for Large Language Models1 Dec 2023
- 2023PipeOptim: Ensuring Effective 1F1B Schedule with Optimizer-Dependent Weight Prediction1 Dec 2023
- 2023Eliminating Domain Bias for Federated Learning in Representation Space25 Nov 2023
- 2023An effective theory of collective deep learning19 Oct 2023
- 2023DRAG: Divergence-based Adaptive Aggregation in Federated learning on Non-IID Data4 Sep 2023
- 2023Distributed Extra-gradient with Optimal Complexity and Communication Guarantees17 Aug 2023
- 2023Accelerating Distributed ML Training via Selective Synchronization16 Jul 2023
- 2023Weighted Averaged Stochastic Gradient Descent: Asymptotic Normality and Optimality13 Jul 2023
- 2023A Survey From Distributed Machine Learning to Distributed Deep Learning11 Jul 2023
- 2023DropCompute: simple and more robust distributed synchronous training via compute variance reduction18 Jun 2023
- 2023A2CiD2: Accelerating Asynchronous Communication in Decentralized Deep Learning14 Jun 2023
- 2023SparDL: Distributed Deep Learning Training with Efficient Sparse Communication3 Apr 2023
- 2023Learning Rate Schedules in the Presence of Distribution Shift27 Mar 2023
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression24 Jan 2023
- 2022Scaling up Stochastic Gradient Descent for Non-convex Optimisation6 Oct 2022
- 2022Shuffle-QUDIO: accelerate distributed VQE with trainability enhancement and measurement reduction26 Sep 2022
- 2022Personalized Federated Learning with Communication Compression12 Sep 2022
50 newest of 90 All 90 (JSON)
- 2021Software-Hardware Co-design for Fast and Scalable Training of Deep Learning Recommendation Modelscited by 78
- 2022Federated Learning via Inexact ADMMcited by 44
- 2021Anarchic Federated Learningcited by 42
- 2021A Bayesian Federated Learning Framework with Online Laplace Approximationcited by 41
- 2021Communication Efficiency in Federated Learning: Achievements and Challengescited by 40
- 2023Eliminating Domain Bias for Federated Learning in Representation Spacecited by 38
- 2022On Privacy and Personalization in Cross-Silo Federated Learningcited by 36
- 2021Communication-efficient SGD: From Local SGD to One-Shot Averagingcited by 19
- 2021ResIST: Layer-Wise Decomposition of ResNets for Distributed Trainingcited by 18
- 2021Distributed Learning Systems with First-order Methodscited by 17
- 2024An Upload-Efficient Scheme for Transferring Knowledge From a Server-Side Pre-trained Generator to Clients in Heterogeneous Federated Learningcited by 17
- 2023On the Role of Server Momentum in Federated Learningcited by 16
12 most cited of 88 All 88 (JSON)
- 2026Training for the Model You Return: Improving Optimization for Iterate-Averaged Language Models23 Jun 2026
- 2026Unifying Local Communications and Local Updates for LLM Pretraining9 Jun 2026
- 2026A Hierarchical Sampling Framework for bounding the Generalization Error of Federated Learning5 May 2026
- 2026Scalable Hyperparameter-Divergent Ensemble Training with Automatic Learning Rate Exploration for Large Models27 Apr 2026
- 2026LoDAdaC: a unified local training-based decentralized framework with adaptive gradients and compressed communication11 Apr 2026
- 2026Demystifying Data-Driven Probabilistic Medium-Range Weather Forecasting26 Jan 2026
- 2025LSAM: Asynchronous Distributed Training with Landscape-Smoothed Sharpness-Aware Minimization3 Sep 2025
- 2025Cooperative SGD with Dynamic Mixing Matrices20 Aug 2025
- 2025FeDaL: Federated Dataset Learning for General Time Series Foundation Models6 Aug 2025
- 2025Communication-Efficient Distributed Training for Collaborative Flat Optima Recovery in Deep Learning27 Jul 2025
- 2025Federated Learning for Commercial Image Sources17 Jul 2025
- 2025HtFLlib: A Comprehensive Heterogeneous Federated Learning Library and Benchmark4 Jun 2025
12 newest of 88 All 88 (JSON)