arXiv:2604.27844cs.DCcs.CL2026-04被引 1

用高效压缩技术降低大模型训练通信开销,不损失精度

ZipCCL: Efficient Lossless Data Compression of Communication Collectives for Accelerating LLM Training

论文配图:ZipCCL: Efficient Lossless Data Compression of Communication Collectives for Accelerating LLM Training
图 1 · 摘自论文原文
  • 基于梯度近似正态分布,设计快速无损编码方法
  • 在64卡集群上通信时间减少1.35倍,训练提速1.18倍
  • 适合大规模分布式训练场景,尤其对通信敏感的模型

大语言模型分布式训练中,通信已成为主要瓶颈。尽管已有多种方法减少通信开销,但无损压缩的潜力尚未被充分挖掘,因其压缩解压开销常超过通信节省。我们观察到训练过程中的激活、梯度和参数等通信数据往往服从近似高斯分布,这为数据压缩提供了关键特性。为此,我们提出ZipCCL,一个面向大模型训练通信集合操作的无损压缩库。ZipCCL采用三项创新技术:(1) 理论基础扎实的指数编码,利用LLM张量的高斯分布实现快速压缩,无需昂贵的在线统计;(2) 针对GPU优化的压缩与解压内核,通过精心设计的内存访问模式和通信感知的数据布局进行流水线调度;(3) 自适应通信策略,根据工作负载特征和系统状态动态切换集体操作。在64卡集群上使用混合专家和密集Transformer模型评估,ZipCCL将通信时间减少最多1.35倍,实现端到端训练速度提升最高1.18倍,且不影响模型质量。

原文摘要 · Abstract (English)

Communication has emerged as a critical bottleneck in the distributed training of large language models (LLMs). While numerous approaches have been proposed to reduce communication overhead, the potential of lossless compression has remained largely underexplored since compression and decompression typically consume larger overheads than the benefits of reduced communication traffic. We observe that the communication data, including activations, gradients and parameters, during training often follows a near-Gaussian distribution, which is a key feature for data compression. Thus, we introduce ZipCCL, a lossless compressed communication library of collectives for LLM training. ZipCCL is equipped with our novel techniques: (1) theoretically grounded exponent coding that exploits the Gaussian distribution of LLM tensors to accelerate compression without expensive online statistics, (2) GPU-optimized compression and decompression kernels that carefully design memory access patterns and pipeline using communication-aware data layout, and (3) adaptive communication strategies that dynamically switch collective operations based on workload patterns and system characteristics. Evaluated on a 64-GPU cluster using both mixture-of-experts and dense transformer models, ZipCCL reduces communication time by up to 1.35$\times$ and achieves end-to-end training speedups of up to 1.18$\times$ without any impact on model quality.

大模型训练通信优化无损压缩

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