Efficient Training on Multiple Consumer GPUs with RoundPipe
cs.DC, cs.AI, cs.LG
Submitted: 2026-04-29
Updated: 2026-10-18
Comments: Github Repo: https://github.com/ITcarrot/RoundPipe Project website: https://itcarrot.github.io/RoundPipe/
Code: https://github.com/ITcarrot/RoundPipe
Project page: https://itcarrot.github.io/RoundPipe
License: http://creativecommons.org/licenses/by/4.0/
The gist: Fine-tuning Large Language Models (LLMs) on consumer-grade GPUs is highly cost-effective, yet constrained by limited GPU memory and slow PCIe interconnects.
Terminology
Abstract
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 Layerwise Model Consistency Protocol, and an automated layer partitioning algorithm. Evaluations on an 8 times RTX 4090 server demonstrate that RoundPipe achieves 1.48--2.16 times 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 has been open-sourced as a Python library.
Sources
- L-Eval: Instituting Standardized Evaluation for Long Context Language Models
- Efficient and Robust Parallel DNN Training through Model Parallelism on Multi-GPU Platform
- Training Deep Nets with Sublinear Memory Cost
- FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioning
- FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness
- XPipe: Efficient Pipeline Model Parallelism for Multi-GPU DNN Training
- PipeDream: Fast and Efficient Pipeline Parallel DNN Training
- GPipe: Efficient Training of Giant Neural Networks using Pipeline Parallelism
- Mixtral of Experts
- MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production
- Scaling Laws for Neural Language Models
- Reducing Activation Recomputation in Large Transformer Models
- Breadth-First Pipeline Parallelism
- ZenFlow: Enabling Stall-Free Offloading for LLM Training
- LoHan: Low-Cost High-Performance Framework to Fine-Tune 100B Model on a Consumer GPU
- Training Report of TeleChat3-MoE
- The Llama 3 Herd of Models
- Mixed Precision Training
- Ray: A Distributed Framework for Emerging AI Applications
- Memory-Efficient Pipeline-Parallel DNN Training
Related papers
- iScheduler: Reinforcement Learning-Driven Continual Optimization for Large-Scale Resource Investment Problems
- Load Balancing with Evolving Workloads in Large-Scale Language Model Serving
- Dooly: Configuration-Agnostic, Redundancy-Aware Profiling for LLM Inference Simulation
- SAMM: Sharded Automated Market Maker
- InferScale: GPU-Native KV Injection for Personalized LLM Serving
- Vigil: Accountable Liveness against Selective Silence