mlsanityData to 14 Oct 2026

Efficient Training on Multiple Consumer GPUs with RoundPipe

arXiv 2604.27085 ↗First posted 29 Apr 2026cs.DC cs.AI cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Fine-tuning Large Language Models (LLMs) on consumer-grade GPUs is highly cost-effective, yet constrained by limited GPU memory and slow PCIe interconnects. Pipeline parallelism combined with CPU offloading mitigates these hardware bottlenecks by reducing communication overhead. However, existing PP schedules suffer from an inherent limitation termed the weight binding issue. Binding uneven model stages (e.g., the LM head is large) to GPUs limits the pipeline's throughput to that of the GPU with the heaviest load, leading to severe pipeline bubbles. In this paper, we propose RoundPipe, a novel pipeline schedule that breaks the weight binding constraint on consumer GPU servers. RoundPipe treats GPUs as a pool of stateless execution workers and dynamically dispatches computation stages across devices in a round-robin manner, achieving a near-zero-bubble pipeline. To ensure training correctness and system efficiency, RoundPipe integrates a priority-aware transfer scheduling engine, a fine-grained distributed event-based synchronization protocol, and an automated layer partitioning algorithm. Evaluations on an 8× RTX 4090 server demonstrate that RoundPipe achieves 1.48--2.16× speedups over state-of-the-art baselines when fine-tuning 1.7B to 32B models. Remarkably, RoundPipe enables LoRA fine-tuning of the Qwen3-235B model with 31K sequence length on a single server. RoundPipe is publicly available as an open-source Python library with comprehensive documentation.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 33 of its 54 references that are in the map. First posted on arXiv 29 Apr 2026.

Builds on

33 of 54 references.

References · 33

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  3. 2022Training language models to follow instructions with human feedbackcited by 15,971
  4. 2024The Llama 3 Herd of Modelscited by 14,715
  5. 2020Scaling Laws for Neural Language Modelscited by 6,586
  6. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  7. 2025Wan: Open and Advanced Large-Scale Video Generative Modelscited by 2,708
  8. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  9. 2023Code Llama: Open Foundation Models for Codecited by 2,139
  10. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
  11. 2025gpt-oss-120b & gpt-oss-20b Model Cardcited by 1,386
  12. 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
21 more
  1. 2017Mixed Precision Trainingcited by 1,156
  2. 2016Training Deep Nets with Sublinear Memory Costcited by 848
  3. 2021Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LMcited by 632
  4. 2017Ray: A Distributed Framework for Emerging AI Applicationscited by 527
  5. 2021ZeRO-Offload: Democratizing Billion-Scale Model Trainingcited by 341
  6. 2022Reducing Activation Recomputation in Large Transformer Modelscited by 302
  7. 2021ZeRO-Infinity: Breaking the GPU Memory Wall for Extreme Scale Deep Learningcited by 256
  8. 2023L-Eval: Instituting Standardized Evaluation for Long Context Language Modelscited by 233
  9. 2020Memory-Efficient Pipeline-Parallel DNN Trainingcited by 134
  10. 2018PipeDream: Fast and Efficient Pipeline Parallel DNN Trainingcited by 117
  11. 2020DAPPLE: A Pipelined Data Parallel Approach for Training Large Modelscited by 101
  12. 2021Chimera: Efficiently Training Large-Scale Neural Networks with Bidirectional Pipelinescited by 68
  13. 2018SuperNeurons: Dynamic GPU Memory Management for Training Deep Neural Networkscited by 55
  14. 2019PipeMare: Asynchronous Pipeline Parallel DNN Trainingcited by 55
  15. 2025MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Productioncited by 31
  16. 2018Efficient and Robust Parallel DNN Training through Model Parallelism on Multi-GPU Platformcited by 28
  17. 2021PatrickStar: Parallel Training of Pre-trained Models via Chunk-based Memory Managementcited by 15
  18. 2023G10: Enabling An Efficient Unified GPU Memory and Storage Architecture with Smart Tensor Migrationscited by 10
  19. 2024LoHan: Low-Cost High-Performance Framework to Fine-Tune 100B Model on a Consumer GPUcited by 10
  20. 2024MEMO: Fine-grained Tensor Management For Ultra-long Context LLM Trainingcited by 5
  21. 2025ZenFlow: Enabling Stall-Free Offloading for LLM Trainingcited by 5

Built on by

Not cited yet.