本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

保持すべきか、破棄すべきか:分離型LLMサービングにおけるKVキャッシュ保持の学習

記事の要約

本論文は、分離型LLMサービング向けの学習ベースKVキャッシュ保持フレームワークKVLearnを提案する。プレフィックスの再利用確率を予測し、再計算・転送・保存の各コストを用いて保持/受け入れ判断へ変換し、さらに適応的しきい値制御でオンライン調整する。テキストおよびマルチモーダル負荷において、エンドツーエンドの初回トークン時間をNo-Cache比で最大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キャッシュ保持の学習
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

プレフィルとデコードを異なるノードプールに分離して実行する分離型(disaggregated)LLMサービングでは、KVキャッシュの計算位置と消費位置の間にネットワークファブリックが介在する。論文は、この構造が従来のキャッシュポリシーの重要な前提を崩すと指摘する。すなわち、キャッシュミスはローカルな再計算だけでなく、デコードノードへのネットワーク転送も伴う。同様に、ブロックをグローバルキャッシュへ受け入れる際にも追加の転送が必要となるため、不適切な保持判断はメモリと帯域の双方を浪費する。

著者らはこの課題に対し、分離型サービングにおけるコストを意識した学習ベースのKVキャッシュ保持フレームワーク「KVLearn」を提案する。KVLearnは三つの要素で構成される。第一に、軽量なPrefix Reuse Predictor(PRP)で、モデル重みを変更することなく、プレフィックスの構造的特徴と時間的特徴から再利用確率を推定する。第二に、Cost-Aware Retention Score(CARS)で、再利用確率に再計算コスト・転送コスト・保存コストを組み合わせ、保持か受け入れかの二者択一の判断へ変換する。第三に、Adaptive Threshold Controller(ATC)で、実際に観測されたヒット率とメモリ圧迫度に応じて受け入れしきい値をオンラインで調整し、負荷変動に追随できるようにする。

研究チームはKVLearnをグローバルな分離型サービングトポロジに統合し、テキストとマルチモーダルの負荷で評価した。評価には、画像や動画由来の大規模なKVブロックで再利用パターンが大きく異なるケースも含まれる。その結果、エンドツーエンドの初回トークン時間(TTFT)は、No-Cache比で最大56%、LRU-Pool比で38%、Mooncake系ベースライン比で33%削減された。ノード間のKV転送量はLRU-Pool比で最大53%削減されている。マルチモーダルのMM-Sessionでは、スループットがオラクルから約5%以内に収まった。

本論文はACM SYSTOR 2026に採択され、コードはGitHubのFastLM/KVLearnリポジトリで公開されている。分離型アーキテクチャにおけるキャッシュ管理のコストモデルを「ローカル再計算」から「再計算+ネットワーク転送+保存」のトレードオフへと拡張し、変化する再利用パターンにオンライン学習で適応する点に意義がある。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • 軽量なPrefix Reuse Predictor(PRP)が、モデル重みを変更せずにプレフィックスの構造的・時間的特徴から再利用確率を推定する
  • Cost-Aware Retention Score(CARS)が、再計算・転送・保存コストを用いて再利用確率を保持/受け入れ判断に変換する
  • Adaptive Threshold Controller(ATC)が、観測されたヒット率とメモリ圧迫度に基づいて受け入れしきい値をオンラインで調整する
  • 初回トークン時間をNo-Cache比56%、LRU-Pool比38%、Mooncake系ベースライン比33%削減し、ノード間KV転送量をLRU-Pool比53%削減する

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。