In plain words: Large AI training sends messages only between small groups of chips, so this design drops the central switch layer and wires those groups straight together. It trains just as fast as a standard datacenter network while cutting network cost by up to 77%.
Abstract · Rail-only: A Low-Cost High-Performance Network for Training LLMs with Trillion Parameters
This paper presents a low-cost network architecture for training large language models (LLMs) at hyperscale. We study the optimal parallelization strategy of LLMs and propose a novel datacenter network design tailored to LLM's unique communication pattern. We show that LLM training generates sparse communication patterns in the network and, therefore, does not require any-to-any full-bisection network to complete efficiently. As a result, our design eliminates the spine layer in traditional GPU clusters. We name this design a Rail-only network and demonstrate that it achieves the same training performance while reducing the network cost by 38% to 77% and network power consumption by 37% to 75% compared to a conventional GPU datacenter. Our architecture also supports Mixture-of-Expert (MoE) models with all-to-all communication through forwarding, with only 8.2% to 11.2% completion time overhead for all-to-all traffic. We study the failure robustness of Rail-only networks and provide insights into the performance impact of different network and training parameters.
Weiyang Wang, Manya Ghobadi, Kayvon Shakeri, Ying Zhang, Naader Hasani
arXiv:2307.12169 · cs.NI, cs.AI, cs.LG · submitted Jul 22, 2023 · updated Sep 15, 2024
abstract · pdf · html
Previous paper by the same team from Meta and MIT but with Billions instead of Trillions of parameters [2].
[1] A RoCE network for distributed AI training at scale:
https://news.ycombinator.com/item?id=41162664
[2] Optimized Network Architectures for Training Large Language Models With Billions of Parameters [PDF]:
https://people.csail.mit.edu/ghobadi/papers/rail_llm_hotnets...