让大模型训练不怕硬件故障,出错后仍能稳定继续
ReCoVer: Resilient LLM Pre-Training System via Fault-Tolerant Collective and Versatile Workload

- 通过保持每轮微批次数量不变,确保梯度统计等价于无故障运行
- 在512张GPU上丢失256张后仍能维持训练轨迹,有效吞吐提升2.23倍
- 适用于多种并行策略,适合大规模分布式训练场景
在大规模GPU集群上预训练大语言模型时,硬件故障已成常态而非例外,亟需具备容错能力的训练系统。现有框架或仅针对特定并行方案,或难以维持无故障训练轨迹。我们提出ReCoVer,一种韧性大模型预训练系统,核心原则是每轮迭代保持微批次数量恒定,确保每轮梯度与无故障运行下统计等价。系统分为三层解耦协议:(1)容错通信原语,隔离故障传播;(2)步骤内细粒度恢复,保留迭代内进度,防止梯度污染;(3)灵活工作负载策略,动态重分配幸存节点的微批次配额。该设计与3D并行和混合分片数据并行(HSDP)兼容,可直接作为即插即用组件。我们在最多512张GPU上评估端到端预训练任务,即使中途丢失256张GPU,ReCoVer仍能保持与无故障参考路径一致的训练轨迹。相比检查点重启基线,其在多次故障后有效吞吐提升2.23倍,训练相同时间下处理74.9%更多token,且随训练延长优势更明显。
原文摘要 · Abstract (English)
Pre-training large language models on massive GPU clusters has made hardware faults routine rather than rare, driving the need for resilient training systems. Yet existing frameworks either focus on specific parallelism schemes or risk drifting away from a failure-free training trajectory. We propose ReCoVer, a resilient LLM pre-training system that upholds a single invariant: each iteration keeps the number of microbatches constant, ensuring per-iteration gradients remain stochastically equivalent to a failure-free run. The framework is organized as three decoupled protocol layers: (1) Fault-tolerant collectives that isolate faults from propagating across replicas; (2) in-step fine-grained recovery that preserves intra-iteration progress and prevents gradient corruption; (3) versatile-workload policy that dynamically redistributes microbatch quotas across the survivors. The design is parallelism-agnostic, integrating directly with both 3D parallelism and Hybrid Sharded Data Parallel (HSDP) as a drop-in substrate. We evaluate our implementation on end-to-end pre-training tasks for up to 512 GPUs, ReCoVer successfully preserves the training trajectory from a failure-free reference despite of 256 GPUs lost spread across the run. For comparison with checkpoint-and-restart baselines, ReCoVer demonstrates $2.23\times$ higher effective throughput after successive failures. This advantage results in ReCoVer processing 74.9% more tokens at 234 GPU-hours, with the gap widening as the training prolongs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。