arXiv:2410.12707cs.DCcs.AI2024-10被引 14

让分布全球的显卡协同训练大模型,自动压缩通信数据提升效率。

FusionLLM: A Decentralized LLM Training System on Geo-distributed GPUs with Adaptive Compression

  • 用操作图结构解耦模型定义与底层实现,支持灵活定制和跨框架运行。
  • 在48张全球分布的GPU上训练时,比基线快1.45至9.39倍且保证收敛。
  • 自适应压缩最慢通信链路的数据,缓解网络瓶颈,适合分布式科研团队使用。

为缓解训练大型深度神经网络(尤其是大语言模型)时的硬件资源短缺问题,我们提出FusionLLM,一个专为跨不同计算集群或单个设备的地理分布式GPU设计的去中心化训练系统。该系统面对四大挑战:远程自动微分需求、灵活模型定义与异构软件支持、异构硬件导致的资源利用率低或慢节点问题、以及缓慢的网络通信。为此,我们采用操作有向无环图(OP-DAG)表示模型,每个节点代表一个操作,边代表数据依赖关系。基于此,用户可自由定制任意深度神经网络而无需关注底层实现;支持更细粒度的任务调度,提升优化空间;并通过图执行器实现远程自动微分,无需统一底层机器学习框架版本。为提高效率,我们引入工作负载估算器,并设计了OP-Fence调度器,将带宽相近的设备聚类并划分任务以提升吞吐量。此外,提出AdaTopK压缩器,动态压缩最慢通信链路上的中间激活值与梯度。我们在三个真实测试平台、48张连接速率8 Mbps~10 Gbps网络的GPU上训练ResNet-101与GPT-2,实验表明,本系统在保证收敛的前提下,相较基线方法提升1.45至9.39倍速度。

原文摘要 · Abstract (English)

To alleviate hardware scarcity in training large deep neural networks (DNNs), particularly large language models (LLMs), we present FusionLLM, a decentralized training system designed and implemented for training DNNs using geo-distributed GPUs across different computing clusters or individual devices. Decentralized training faces significant challenges regarding system design and efficiency, including: 1) the need for remote automatic differentiation (RAD), 2) support for flexible model definitions and heterogeneous software, 3) heterogeneous hardware leading to low resource utilization or the straggler problem, and 4) slow network communication. To address these challenges, in the system design, we represent the model as a directed acyclic graph of operators (OP-DAG). Each node in the DAG represents the operator in the DNNs, while the edge represents the data dependency between operators. Based on this design, 1) users are allowed to customize any DNN without caring low-level operator implementation; 2) we enable the task scheduling with the more fine-grained sub-tasks, offering more optimization space; 3) a DAG runtime executor can implement RAD withour requiring the consistent low-level ML framework versions. To enhance system efficiency, we implement a workload estimator and design an OP-Fence scheduler to cluster devices with similar bandwidths together and partition the DAG to increase throughput. Additionally, we propose an AdaTopK compressor to adaptively compress intermediate activations and gradients at the slowest communication links. To evaluate the convergence and efficiency of our system and algorithms, we train ResNet-101 and GPT-2 on three real-world testbeds using 48 GPUs connected with 8 Mbps~10 Gbps networks. Experimental results demonstrate that our system and method can achieve 1.45 - 9.39x speedup compared to baseline methods while ensuring convergence.

分布式训练大模型通信优化去中心化

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