arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.09109cs.LGcs.AIcs.CVcs.DC
收藏

大规模语言模型的分布式混合并行:比较研究与系统设计指南

Distributed Hybrid Parallelism for Large Language Models: Comparative Study and System Design Guide

Huawei Canada(华为加拿大)

Hossam Amer, Rezaul Karim, Ali Pourranjbar, Weiwei Zhang, Walid Ahmed, Boxing Chen

AI总结:

本文研究大规模语言模型的分布式混合并行方法,通过比较分析和系统设计指南,探讨优化并行策略以提升训练和推理效率。

摘要:

随着大规模语言模型(LLMs)的迅速发展,各种方法已被开发出来在硬件设备上分布计算和内存以实现高效的训练和推理。尽管现有的综述提供了这些技术的描述性概述,但对这些技术的收益和权衡的系统分析以及如何利用这些见解来指导设计最优分布式系统的方法仍然有限。本文对集体操作和分布式并行策略进行了全面回顾,并辅以数学公式以加深理论理解。我们进一步考察了混合并行化设计,强调在模型部署的不同阶段中的通信与计算重叠,包括训练和推理。最近利用成本模型自动搜索最优混合并行化策略的进展也进行了讨论。此外,我们通过主流架构类别进行案例研究,揭示经验见解以指导研究者和实践者选择并行策略。最后,我们强调了当前LLM训练范式的开放挑战和限制,并概述了大规模模型开发下一代的有前途方向。

英文摘要:

With the rapid growth of large language models (LLMs), a wide range of methods have been developed to distribute computation and memory across hardware devices for efficient training and inference. While existing surveys provide descriptive overviews of these techniques, systematic analysis of their benefits and trade offs and how such insights can inform principled methodology for designing optimal distributed systems remain limited. This paper offers a comprehensive review of collective operations and distributed parallel strategies, complemented by mathematical formulations to deepen theoretical understanding. We further examine hybrid parallelization designs, emphasizing communication computation overlap across different stages of model deployment, including both training and inference. Recent advances in automated search for optimal hybrid parallelization strategies using cost models are also discussed. Moreover, we present case studies with mainstream architecture categories to reveal empirical insights to guide researchers and practitioners in parallelism strategy selection. Finally, we highlight open challenges and limitations of current LLM training paradigms and outline promising directions for the next generation of large scale model development.

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

↑