低带宽下训练大模型,用稀疏通信与压缩提升效率
Heterogeneous Low-Bandwidth Pre-Training of LLMs
- 采用稀疏伪梯度交换与间歇同步降低通信量
- 激活值压缩在178M-1B参数模型上保持性能损失可控
- 异构节点协作训练,适合资源受限设备参与
大规模语言模型预训练日益依赖分布式计算,但带宽限制使其难以突破高性能数据中心的范围——尤其当模型并行引发频繁的大规模设备间通信时。本文研究将低通信数据并行方法 SparseLoCo(基于不频繁同步和稀疏伪梯度交换)与低带宽流水线模型并行结合,通过激活值及梯度压缩实现。提出一种异构分布式训练框架:部分参与者在高带宽互联上托管完整副本,而资源受限参与者则通过子空间投影的流水线并行联合构建副本。为使新近提出的子空间流水线压缩与 SparseLoCo 兼容,我们探索多种适配策略。在标准预训练语料库上的大规模语言建模实验(178M–1B参数)表明,激活压缩可与 SparseLoCo 有效结合,仅带来适度性能损失;而选择性(异构)压缩相比全副本压缩,在极端压缩比下持续改善损失-通信权衡。结果表明,该方案为将低带宽模型并行与异构参与方引入大模型预训练提供了可行路径。
原文摘要 · Abstract (English)
Pre-training large language models (LLMs) increasingly requires distributed compute, yet bandwidth constraints make it difficult to scale beyond well-provisioned datacenters-especially when model parallelism forces frequent, large inter-device communications. We study whether SparseLoCo, a low-communication data parallel method based on infrequent synchronization and sparse pseudo-gradient exchange, can be combined with low-bandwidth pipeline model parallelism via activation and activation-gradient compression. We introduce a heterogeneous distributed training framework where some participants host full replicas on high-bandwidth interconnects, while resource-limited participants are grouped to jointly instantiate a replica using pipeline parallelism with subspace-projected inter-stage communication. To make the recently introduced subspace pipeline compression compatible with SparseLoCo, we study a number of adaptations. Across large-scale language modeling experiments (178M-1B parameters) on standard pretraining corpora, we find that activation compression composes with SparseLoCo at modest cost, while selective (heterogeneous) compression consistently improves the loss-communication tradeoff relative to compressing all replicas-especially at aggressive compression ratios. These results suggest a practical path to incorporating low-bandwidth model parallelism and heterogeneous participants into LLM pre-training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。