拆分大模型推理阶段,发现能耗反而更高
Revisiting Disaggregated Large Language Model Serving for Performance and Energy Implications
- 将预填充与解码分离到不同GPU,通过不同缓存传输路径对比性能
- 在高负载下拆分可提升性能,但整体能耗显著高于传统部署
- 独立调频未节能,因拆分本身增加能耗,适合特定场景优化
不同于传统将预填充和解码阶段共置于同一GPU的LLM服务方式,拆分式服务将预填充与解码分别分配至不同GPU。预填充完成后,需将键值缓存(KV cache)传输至解码GPU。尽管已有研究提出多种跨内存与存储层级的KV缓存传输路径,但缺乏对性能与能效的系统性评估。同时,虽有如缓存复用、频率调节等优化技术应用于拆分式服务,其实际性能与能耗影响尚未被严谨验证。本文通过重新评估不同缓存传输介质及优化策略下的预填充-解码拆分架构,引入新的共置服务基线,对比多种拆分配置。结合动态电压频率调节(DVFS)的GPU性能分析,识别各方案的性能-能耗帕累托前沿,评估拆分带来的潜在节能效果。结果表明,拆分带来的性能优势并非必然,取决于请求负载与缓存传输路径;此外,拆分带来的逐阶段独立频率调节并未实现节能,因拆分架构本身能耗更高。
原文摘要 · Abstract (English)
Different from traditional Large Language Model (LLM) serving that colocates the prefill and decode stages on the same GPU, disaggregated serving dedicates distinct GPUs to prefill and decode workload. Once the prefill GPU completes its task, the KV cache must be transferred to the decode GPU. While existing works have proposed various KV cache transfer paths across different memory and storage tiers, there remains a lack of systematic benchmarking that compares their performance and energy efficiency. Meanwhile, although optimization techniques such as KV cache reuse and frequency scaling have been utilized for disaggregated serving, their performance and energy implications have not been rigorously benchmarked. In this paper, we fill this research gap by re-evaluating prefill-decode disaggregation under different KV transfer mediums and optimization strategies. Specifically, we include a new colocated serving baseline and evaluate disaggregated setups under different KV cache transfer paths. Through GPU profiling using dynamic voltage and frequency scaling (DVFS), we identify and compare the performance-energy Pareto frontiers across all setups to evaluate the potential energy savings enabled by disaggregation. Our results show that performance benefits from prefill-decode disaggregation are not guaranteed and depend on the request load and KV transfer mediums. In addition, stage-wise independent frequency scaling enabled by disaggregation does not lead to energy saving due to inherently higher energy consumption of disaggregated serving.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。