跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

用於混合和迴圈LLM服務的稀疏字首快取

文章摘要

一篇新論文介紹了稀疏字首快取,利用迴圈層從單一狀態恢復的能力,透過在稀疏檢查點儲存精確的迴圈狀態來最佳化LLM推理。一個O(NM)動態規劃演算法確定最佳檢查點位置,在實際任務上優於標準啟發式方法,減少了延遲。

來源arXiv Machine Learning作者: Mikhail Shirokikh, Sergey Nikolenko
用於混合和迴圈LLM服務的稀疏字首快取
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

大型語言模型(LLM)推理中,字首快取是一種關鍵的延遲最佳化技術。然而,現有系統假設每令牌的鍵值(KV)快取可以密集複用。狀態空間模型(SSM)和迴圈層改變了問題結構:這些模型可以從單個儲存的狀態恢復,而不需要整個令牌歷史。這種不對稱性開啟了一個新的設計點——在無複用和密集快取之間:在稀疏的檢查點位置儲存精確的迴圈狀態,並在快取命中時從最深的儲存檢查點恢復,然後精確重新計算剩餘字尾。

本文正式將稀疏字首快取定義為在重疊深度分佈下的檢查點放置問題,並提出了一個精確的O(NM)動態規劃演算法。在多個請求共享顯著字首(例如,針對同一長文件提出不同問題)的應用場景中,該方法在真實資料上持續改善了標準啟發式方法所追蹤的Pareto前沿。在QuALITY和System Prompts資料集上,基於分佈感知的放置方法在每層組Pareto前沿上優於所有固定預算基線,並且匹配或超越了最強啟發式方法(塊快取),同時使用的檢查點數量通常更少。在低檢查點預算下,當重疊分佈最不均勻時,增益最大。

該方法在多個請求共享大量但不完全相同的字首且這些字首保留在快取條目中時最為適用。它保持精確輸出,不改變迴圈計算本身,也不需要新的迴圈更新核心。其適用於迴圈/SSM層,這些層的隱藏狀態可以精確提取和恢復。對於混合模型,該方法可以與現有的KV快取壓縮技術結合使用。

此外,該方法的實現相對簡單,因為不需要修改底層的迴圈計算方式,只需在檢查點位置儲存和恢復隱藏狀態。這降低了部署門檻,使得在現有的推理系統中整合稀疏字首快取成為可能。論文中的實驗表明,即使在較少的檢查點預算下,該方法也能顯著減少快取未命中時的計算量,從而降低整體推理延遲。這對於需要處理大量具有公共字首請求的服務(如多輪對話、文件問答等)尤其有價值。

展開要點與分析

文章情報

工程師進階

要點

  • 現有字首快取假設密集的鍵/值重用,但狀態空間模型可以從單一狀態恢復。
  • 稀疏快取在選定檢查點儲存迴圈狀態,快取命中時重新計算字尾。
  • 基於分佈感知的檢查點放置透過動態規劃優於固定預算基線。
  • 方法保持精確輸出,並與現有KV快取壓縮相容。

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