跳到主要內容
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%

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。

留還是不留:分離式LLM服務系統中的KV快取保留學習 | AI News Hub