Tarragon让MoE大模型推理抗故障,单个失败不拖垮全系统。
Making MoE-based LLM Inference Resilient with Tarragon
- 分离注意力与专家计算,分治故障影响范围。
- 故障后恢复仅需0.3-0.4秒,比现有方案快160-213倍。
- 适合高可用、低延迟的大模型推理服务场景。
Mixture-of-Experts(MoE)模型正被广泛用于大规模部署大语言模型(LLM),但随着规模扩大,故障频发。现有系统故障容错能力差:单个工作节点失效即触发全局粗粒度重启,导致已积累进度丢失,推理流水线停滞——这显然不适用于对延迟敏感的LLM服务。本文提出Tarragon,一种具备韧性的MoE推理框架,将故障影响限制在单个工作节点内,其余部分仍可继续推进。Tarragon利用MoE Transformer中注意力计算与专家计算的天然分离特性,将注意力工作节点(AWs)和专家工作节点(EWs)视为独立的故障域。通过可重构数据路径,自动将请求重定向至健康节点以掩盖故障。在此基础上,Tarragon实现自愈机制:对有状态的AWs,采用异步增量式KV缓存检查点并支持按请求恢复;对无状态的EWs,利用残余GPU内存部署影子专家。上述设计使恢复成本和重计算开销极低。评估显示,相比最先进的MegaScale-Infer,Tarragon将故障引发的停顿时间降低160-213倍(从约64秒降至0.3-0.4秒),且在无故障时性能保持不变。
原文摘要 · Abstract (English)
Mixture-of-Experts (MoE) models are increasingly used to serve LLMs at scale, but failures become common as deployment scale grows. Existing systems exhibit poor failure resilience: even a single worker failure triggers a coarse-grained, service-wide restart, discarding accumulated progress and halting the entire inference pipeline during recovery--an approach clearly ill-suited for latency-sensitive, LLM services. We present Tarragon, a resilient MoE inference framework that confines the failures impact to individual workers while allowing the rest of the pipeline to continue making forward progress. Tarragon exploits the natural separation between the attention and expert computation in MoE-based transformers, treating attention workers (AWs) and expert workers (EWs) as distinct failure domains. Tarragon introduces a reconfigurable datapath to mask failures by rerouting requests to healthy workers. On top of this datapath, Tarragon implements a self-healing mechanism that relaxes the tightly synchronized execution of existing MoE frameworks. For stateful AWs, Tarragon performs asynchronous, incremental KV cache checkpointing with per-request restoration, and for stateless EWs, it leverages residual GPU memory to deploy shadow experts. These together keep recovery cost and recomputation overhead extremely low. Our evaluation shows that, compared to state-of-the-art MegaScale-Infer, Tarragon reduces failure-induced stalls by 160-213x (from ~64 s down to 0.3-0.4 s) while preserving performance when no failures occur.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。