CIERA: Cross-Iteration Exponent Reuse for Lossless Allgather in Sharded MoE Training

2026-09-03

AI Summary

为解决MoE模型训练中Allgather操作导致的通信开销问题，提出CIERA方法，通过跨迭代指数重用来实现无损压缩，显著提高训练速度。

Abstract

In training Mixture-of-Experts (MoE) models, sharded data parallelism partitions each expert's parameters across GPUs, requiring an Allgather operation to reconstruct the full weight matrix before each layer executes. This communication often dominates iteration time. Prior work often reduces this overhead using lossy compression methods that sacrifice numerical fidelity, while existing lossless methods do not exploit cross-iteration exponent stability. In this paper, we propose Cross-Iteration Exponent Reuse Allgather (CIERA), a lossless, system-aware communication method for sharded MoE training. We observe that after a brief warmup phase, the exponent values of most weights remain unchanged across iterations. Based on this, we cache exponents locally and transmit only the sign and mantissa when exponents are unchanged. The receiver reconstructs the original weights exactly by combining the cached exponents with the received data. Since parameter matrices vary in shape across layers, compression is applied only when it yields net time savings. Moreover, the compression operation is overlapped with both Allgather communication and computation. Our real experiments and large-scale trace-driven simulator show that on OLMoE-1B-7B at 16 GPUs, CIERA achieves a 3.70x speedup over the lossless baseline and 3.68x over the lossy baseline, projected to reach 4.28x and 4.42x respectively at 128 GPUs, while preserving bitwise-exact parameter reconstruction in all evaluated runs.

Problem

• Mixture-of-Experts
• Sharded Data Parallelism
• Allgather Operation
• Lossless Compression
• Cross-Iteration Exponent Stability

Innovation

• Cross-Iteration Exponent Reuse
• Lossless Communication
• Sharded MoE Training