Federated Aggregation Reduces Traffic in Sharded Training 联邦聚合降低分片训练通信量
arXiv:2609.20359 · 2026-09-17T13:21:52Z
FL+FSDP and FL+HSDP divide large sharded-training jobs into loosely coupled federation groups, using FedAvg-style aggregation to reduce inter-group traffic while bounding global batch size. The authors evaluate Llama 3.1 8B pretraining on 512 A100 GPUs under identical hyperparameters and report better scaling than conventional sharded data parallelism. The result directly addresses heterogeneous multi-tier interconnects, but broader convergence behavior and production fault tolerance still need validation. FL+FSDP与FL+HSDP把大规模分片训练拆成松耦合的联邦组,以类似FedAvg的聚合减少组间流量,同时限制全局批量大小。作者在512张A100 GPU上以相同超参数进行Llama 3.1 8B预训练,并报告了优于传统分片数据并行的扩展表现。该方法直指异构多层互连瓶颈,但更广泛的收敛特性与生产环境容错仍需验证。