English

ElasWave: An Elastic-Native System for Scalable Hybrid-Parallel Training

Distributed, Parallel, and Cluster Computing 2025-10-09 v3

Abstract

Large-scale LLM pretraining now runs across 10510^5--10610^6 accelerators, making failures routine and elasticity mandatory. We posit that an elastic-native training system must jointly deliver (i) parameter consistency, (ii) low mean time to recovery (MTTR), (iii) high post-change throughput, and (iv) computation consistency. No prior system achieves all four simultaneously. To achieve these goals, we present ElasWave, which delivers per-step fault tolerance via multi-dimensional scheduling across graph, dataflow, DVFS, and RNG. ElasWave reshapes and reshards micro-batches while preserving the global batch size and gradient scale. It performs online pipeline resharding with asynchronous parameter migration and interleaves ZeRO partitions, reducing parameter recovery processes to disjoint rank-to-rank transfers. It further leverages DVFS to absorb pipeline bubbles and reshards RNG to keep computation consistency. Together, a dynamic communicator enables in-place communication group edits, while per-step in-memory snapshots support online verification and redistribution. We evaluate ElasWave on 96 NPUs and benchmark it against state-of-the-art baselines: throughput improves by 1.35×1.35\times over ReCycle and 1.60×1.60\times over TorchFT; communicator recovery completes within one second (up to 82×/3.6×82\times/3.6\times faster than full/partial rebuilds); migration MTTR drops by as much as 51%51\%; and convergence deviation is reduced by approximately 78%78\%.

Keywords

Cite

@article{arxiv.2510.00606,
  title  = {ElasWave: An Elastic-Native System for Scalable Hybrid-Parallel Training},
  author = {Xueze Kang and Guangyu Xiang and Yuxin Wang and Hao Zhang and Yuchu Fang and Yuhang Zhou and Zhenheng Tang and Youhui Lv and Eliran Maman and Mark Wasserman and Alon Zameret and Zhipeng Bian and Shushu Chen and Zhiyou Yu and Jin Wang and Xiaoyu Wu and Yang Zheng and Chen Tian and Xiaowen Chu},
  journal= {arXiv preprint arXiv:2510.00606},
  year   = {2025}
}
R2 v1 2026-07-01T06:09:50.909Z