arXiv:2411.12780cs.CV2024-11被引 5

用局部学习提升多卡训练速度,突破传统流水线瓶颈。

Faster Multi-GPU Training with PPLL: A Pipeline Parallelism Framework Leveraging Local Learning

  • 将模型分块部署于多GPU,通过队列实现无阻塞流水线通信。
  • 4卡下ViT训练提速162%,ResNet提速33%,接近传统流水线效率。
  • 适合追求高效训练的大模型研究者,尤其适用于局部学习场景。

当前大规模深度学习模型训练通常依赖多GPU并行。然而,传统模型并行方法存在固有的通信开销和同步延迟,难以实现无缝并行,影响整体训练效率。为此,我们提出PPLL(基于局部学习的流水线并行框架),利用局部学习算法实现多GPU间的高效并行训练。PPLL将模型划分为多个独立模块,分别分配至不同GPU;通过队列管理跨GPU数据传输,使各模块可流水式执行前向与反向传播。该设计有效减少空闲时间,避免由串行梯度更新导致的瓶颈,从而加速训练过程。我们在CIFAR-10、SVHN和STL-10数据集上,针对ResNet与视觉变换器(Vision Transformer, ViT)进行了大量实验。结果表明,PPLL显著提升了局部学习方法的训练速度,在保持模型性能的前提下,其训练速度达到或超过传统流水线并行(PP)。在4卡配置下,对ViT和ResNet的训练分别提速162%和33%,分别达到传统流水线并行的1.25倍和0.85倍。

原文摘要 · Abstract (English)

Currently, training large-scale deep learning models is typically achieved through parallel training across multiple GPUs. However, due to the inherent communication overhead and synchronization delays in traditional model parallelism methods, seamless parallel training cannot be achieved, which, to some extent, affects overall training efficiency. To address this issue, we present PPLL (Pipeline Parallelism based on Local Learning), a novel framework that leverages local learning algorithms to enable effective parallel training across multiple GPUs. PPLL divides the model into several distinct blocks, each allocated to a separate GPU. By utilizing queues to manage data transfers between GPUs, PPLL ensures seamless cross-GPU communication, allowing multiple blocks to execute forward and backward passes in a pipelined manner. This design minimizes idle times and prevents bottlenecks typically caused by sequential gradient updates, thereby accelerating the overall training process. We validate PPLL through extensive experiments using ResNet and Vision Transformer (ViT) architectures on CIFAR-10, SVHN, and STL-10 datasets. Our results demonstrate that PPLL significantly enhances the training speed of the local learning method while achieving comparable or even superior training speed to traditional pipeline parallelism (PP) without sacrificing model performance. In a 4-GPU training setup, PPLL accelerated local learning training on ViT and ResNet by 162% and 33%, respectively, achieving 1.25x and 0.85x the speed of traditional pipeline parallelism.

流水线并行多卡训练局部学习效率优化

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