arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.04594cs.DCcs.AIcs.LG
收藏

提升大规模AI训练效率:面向实时异常检测与通信优化的C4解决方案

Enhancing Large-Scale AI Training Efficiency: The C4 Solution for Real-Time Anomaly Detection and Communication Optimization

Alibaba Group(阿里巴巴集团);Hong Kong University of Science and Technology(香港科技大学)
Jianbo Dong, Bin Luo, Jun Zhang, Pengcheng Zhang, Fei Feng, Yikai Zhu, Ang Liu, Zian Chen, Yi Shi, Hairong Jiao, Gang Lu, Yu Guan, Ennan Zhai, Wencong Xiao, Han… 展开作者

Jianbo Dong, Bin Luo, Jun Zhang, Pengcheng Zhang, Fei Feng, Yikai Zhu, Ang Liu, Zian Chen, Yi Shi, Hairong Jiao, Gang Lu, Yu Guan, Ennan Zhai, Wencong Xiao, Hanyu Zhao, Man Yuan, Siran Yang, Xiang Li, Jiamang Wang, Rui Men, Jianwei Zhang, Chang Zhou, Dennis Cai, Yuan Xie, Binzhang Fu

AI总结:

针对大规模分布式训练中硬件故障检测延迟和通信冲突导致的效率低下问题,提出通信驱动的C4方案,利用负载同质性和可预测通信模型实现快速异常定位与流量规划,在实际系统中将效率从30%提升至45%。

摘要:

大规模语言模型(LLM)的出现使得采用分布式训练技术成为必要,这涉及部署数千块GPU来训练单个模型。然而,由于高端GPU产品中硬件错误发生概率的增加以及网络流量冲突风险的加剧,大规模分布式训练系统的效率往往不尽如人意。此外,任何局部硬件故障都可能中断训练任务,而无法迅速识别故障组件会导致GPU资源的严重浪费。同时,由流量冲突引起的长时间通信会显著增加GPU的等待时间。为应对这些挑战,我们提出了一种通信驱动的解决方案,即C4。C4的关键洞察有两点。第一,分布式训练中的负载具有同质性特征,并通过周期性同步被划分为迭代,因此硬件异常会在集合通信中产生特定的综合征。利用这一特性,C4能够快速识别故障组件,迅速隔离异常,并重启任务,从而避免因异常检测延迟而造成的资源浪费。第二,集合通信的可预测通信模型涉及有限数量的长寿命流,这使得C4能够高效地执行流量规划,大幅减少这些流之间的带宽竞争。C4已在超大规模云提供商的实际生产系统中广泛部署,使系统效率从30%提升至45%。这一提升归功于错误引发开销的30%减少以及通信成本的15%降低。

英文摘要:

The emergence of Large Language Models (LLMs) has necessitated the adoption of distributed training techniques, involving the deployment of thousands of GPUs to train a single model. Unfortunately, the efficiency of large-scale distributed training systems is often suboptimal due to the increased likelihood of hardware errors in high-end GPU products and the heightened risk of network traffic collisions. Moreover, any local hardware failure can disrupt training tasks, and the inability to swiftly identify faulty components leads to a significant waste of GPU resources. And, prolonged communication due to traffic collisions can substantially increase GPU waiting times. To address these challenges, we propose a communication-driven solution, namely the C4. The key insights of C4 are twofold. First, the load in distributed training exhibits homogeneous characteristics and is divided into iterations through periodic synchronization, therefore hardware anomalies would incur certain syndrome in collective communication. By leveraging this feature, C4 can rapidly identify the faulty components, swiftly isolate the anomaly, and restart the task, thereby avoiding resource wastage caused by delays in anomaly detection. Second, the predictable communication model of collective communication, involving a limited number of long-lived flows, allows C4 to efficiently execute traffic planning, substantially reducing bandwidth competition among these flows. The C4 has been extensively deployed across real-world production systems in a hyperscale cloud provider, yielding a significant improvement in system efficiency, from 30% to 45%. This enhancement is attributed to a 30% reduction in error-induced overhead and a 15% reduction in communication costs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑