一键优化大模型推理配置,30秒内提速50%
AIConfigurator: Lightning-Fast Configuration Optimization for Multi-Framework LLM Serving
- 将推理拆解为可分析的计算原语,建模框架调度特性
- 跨硬件与模型测试,实现密集型模型提速40%、MoE达50%
- 无需实机调试,适配生产环境,支持多框架自动配置
由于动态负载、严苛的延迟/吞吐要求以及不断扩大的配置空间,大型语言模型(LLM)在生产系统中的推理优化日益困难。这一复杂性不仅涉及分布式并行策略(张量/流水线/专家),还包括各框架特有的运行时参数,如CUDA图启用、可用KV缓存内存比例和最大令牌容量,这些参数对性能影响巨大。现代推理框架(如TRT-LLM、vLLM、SGLang)采用不同内核和执行策略,手动调优既框架特异又计算成本高昂。本文提出AIConfigurator,一个统一的性能建模系统,可在无需基于GPU的性能测试的前提下,快速实现跨框架的推理配置搜索。该系统包含:(1) 将推理分解为可解析建模的基本操作——GEMM、注意力、通信与内存操作,并捕捉框架特定调度动态;(2) 覆盖多种硬件平台与主流开源模型(GPT-OSS、Qwen、DeepSeek、LLama、Mistral)的校准级内核性能数据库;(3) 自动解析目标后端最优启动参数的抽象层,无缝集成至生产级编排系统。在真实生产负载上的评估表明,AIConfigurator能在平均30秒内找到更优配置,使密集模型(如Qwen3-32B)性能提升最高40%,MoE架构(如DeepSeek-V3)提升最高50%,并支持从集群拓扑到引擎特定标志的广阔设计空间探索。
原文摘要 · Abstract (English)
Optimizing Large Language Model (LLM) inference in production systems is increasingly difficult due to dynamic workloads, stringent latency/throughput targets, and a rapidly expanding configuration space. This complexity spans not only distributed parallelism strategies (tensor/pipeline/expert) but also intricate framework-specific runtime parameters such as those concerning the enablement of CUDA graphs, available KV-cache memory fractions, and maximum token capacity, which drastically impact performance. The diversity of modern inference frameworks (e.g., TRT-LLM, vLLM, SGLang), each employing distinct kernels and execution policies, makes manual tuning both framework-specific and computationally prohibitive. We present AIConfigurator, a unified performance-modeling system that enables rapid, framework-agnostic inference configuration search without requiring GPU-based profiling. AIConfigurator combines (1) a methodology that decomposes inference into analytically modelable primitives - GEMM, attention, communication, and memory operations while capturing framework-specific scheduling dynamics; (2) a calibrated kernel-level performance database for these primitives across a wide range of hardware platforms and popular open-weights models (GPT-OSS, Qwen, DeepSeek, LLama, Mistral); and (3) an abstraction layer that automatically resolves optimal launch parameters for the target backend, seamlessly integrating into production-grade orchestration systems. Evaluation on production LLM serving workloads demonstrates that AIConfigurator identifies superior serving configurations that improve performance by up to 40% for dense models (e.g., Qwen3-32B) and 50% for MoE architectures (e.g., DeepSeek-V3), while completing searches within 30 seconds on average. Enabling the rapid exploration of vast design spaces - from cluster topology down to engine specific flags.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。