PipeMare: Asynchronous Pipeline Parallel DNN Training
Not on the map. Why?
Abstract (arXiv)
Pipeline parallelism (PP) when training neural networks enables larger models to be partitioned spatially, leading to both lower network communication and overall higher hardware utilization. Unfortunately, to preserve the statistical efficiency of sequential training, existing PP techniques sacrifice hardware efficiency by decreasing pipeline utilization or incurring extra memory costs. In this paper, we investigate to what extent these sacrifices are necessary. We devise PipeMare, a simple yet robust training method that tolerates asynchronous updates during PP execution without sacrificing utilization or memory, which allows efficient use of fine-grained pipeline parallelism. Concretely, when tested on ResNet and Transformer networks, asynchrony enables PipeMare to use up to 2.7× less memory or get 4.3× higher pipeline utilization, with similar model quality, when compared to state-of-the-art synchronous PP training techniques.
Key facts. Cited by 54 papers in the map (data through 7 Oct 2026). First posted on arXiv 9 Oct 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 1; 2021: 7; 2022: 13; 2023: 15; 2024: 8; 2025: 4; 2026: 6.
54 papers.
Sort:
By area:
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 621
- 2022Pathways: Asynchronous Distributed Dataflow for MLcited by 81
- 2021Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelinescited by 68
- 2023SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficientcited by 58
- 2024FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusioncited by 53
- 2022Galvatron: Efficient Transformer Training over Multiple GPUs Using Automatic Parallelismcited by 51
- 2022Bamboo: Making Preemptible Instances Resilient for Affordable Training of Large DNNscited by 49
- 2022PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communicationcited by 45
- 2020Towards Crowdsourced Training of Large Neural Networks using Decentralized Mixture-of-Expertscited by 43
- 2023Distributed Inference and Fine-tuning of Large Language Models Over The Internetcited by 42
- 2023Zero Bubble Pipeline Parallelismcited by 39
- 2022Fine-tuning Language Models over Slow Networks using Activation Compression with Guaranteescited by 29
38 more
- 2023Understanding the Potential of FPGA-Based Spatial Acceleration for Large Language Model Inferencecited by 29
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Surveycited by 25
- 2022Egeria: Efficient DNN Training with Knowledge-Guided Layer Freezingcited by 22
- 2021AxoNN: An asynchronous, message-driven parallel framework for extreme-scale deep learningcited by 15
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compressioncited by 14
- 2024Smart-Infinity: Fast Large Language Model Training using Near-Storage Processing on a Real Systemcited by 14
- 2021Pipeline Parallelism for Inference on Heterogeneous Edge Computingcited by 13
- 2023Improving Automatic Parallel Training via Balanced Memory Workload Optimizationcited by 13
- 2021Automap: Towards Ergonomic Automated Parallelism for ML Modelscited by 12
- 2023mLoRA: Fine-Tuning LoRA Adapters via Highly-Efficient Pipeline Parallelism in Multiple GPUscited by 12
- 2022Merak: An Efficient Distributed DNN Training Framework with Automated 3D Parallelism for Giant Foundation Modelscited by 11
- 2024Pipeline Parallelism with Controllable Memorycited by 11
- 2022HammingMesh: A Network Topology for Large-Scale Deep Learningcited by 8
- 2022RevBiFPN: The Fully Reversible Bidirectional Feature Pyramid Networkcited by 6
- 2024PETRA: Parallel End-to-end Training with Reversible Architecturescited by 6
- 2024ACCO: Accumulate While You Communicate for Communication-Overlapped Sharded LLM Trainingcited by 6
- 2025Nesterov Method for Asynchronous Pipeline Parallel Optimizationcited by 6
- 2023Exploiting Sparsity in Pruned Neural Networks to Optimize Large Model Trainingcited by 5
- 2021Parareal Neural Networks Emulating a Parallel-in-time Algorithmcited by 4
- 2022Survey on Large Scale Neural Network Trainingcited by 4
- 2023Systems for Parallel and Distributed Large-Model Deep Learning Trainingcited by 4
- 2023PipeOptim: Ensuring Effective 1F1B Schedule with Optimizer-Dependent Weight Predictioncited by 4
- 2021Hydra: A System for Large Multi-Model Deep Learningcited by 3
- 2022Layer-Wise Partitioning and Merging for Efficient and Scalable Deep Learningcited by 2
- 2022Automatic Discovery of Composite SPMD Partitioning Strategies in PartIRcited by 2
- 2023Baechi: Fast Device Placement of Machine Learning Graphscited by 2
- 2022PiPar: Pipeline Parallelism for Collaborative Machine Learningcited by 2
- 2023Saturn: An Optimized Data System for Large Model Deep Learning Workloadscited by 2
- 2024Ravnest: Decentralized Asynchronous Training on Heterogeneous Devicescited by 2
- 2026AsyncMesh: Fully Asynchronous Optimization for Data and Pipeline Parallelismcited by 2
- 2026Mitigating Staleness in Asynchronous Pipeline Parallelism via Basis Rotationcited by 2
- 2023Exploring the Robustness of Decentralized Training for Large Language Modelscited by 1
- 2025Coded Deep Learning: Framework and Algorithmcited by 1
- 2023Practical Performance Guarantees for Pipelined DNN Inferencecited by 0
- 2024Cyclic Data Parallelism for Efficient Parallelism of Deep Neural Networkscited by 0
- 20242BP: 2-Stage Backpropagationcited by 0
- 2025Advancing Compositional LLM Reasoning with Structured Task Relations in Interactive Multimodal Communicationscited by 0
- 2025Optimizing Resource Allocation for Geographically-Distributed Inference by Large Language Modelscited by 0
50 most cited of 54 All 54 (JSON)
- 2026One-Step Gradient Delay is Not a Barrier for Large-Scale Asynchronous Pipeline Parallel LLM Pretraining29 Jun 2026
- 2026Breaking the Bubble: Asynchronous Pipeline Parallel Training with Bounded Weight Inconsistency5 Jun 2026
- 2026A Readiness-Driven Runtime for Pipeline-Parallel Training under Runtime Variability18 May 2026
- 2026Efficient Training on Multiple Consumer GPUs with RoundPipe29 Apr 2026
- 2026Mitigating Staleness in Asynchronous Pipeline Parallelism via Basis Rotation3 Feb 2026
- 2026AsyncMesh: Fully Asynchronous Optimization for Data and Pipeline Parallelism30 Jan 2026
- 2025Optimizing Resource Allocation for Geographically-Distributed Inference by Large Language Models26 Dec 2025
- 2025Advancing Compositional LLM Reasoning with Structured Task Relations in Interactive Multimodal Communications28 Jul 2025
- 2025Nesterov Method for Asynchronous Pipeline Parallel Optimization2 May 2025
- 2025Coded Deep Learning: Framework and Algorithm16 Jan 2025
- 2024FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusion11 Jun 2024
- 2024PETRA: Parallel End-to-end Training with Reversible Architectures4 Jun 2024
38 more
- 2024ACCO: Accumulate While You Communicate for Communication-Overlapped Sharded LLM Training3 Jun 2024
- 20242BP: 2-Stage Backpropagation28 May 2024
- 2024Pipeline Parallelism with Controllable Memory24 May 2024
- 2024Cyclic Data Parallelism for Efficient Parallelism of Deep Neural Networks13 Mar 2024
- 2024Smart-Infinity: Fast Large Language Model Training using Near-Storage Processing on a Real System11 Mar 2024
- 2024Ravnest: Decentralized Asynchronous Training on Heterogeneous Devices3 Jan 2024
- 2023Understanding the Potential of FPGA-Based Spatial Acceleration for Large Language Model Inference23 Dec 2023
- 2023Distributed Inference and Fine-tuning of Large Language Models Over The Internet13 Dec 2023
- 2023mLoRA: Fine-Tuning LoRA Adapters via Highly-Efficient Pipeline Parallelism in Multiple GPUs5 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
- 2023The Efficiency Spectrum of Large Language Models: An Algorithmic Survey1 Dec 2023
- 2023Zero Bubble Pipeline Parallelism30 Nov 2023
- 2023Practical Performance Guarantees for Pipelined DNN Inference7 Nov 2023
- 2023Saturn: An Optimized Data System for Large Model Deep Learning Workloads3 Sep 2023
- 2023Improving Automatic Parallel Training via Balanced Memory Workload Optimization5 Jul 2023
- 2023Exploiting Sparsity in Pruned Neural Networks to Optimize Large Model Training10 Feb 2023
- 2023SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficient27 Jan 2023
- 2023Optimus-CC: Efficient Large NLP Model Training with 3D Parallelism Aware Communication Compression24 Jan 2023
- 2023Baechi: Fast Device Placement of Machine Learning Graphs20 Jan 2023
- 2023Systems for Parallel and Distributed Large-Model Deep Learning Training6 Jan 2023
- 2022PiPar: Pipeline Parallelism for Collaborative Machine Learning1 Dec 2022
- 2022Galvatron: Efficient Transformer Training over Multiple GPUs Using Automatic Parallelism25 Nov 2022
- 2022Automatic Discovery of Composite SPMD Partitioning Strategies in PartIR7 Oct 2022
- 2022HammingMesh: A Network Topology for Large-Scale Deep Learning3 Sep 2022
- 2022Layer-Wise Partitioning and Merging for Efficient and Scalable Deep Learning22 Jul 2022
- 2022RevBiFPN: The Fully Reversible Bidirectional Feature Pyramid Network28 Jun 2022
- 2022Merak: An Efficient Distributed DNN Training Framework with Automated 3D Parallelism for Giant Foundation Models10 Jun 2022
- 2022Fine-tuning Language Models over Slow Networks using Activation Compression with Guarantees2 Jun 2022
- 2022Bamboo: Making Preemptible Instances Resilient for Affordable Training of Large DNNs26 Apr 2022
- 2022Pathways: Asynchronous Distributed Dataflow for ML23 Mar 2022
- 2022PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communication20 Mar 2022
- 2022Survey on Large Scale Neural Network Training21 Feb 2022
- 2022Egeria: Efficient DNN Training with Knowledge-Guided Layer Freezing17 Jan 2022
- 2021Automap: Towards Ergonomic Automated Parallelism for ML Models6 Dec 2021
- 2021Pipeline Parallelism for Inference on Heterogeneous Edge Computing28 Oct 2021
- 2021AxoNN: An asynchronous, message-driven parallel framework for extreme-scale deep learning25 Oct 2021
- 2021Hydra: A System for Large Multi-Model Deep Learning16 Oct 2021
50 newest of 54 All 54 (JSON)
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 621
- 2022Pathways: Asynchronous Distributed Dataflow for MLcited by 81
- 2021Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelinescited by 68
- 2023SWARM Parallelism: Training Large Models Can Be Surprisingly Communication-Efficientcited by 58
- 2024FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusioncited by 53
- 2022Galvatron: Efficient Transformer Training over Multiple GPUs Using Automatic Parallelismcited by 51
- 2022Bamboo: Making Preemptible Instances Resilient for Affordable Training of Large DNNscited by 49
- 2022PipeGCN: Efficient Full-Graph Training of Graph Convolutional Networks with Pipelined Feature Communicationcited by 45
- 2023Distributed Inference and Fine-tuning of Large Language Models Over The Internetcited by 42
- 2023Zero Bubble Pipeline Parallelismcited by 39
- 2022Fine-tuning Language Models over Slow Networks using Activation Compression with Guaranteescited by 29
- 2023Understanding the Potential of FPGA-Based Spatial Acceleration for Large Language Model Inferencecited by 29
12 most cited of 52 All 52 (JSON)
- 2026One-Step Gradient Delay is Not a Barrier for Large-Scale Asynchronous Pipeline Parallel LLM Pretraining29 Jun 2026
- 2026Breaking the Bubble: Asynchronous Pipeline Parallel Training with Bounded Weight Inconsistency5 Jun 2026
- 2026A Readiness-Driven Runtime for Pipeline-Parallel Training under Runtime Variability18 May 2026
- 2026Efficient Training on Multiple Consumer GPUs with RoundPipe29 Apr 2026
- 2026Mitigating Staleness in Asynchronous Pipeline Parallelism via Basis Rotation3 Feb 2026
- 2026AsyncMesh: Fully Asynchronous Optimization for Data and Pipeline Parallelism30 Jan 2026
- 2025Optimizing Resource Allocation for Geographically-Distributed Inference by Large Language Models26 Dec 2025
- 2025Nesterov Method for Asynchronous Pipeline Parallel Optimization2 May 2025
- 2025Coded Deep Learning: Framework and Algorithm16 Jan 2025
- 2024FLUX: Fast Software-based Communication Overlap On GPUs Through Kernel Fusion11 Jun 2024
- 2024PETRA: Parallel End-to-end Training with Reversible Architectures4 Jun 2024
- 2024ACCO: Accumulate While You Communicate for Communication-Overlapped Sharded LLM Training3 Jun 2024
12 newest of 52 All 52 (JSON)