MIST模拟异构硬件上多阶段大模型推理,帮设计者低成本优化系统。
MIST: A Co-Design Framework for Heterogeneous, Multi-Stage LLM Inference
- 构建可模拟多阶段推理的异构硬件仿真框架
- 支持混合批处理与多级内存,捕捉带宽争用等关键瓶颈
- 适合系统架构师做软硬协同优化,尤其云部署场景
现代大模型服务包含检索增强生成(RAG)、KV缓存重用等多阶段流程,各阶段对计算、内存和延迟需求各异。现有推理引擎配置空间巨大,全面基准测试成本超40,000美元。同时,硬件平台快速多样化,涵盖AMD GPU、TPU及定制ASIC,跨厂商预填充-解码(PD)分离配置缺乏统一软件栈进行端到端评估。为此,我们提出MIST——一种异构多阶段大模型推理执行仿真框架。MIST可建模包括RAG、KV检索、推理、预填充和解码在内的多种请求阶段,支持异构客户端并发执行多个模型,集成先进批处理策略与多层次内存体系。通过结合真实硬件轨迹与解析建模,精准捕捉内存带宽竞争、跨集群通信延迟及混合CPU-加速器部署下的批处理效率等关键权衡。案例研究揭示了推理阶段对端到端延迟的影响、混合流水线最优批处理策略,以及远程KV缓存检索的架构影响。MIST助力系统设计师在不断演进的大模型推理生态中做出明智决策,为下一代AI负载提供可行动的软硬协同优化洞察。
原文摘要 · Abstract (English)
Modern LLM serving now spans multi-stage pipelines including RAG retrieval and KV cache reuse, each with distinct compute, memory, and latency demands. Inference engines expose a large configuration space with no systematic navigation methodology, and exhaustively benchmarking configurations can exceed 40K in cloud costs. Simultaneously, the hardware landscape is rapidly diversifying across AMD GPUs, TPUs, and custom ASICs, while cross-vendor prefill-decode (PD) disaggregated configurations lack unified software stacks for end-to-end evaluation today. To address this gap, we present MIST, a Heterogeneous Multi-stage LLM inference Execution Simulator. MIST models diverse request stages; including RAG, KV retrieval, reasoning, prefill, and decode across complex hardware hierarchies. MIST supports heterogeneous clients executing multiple models concurrently unlike prior frameworks while incorporating advanced batching strategies and multi-level memory hierarchies. By integrating real hardware traces with analytical modeling, MIST captures critical trade-offs such as memory bandwidth contention, inter-cluster communication latency, and batching efficiency in hybrid CPU-accelerator deployments. Through case studies, we explore the impact of reasoning stages on end-to-end latency, optimal batching strategies for hybrid pipelines, and the architectural implications of remote KV cache retrieval. MIST empowers system designers to navigate the evolving landscape of LLM inference, providing actionable insights into optimizing hardware-software co-design for next-generation AI workloads.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。