HetPipe: Enabling Large DNN Training on (Whimpy) Heterogeneous GPU Clusters through Integration of Pipelined Model Parallelism and Data Parallelism
Not on the map. Why?
Abstract (arXiv)
Deep Neural Network (DNN) models have continuously been growing in size in order to improve the accuracy and quality of the models. Moreover, for training of large DNN models, the use of heterogeneous GPUs is inevitable due to the short release cycle of new GPU architectures. In this paper, we investigate how to enable training of large DNN models on a heterogeneous GPU cluster that possibly includes whimpy GPUs that, as a standalone, could not be used for training. We present a DNN training system, HetPipe (Heterogeneous Pipeline), that integrates pipelined model parallelism (PMP) with data parallelism (DP). In HetPipe, a group of multiple GPUs, called a virtual worker, processes minibatches in a pipelined manner, and multiple such virtual workers employ data parallelism for higher performance. We also propose a novel parameter synchronization model, which we refer to as Wave Synchronous Parallel (WSP) to accommodate both PMP and DP for virtual workers, and provide convergence proof of WSP. Our experimental results on a given heterogeneous setting show that with HetPipe, DNN models converge up to 49% faster compared to the state-of-the-art DP technique.
Key facts. Cited by 46 papers in the map (data through 7 Oct 2026). First posted on arXiv 28 May 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 7; 2023: 6; 2024: 12; 2025: 9; 2026: 4.
46 papers.
Sort:
By area:
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 621
- 2021PanGu-α: Large-scale Autoregressive Pretrained Chinese Language Models with Auto-parallel Computationcited by 134
- 2022Decentralized Training of Foundation Models in Heterogeneous Environmentscited by 87
- 2024A Survey of Resource-efficient LLM and Multimodal Foundation Modelscited by 84
- 2024Helix: Serving Large Language Models over Heterogeneous GPUs and Network via Max-Flowcited by 54
- 2023Oobleck: Resilient Distributed Training of Large Models Using Pipeline Templatescited by 40
- 2021Towards Efficient Post-training Quantization of Pre-trained Language Modelscited by 37
- 2022Aryl: An Elastic Cluster Scheduler for Deep Learningcited by 27
- 2021PipeTransformer: Automated Elastic Pipelining for Distributed Training of Transformerscited by 23
- 2023FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUscited by 20
- 2024FlexSP: Accelerating Large Language Model Training via Flexible Sequence Parallelismcited by 19
- 2021Persia: An Open, Hybrid System Scaling Deep Learning-based Recommenders up to 100 Trillion Parameterscited by 17
34 more
- 2021HeterPS: Distributed Deep Learning With Reinforcement Learning Based Scheduling in Heterogeneous Environmentscited by 15
- 2021Pipeline Parallelism for Inference on Heterogeneous Edge Computingcited by 13
- 2024Asteroid: Resource-Efficient Hybrid Pipeline Parallelism for Collaborative DNN Training on Heterogeneous Edge Devicescited by 13
- 2022Merak: An Efficient Distributed DNN Training Framework with Automated 3D Parallelism for Giant Foundation Modelscited by 11
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Surveycited by 9
- 2024Acceleration for Deep Reinforcement Learning using Parallel and Distributed Computing: A Surveycited by 8
- 2024Spindle: Efficient Distributed Training of Multi-Task Large Models via Wavefront Schedulingcited by 7
- 2024FusionLLM: A Decentralized LLM Training System on Geo-distributed GPUs with Adaptive Compressioncited by 7
- 2021Automatic Graph Partitioning for Very Large-scale Deep Learningcited by 6
- 2022Demand Layering for Real-Time DNN Inference with Minimized Memory Usagecited by 6
- 2025Batch Query Processing and Optimization for Agentic Workflowscited by 6
- 2023Shared Memory-contention-aware Concurrent DNN Execution for Diversely Heterogeneous System-on-Chipscited by 5
- 2022Survey on Large Scale Neural Network Trainingcited by 4
- 2022Enabling All In-Edge Deep Learning: A Literature Reviewcited by 4
- 2025SkipPipe: Partial and Reordered Pipelining Framework for Training LLMs in Heterogeneous Networkscited by 4
- 2024QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devicescited by 3
- 2022A Multiagent Framework for the Asynchronous and Collaborative Extension of Multitask ML Systemscited by 2
- 2023Tessel: Boosting Distributed Execution of Large DNN Models via Flexible Schedule Searchcited by 2
- 2024Resource Allocation and Workload Scheduling for Large-Scale Distributed Deep Learning: A Surveycited by 2
- 2024Resource-Efficient Personal Large Language Models Fine-Tuning with Collaborative Edge Computingcited by 1
- 2024PreNeT: Leveraging Computational Features to Predict Deep Neural Network Training Timecited by 1
- 2025KAITIAN: A Unified Communication Framework for Enabling Efficient Collaboration Across Heterogeneous Accelerators in Embodied AI Systemscited by 1
- 2025CollaPipe: Adaptive Segment-Optimized Pipeline Parallelism for Collaborative LLM Training in Heterogeneous Edge Networkscited by 1
- 2026HetCCL: Accelerating LLM Training with Heterogeneous GPUscited by 1
- 2023Toward efficient resource utilization at edge nodes in federated learningcited by 0
- 2023Practical Performance Guarantees for Pipelined DNN Inferencecited by 0
- 2025Nonuniform-Tensor-Parallelism: Mitigating GPU failure impact for Scaled-up LLM Trainingcited by 0
- 2025MoLink: Distributed and Efficient Serving Framework for Large Modelscited by 0
- 2025Advancing Compositional LLM Reasoning with Structured Task Relations in Interactive Multimodal Communicationscited by 0
- 2025Tri-Accel: Curvature-Aware Precision-Adaptive and Memory-Elastic Optimization for Efficient GPU Usagecited by 0
- 2025TeamFormer: Shallow Parallel Transformers with Progressive Approximationcited by 0
- 2026A Readiness-Driven Runtime for Pipeline-Parallel Training under Runtime Variabilitycited by 0
- 2026HetCCL: Enabling Collective Communication For Mixed-Vendor Heterogeneous Clusterscited by 0
- 2026CogGuard: Cognitive and Operational Profiling for Proactive Warning in Edge Intelligent Servicescited by 0
- 2026CogGuard: Cognitive and Operational Profiling for Proactive Warning in Edge Intelligent Services13 Jun 2026
- 2026HetCCL: Enabling Collective Communication For Mixed-Vendor Heterogeneous Clusters29 May 2026
- 2026A Readiness-Driven Runtime for Pipeline-Parallel Training under Runtime Variability18 May 2026
- 2026HetCCL: Accelerating LLM Training with Heterogeneous GPUs30 Jan 2026
- 2025TeamFormer: Shallow Parallel Transformers with Progressive Approximation17 Oct 2025
- 2025CollaPipe: Adaptive Segment-Optimized Pipeline Parallelism for Collaborative LLM Training in Heterogeneous Edge Networks24 Sep 2025
- 2025Batch Query Processing and Optimization for Agentic Workflows2 Sep 2025
- 2025Tri-Accel: Curvature-Aware Precision-Adaptive and Memory-Elastic Optimization for Efficient GPU Usage23 Aug 2025
- 2025Advancing Compositional LLM Reasoning with Structured Task Relations in Interactive Multimodal Communications28 Jul 2025
- 2025MoLink: Distributed and Efficient Serving Framework for Large Models7 Jul 2025
- 2025KAITIAN: A Unified Communication Framework for Enabling Efficient Collaboration Across Heterogeneous Accelerators in Embodied AI Systems15 May 2025
- 2025Nonuniform-Tensor-Parallelism: Mitigating GPU failure impact for Scaled-up LLM Training8 Apr 2025
34 more
- 2025SkipPipe: Partial and Reordered Pipelining Framework for Training LLMs in Heterogeneous Networks27 Feb 2025
- 2024PreNeT: Leveraging Computational Features to Predict Deep Neural Network Training Time20 Dec 2024
- 2024FlexSP: Accelerating Large Language Model Training via Flexible Sequence Parallelism2 Dec 2024
- 2024Acceleration for Deep Reinforcement Learning using Parallel and Distributed Computing: A Survey8 Nov 2024
- 2024FusionLLM: A Decentralized LLM Training System on Geo-distributed GPUs with Adaptive Compression16 Oct 2024
- 2024Spindle: Efficient Distributed Training of Multi-Task Large Models via Wavefront Scheduling5 Sep 2024
- 2024Resource-Efficient Personal Large Language Models Fine-Tuning with Collaborative Edge Computing20 Aug 2024
- 2024Asteroid: Resource-Efficient Hybrid Pipeline Parallelism for Collaborative DNN Training on Heterogeneous Edge Devices15 Aug 2024
- 2024QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices2 Jul 2024
- 2024Resource Allocation and Workload Scheduling for Large-Scale Distributed Deep Learning: A Survey12 Jun 2024
- 2024Helix: Serving Large Language Models over Heterogeneous GPUs and Network via Max-Flow3 Jun 2024
- 2024Communication-Efficient Large-Scale Distributed Deep Learning: A Comprehensive Survey9 Apr 2024
- 2024A Survey of Resource-efficient LLM and Multimodal Foundation Models16 Jan 2024
- 2023Tessel: Boosting Distributed Execution of Large DNN Models via Flexible Schedule Search26 Nov 2023
- 2023Practical Performance Guarantees for Pipelined DNN Inference7 Nov 2023
- 2023Toward efficient resource utilization at edge nodes in federated learning19 Sep 2023
- 2023Oobleck: Resilient Distributed Training of Large Models Using Pipeline Templates15 Sep 2023
- 2023FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUs3 Sep 2023
- 2023Shared Memory-contention-aware Concurrent DNN Execution for Diversely Heterogeneous System-on-Chips10 Aug 2023
- 2022Demand Layering for Real-Time DNN Inference with Minimized Memory Usage8 Oct 2022
- 2022A Multiagent Framework for the Asynchronous and Collaborative Extension of Multitask ML Systems29 Sep 2022
- 2022Merak: An Efficient Distributed DNN Training Framework with Automated 3D Parallelism for Giant Foundation Models10 Jun 2022
- 2022Decentralized Training of Foundation Models in Heterogeneous Environments2 Jun 2022
- 2022Enabling All In-Edge Deep Learning: A Literature Review7 Apr 2022
- 2022Survey on Large Scale Neural Network Training21 Feb 2022
- 2022Aryl: An Elastic Cluster Scheduler for Deep Learning16 Feb 2022
- 2021HeterPS: Distributed Deep Learning With Reinforcement Learning Based Scheduling in Heterogeneous Environments20 Nov 2021
- 2021Persia: An Open, Hybrid System Scaling Deep Learning-based Recommenders up to 100 Trillion Parameters10 Nov 2021
- 2021Pipeline Parallelism for Inference on Heterogeneous Edge Computing28 Oct 2021
- 2021Towards Efficient Post-training Quantization of Pre-trained Language Models30 Sep 2021
- 2021PanGu-α: Large-scale Autoregressive Pretrained Chinese Language Models with Auto-parallel Computation26 Apr 2021
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LM9 Apr 2021
- 2021Automatic Graph Partitioning for Very Large-scale Deep Learning30 Mar 2021
- 2021PipeTransformer: Automated Elastic Pipelining for Distributed Training of Transformers5 Feb 2021
- 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 621
- 2021PanGu-α: Large-scale Autoregressive Pretrained Chinese Language Models with Auto-parallel Computationcited by 134
- 2022Decentralized Training of Foundation Models in Heterogeneous Environmentscited by 87
- 2024A Survey of Resource-efficient LLM and Multimodal Foundation Modelscited by 84
- 2024Helix: Serving Large Language Models over Heterogeneous GPUs and Network via Max-Flowcited by 54
- 2023Oobleck: Resilient Distributed Training of Large Models Using Pipeline Templatescited by 40
- 2021Towards Efficient Post-training Quantization of Pre-trained Language Modelscited by 37
- 2022Aryl: An Elastic Cluster Scheduler for Deep Learningcited by 27
- 2021PipeTransformer: Automated Elastic Pipelining for Distributed Training of Transformerscited by 23
- 2023FusionAI: Decentralized Training and Deploying LLMs with Massive Consumer-Level GPUscited by 20
- 2024FlexSP: Accelerating Large Language Model Training via Flexible Sequence Parallelismcited by 19
- 2021Persia: An Open, Hybrid System Scaling Deep Learning-based Recommenders up to 100 Trillion Parameterscited by 17
12 most cited of 43 All 43 (JSON)
- 2026CogGuard: Cognitive and Operational Profiling for Proactive Warning in Edge Intelligent Services13 Jun 2026
- 2026HetCCL: Enabling Collective Communication For Mixed-Vendor Heterogeneous Clusters29 May 2026
- 2026A Readiness-Driven Runtime for Pipeline-Parallel Training under Runtime Variability18 May 2026
- 2026HetCCL: Accelerating LLM Training with Heterogeneous GPUs30 Jan 2026
- 2025TeamFormer: Shallow Parallel Transformers with Progressive Approximation17 Oct 2025
- 2025CollaPipe: Adaptive Segment-Optimized Pipeline Parallelism for Collaborative LLM Training in Heterogeneous Edge Networks24 Sep 2025
- 2025Tri-Accel: Curvature-Aware Precision-Adaptive and Memory-Elastic Optimization for Efficient GPU Usage23 Aug 2025
- 2025KAITIAN: A Unified Communication Framework for Enabling Efficient Collaboration Across Heterogeneous Accelerators in Embodied AI Systems15 May 2025
- 2025Nonuniform-Tensor-Parallelism: Mitigating GPU failure impact for Scaled-up LLM Training8 Apr 2025
- 2025SkipPipe: Partial and Reordered Pipelining Framework for Training LLMs in Heterogeneous Networks27 Feb 2025
- 2024PreNeT: Leveraging Computational Features to Predict Deep Neural Network Training Time20 Dec 2024
- 2024FlexSP: Accelerating Large Language Model Training via Flexible Sequence Parallelism2 Dec 2024
12 newest of 43 All 43 (JSON)