English

Learning In Chaos: Efficient Autoscaling and Self-Healing for Multi-Party Distributed Training

Distributed, Parallel, and Cluster Computing 2025-09-16 v2 Artificial Intelligence

Abstract

Node and link churn in multi-party, cross-region clusters over wide-area networks (WANs) often disrupts distributed training. However, checkpoint-based recovery and cloud-centric autoscaling react slowly and assume centralized control, which is misaligned with the self-governed setup where institutions can freely join and leave. This paper proposes Chaos, a multi-party distributed training system with self-healing and autoscaling, enabling robust and elastic training under churn. It speeds up autoscaling via multi-neighbor state replication and model sharding. We formalize the sharding and assignment as a MINLP that captures WAN heterogeneity, and reduce it to a tractable MILP by analyzing its monotonicity on a divisibility chain. By establishing an equivalence, we derive a greedy algorithm that follows optimality rules and yields the optimal solution in polynomial time. Chaos uses a cluster monitor to track resource and topology changes, and handles scaling events through peer negotiation protocols, enabling fully self-governed autoscaling among institutions. Experiments show that Chaos has substantially lower scale-out delay than Pollux, Elan, and Autoscaling, and handles scale-in, connect-link, and disconnect-link events within 20ms. It also delivers the lowest idle time, showing superior resource use and scalability as the cluster grows.

Keywords

Cite

@article{arxiv.2505.12815,
  title  = {Learning In Chaos: Efficient Autoscaling and Self-Healing for Multi-Party Distributed Training},
  author = {Wenjiao Feng and Rongxing Xiao and Zonghang Li and Hongfang Yu and Gang Sun and Long Luo and Mohsen Guizani and Qirong Ho and Steve Liu},
  journal= {arXiv preprint arXiv:2505.12815},
  year   = {2025}
}

Comments

14 pages, 16 figures

R2 v1 2026-07-01T02:21:07.637Z