探索注意力与专家网络分离对高效MoE大模型推理的极限
How Far Can Disaggregation Go? A Design-Space Exploration of Attention-FFN Disaggregation for Efficient MoE LLM Serving

- 将注意力和专家前馈层分置不同GPU组,实现细粒度资源调度
- 在严格延迟要求下,系统吞吐达每秒4000个词元,非分离方案无法实现
- 给出跨工作负载的部署策略,适合作为大规模推理系统设计参考
当前大语言模型推理持续走向拆分以应对模型规模增长和严格的首字节时间(TTFT)与每字节时间(TPOT)服务目标:从分块预填充聚合,到预填充-解码(P/D)拆分,再到最近的算子级注意力-专家前馈(AFD)拆分。这一趋势对混合专家(MoE)模型尤为关键,因其存在内存密集型注意力、计算密集型专家前馈网络以及专家调度/合并通信等不同资源需求。AFD通过将注意力和MoE-FFN执行分配至独立GPU组,进一步暴露了这种异构性。每一级拆分都加深了调度设计空间,涉及工作负载特性、资源分配与互联拓扑。本文通过融合设备端内核测量与高保真网络仿真,系统评估了在真实工作负载(涵盖输入输出序列长度、前缀键值复用及用户延迟约束)下,各拆分层级的实际收益。在严格满足TTFT/TPOT SLO的前提下,基于DeepSeek-V3.2,在聊天、编程与智能体编程任务中,AFD维持约4000词元/秒的系统吞吐,而传统非拆分部署不可行。研究提炼出联合优化吞吐与交互性的具体原则,包括根据工作负载与模型结构动态划分注意力与前馈层,为当前机架与集群级部署及未来分布式AI基础设施提供设计指导。
原文摘要 · Abstract (English)
Modern large language model (LLM) inference has progressively disaggregated to keep pace with growing model sizes and tight TTFT and TPOT service-level objectives: from chunked-prefill aggregation, to prefill-decode (P/D) disaggregation, and most recently to operator-level Attention-FFN Disaggregation (AFD). This trend is especially important for mixture-of-experts (MoE) models, where memory-bound attention, compute-intensive expert FFNs, and MoE dispatch/combine communication create distinct resource demands. AFD further exposes this heterogeneity by placing attention and MoE-FFN execution on separate GPU groups. Each level of disaggregation deepens the scheduling design space across workload characteristics, resource allocation, and interconnect topology, raising the central question: when does each level actually pay off? We systematically characterize this trade-off for MoE inference across realistic workloads spanning input/output sequence lengths, prefix-KV reuse, and per-user latency constraints. Using chunked-prefill and P/D disaggregation as baselines, we study the benefits and limits of AFD at scale through a framework that fuses on-device kernel measurements with high-fidelity network simulation. Under strict TTFT/TPOT SLOs, AFD sustains around 4k tokens/s of system throughput on DeepSeek-V3.2 across chat, coding, and agentic-coding workloads, where non-AFD deployments are infeasible. We distill concrete takeaways for jointly optimizing throughput and interactivity, including how to partition attention and FFN across GPUs as a function of workload and model architecture, providing design principles for current rack- and cluster-scale deployments as well as future disaggregated AI infrastructure.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。