arXiv:2604.27085cs.DCcs.AI2026-04

突破消费级显卡训练瓶颈,实现高效多卡协同微调大模型。

Efficient Training on Multiple Consumer GPUs with RoundPipe

论文配图:Efficient Training on Multiple Consumer GPUs with RoundPipe
图 1 · 摘自论文原文
  • 将显卡视为无状态计算池,动态轮转分配计算任务。
  • 在8×RTX 4090上训练1.7B至32B模型提速1.48–2.16倍。
  • 可单机微调Qwen3-235B模型,支持31K序列长度。

在消费级显卡上微调大语言模型(LLMs)成本低,但受限于显存不足和慢速PCIe通信。流水线并行结合CPU卸载可缓解硬件瓶颈,但现有流水线调度存在固有缺陷——权重绑定问题:将不均等的模型阶段(如语言模型头较大)绑定到特定显卡,导致流水线吞吐受限于负载最重的显卡,产生严重流水线空洞。本文提出RoundPipe,一种新型流水线调度方案,打破消费级服务器上的权重绑定限制。RoundPipe将显卡视为无状态执行工作节点,以轮询方式动态调度计算阶段,实现近零空洞流水线。为保障训练正确性与系统效率,其集成优先级感知传输调度引擎、细粒度分布式事件同步协议及自动层划分算法。在8×RTX 4090服务器上的评估显示,微调1.7B至32B模型时,性能比当前最优基线提升1.48–2.16倍。显著地,RoundPipe实现了单服务器对Qwen3-235B模型(31K序列长度)的LoRA微调。代码已开源,附完整文档。

原文摘要 · Abstract (English)

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$\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 is publicly available as an open-source Python library with comprehensive documentation.

大模型微调流水线并行消费级显卡LoRA

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。