English

DynamiQ: Accelerating Gradient Synchronization using Compressed Multi-hop All-reduce

Machine Learning 2026-02-10 v1 Distributed, Parallel, and Cluster Computing Networking and Internet Architecture

Abstract

Multi-hop all-reduce is the de facto backbone of large model training. As the training scale increases, the network often becomes a bottleneck, motivating reducing the volume of transmitted data. Accordingly, recent systems demonstrated significant acceleration of the training process using gradient quantization. However, these systems are not optimized for multi-hop aggregation, where entries are partially summed multiple times along their aggregation topology. This paper presents DynamiQ, a quantization framework that bridges the gap between quantization best practices and multi-hop aggregation. DynamiQ introduces novel techniques to better represent partial sums, co-designed with a decompress-accumulate-recompress fused kernel to facilitate fast execution. We extended PyTorch DDP to support DynamiQ over NCCL P2P, and across different LLMs, tasks, and scales, we demonstrate consistent improvement of up to 34.2% over the best among state-of-the-art methods such as Omni-Reduce, THC, and emerging standards such as MXFP4, MXFP6, and MXFP8. Further, DynamiQ is the only evaluated method that consistently reaches near-baseline accuracy (e.g., 99.9% of the BF16 baseline) and does so while significantly accelerating the training.

Keywords

Cite

@article{arxiv.2602.08923,
  title  = {DynamiQ: Accelerating Gradient Synchronization using Compressed Multi-hop All-reduce},
  author = {Wenchen Han and Shay Vargaftik and Michael Mitzenmacher and Ran Ben Basat},
  journal= {arXiv preprint arXiv:2602.08923},
  year   = {2026}
}

Comments

18 pages, 18 figures

R2 v1 2026-07-01T10:28:20.854Z