针对混合精度大模型推理,提出动态内存分配与双层调度,提升显存利用率和吞吐。
FineServe: Precision-Aware KV Slab and Two-Level Scheduling for Heterogeneous Precision LLM Serving

- 根据量化特性动态分配KV缓存,减少显存碎片
- 双层调度使服务达标率提升2.2倍,吞吐提高1.8倍
- 适合需高效共享GPU资源的量化大模型部署场景
后训练量化(PTQ)技术显著提升了对量化大语言模型(LLM)的服务需求,可在极小精度损失下实现更高吞吐量和大幅降低内存占用。量化模型通过更高效的GPU共享提升了资源利用率,但其KV块尺寸小于非量化模型,导致内存碎片化严重,影响内存效率。同时,量化与非量化模型在资源使用上差异明显,需高效调度以最大化吞吐。为此,我们提出FineServe,一个面向混合精度LLM推理的服务框架。核心贡献包括:(1) KV Slab,一种基于精度感知的自适应内存管理机制,根据模型量化特性动态分配KV缓存,显著降低显存碎片;(2) 双层调度框架,包含全局调度器(依据请求速率、延迟SLO和内存约束分配模型到GPU)和局部调度器(根据实时请求波动自适应调整批处理大小)。实验表明,FineServe相比当前最优的GPU共享系统,最高实现2.2倍的SLO达成率和1.8倍的生成吞吐量提升。
原文摘要 · Abstract (English)
Recent advances in Post-Training Quantization (PTQ) techniques have significantly increased demand for serving quantized large language models (LLMs), enabling higher throughput and substantially reduced memory usage with minimal accuracy loss. Quantized models address memory constraints in LLMs and enhance GPU resource utilization through efficient GPU sharing. However, quantized models have smaller KV block sizes than non-quantized models, causing limited memory efficiency due to memory fragmentation. Also, distinct resource usage patterns between quantized and non-quantized models require efficient scheduling to maximize throughput. To address these challenges, we propose FineServe, an inference serving framework for mixed-precision LLMs. FineServe's key contributions include: (1) KV Slab, a precision-aware adaptive memory management technique dynamically allocating KV cache based on model quantization characteristics, significantly reducing GPU memory fragmentation, and (2) a two-level scheduling framework comprising a global scheduler that places models to GPUs based on request rates, latency SLOs, and memory constraints and efficiency, and a local scheduler that adaptively adjusts batch sizes according to real-time request fluctuations. Experimental results demonstrate that FineServe achieves up to 2.2x higher SLO attainment and 1.8x higher token generation throughput compared to the state-of-the-art GPU sharing systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。