跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

留还是不留:分离式LLM服务系统中的KV缓存保留学习

文章摘要

该论文提出KVLearn,一个面向分离式LLM服务架构的学习型KV缓存保留框架,通过预测前缀复用概率并结合重计算、传输与存储成本做出保留/准入决策,并在线自适应调整阈值。在文本与多模态负载上,它将首token端到端时延较无缓存降低最多56%,较LRU-Pool降低38%,较Mooncake式基线降低33%,同时将节点间KV传输量较LRU-Pool减少最多53%。

来源arXiv Machine Learning作者: Dong Liu, Yanxuan Yu, Eric Jiang, Shu Wang, Ying Nian Wu
留还是不留:分离式LLM服务系统中的KV缓存保留学习
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

arXiv 于 2026 年 8 月 10 日发布了一篇题为《To Keep or Not to Keep: Learning KV Cache Retention in Disaggregated LLM Serving Systems》(留存还是不留存:在解耦式 LLM 服务系统中学习 KV 缓存留存)的论文,作者为 Dong Liu、Yanxuan Yu、Eric Jiang、Shu Wang 和 Ying Nian Wu,论文已被 ACM SYSTOR 2026 接收,归类于机器学习(cs.LG)以及分布式、并行与集群计算(cs.DC)两个方向,编号为 arXiv:2610.28637v1,DOI 为 10.48550/arXiv.2610.28637,提交时为 89 KB,作者联系邮箱标注为 Dong Liu。

论文的出发点在于解耦式(disaggregated)大语言模型服务架构带来的根本性变化。这类系统将预填充(prefill)阶段与解码(decode)阶段拆分到不同的节点池中运行,从而在 KV 缓存的计算位置与消费位置之间引入了一层网络互联结构。作者指出,这一架构打破了传统缓存策略的一条关键假设:缓存未命中(miss)不再只意味着本地重新计算,还会额外产生把数据传送到解码节点的网络开销。同样地,把一个数据块接纳(admit)进全局缓存也需要一次额外的传输,因此糟糕的留存决策会同时浪费内存与带宽这两类宝贵资源。

为解决这一问题,作者提出了名为 KVLearn 的基于学习的框架,用于在解耦式服务中实现成本感知的 KV 缓存留存。KVLearn 由三个核心组件构成:第一是轻量级前缀复用预测器(Prefix Reuse Predictor,PRP),它依据前缀的结构特征与时间特征来估计复用概率,且无需修改模型权重;第二是成本感知留存评分(Cost-Aware Retention Score,CARS),它把复用概率转换为保留或接纳的决策,转换过程综合考虑重新计算成本、传输成本和存储成本;第三是自适应阈值控制器(Adaptive Threshold Controller,ATC),它根据观测到的命中率和内存压力在线调整接纳阈值。

在实验方面,作者将 KVLearn 集成到一个全局解耦式服务拓扑中,并在文本与多模态两类负载上进行了评测,其中包含由图像和视频衍生出的、复用模式高度异构的大型 KV 块。结果显示,与无缓存方案(No-Cache)相比,KVLearn 将端到端首 token 时延(time-to-first-token)最多降低 56%;与 LRU-Pool 相比降低 38%;与 Mooncake 风格基线相比降低 33%。同时,相对于 LRU-Pool,KVLearn 将节点间 KV 传输量最多削减 53%。在 MM-Session 工作负载上,其吞吐量仍保持在 oracle 方案的约 5% 以内。论文代码已在 https://github.com/FastLM/KVLearn 公开。

从更广的背景看,这项工作触及了解耦式服务架构下缓存管理的核心权衡:经典缓存策略(如 LRU)只针对本地重算成本做优化,而 KVLearn 试图将网络传输成本、存储成本与重算成本统一纳入同一个决策框架,并通过在线阈值调整来适应不断变化的命中率与内存压力。论文摘要未披露具体的评测硬件、集群规模或负载规模等细节,感兴趣的读者可查阅全文、TeX 源码及音频摘要等附加材料。

展开要点与分析

文章情报

工程师进阶

要点

  • 提出Prefix Reuse Predictor(PRP),无需修改模型权重即可从前缀的结构与时序特征估计复用概率
  • 提出Cost-Aware Retention Score(CARS),将复用概率结合重计算、传输和存储成本转化为保留/准入决策
  • 引入Adaptive Threshold Controller(ATC),依据观测到的命中率与内存压力在线调整准入阈值
  • 首token时延较无缓存降低56%、较LRU-Pool降低38%、较Mooncake式基线降低33%,节点间KV传输量较LRU-Pool减少53%

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。