論文の概要: AbridgeKV: Compressing Long Reasoning into the KV Cache of Learned Pause Tokens
- arxiv url: http://arxiv.org/abs/2610.20641v1
- Date: Wed, 07 Oct 2026 17:39:41 GMT
- ステータス: 翻訳完了
- システム内更新日: 2026-10-16 19:13:15.967085
- Title: AbridgeKV: Compressing Long Reasoning into the KV Cache of Learned Pause Tokens
- Title(参考訳): AbridgeKV:学習したパストークンのKVキャッシュに長い推論を圧縮する
- Abstract要約: AbridgeKVは生成されたテキストの各ブロックを、いくつかの学習された一時停止トークンのKVエントリに圧縮する。
生成されたキャッシュの4分の1は、同じ訓練を受けたフルKVモデルと精度と出力長で一致します。
一致した予算で発行された最強のエビクターは、はるかに長く書き直され、同じデータで再訓練された最も近い訓練された圧縮機は、10ポイントを失う。
- 参考スコア(独自算出の注目度): 16.847761646903574
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Long chain-of-thought decoding is memory-bound: the KV cache of the generated tokens grows linearly with the test-time compute being spent. Evicting cache entries does not fix this, because a model never trained on a cache with entries removed generates longer, often up to the length limit. We instead train the model to compress its own output. AbridgeKV compresses each completed block of generated text into the KV entries of a few learned pause tokens. These are extra input tokens with trained embeddings that produce no output. Once the pause tokens have encoded the block, AbridgeKV frees the block's original cache; subsequent tokens attend to the pause tokens' KV entries in its place. Three novel choices make this work: the pause tokens' KV entries take positions inside the block they replace, the cache budget is allocated across layers according to the measured value of each layer's entries, and compression operates on relatively large blocks. On 446 competition problems from AIME, HMMT, AMC, BRUMO, CMIMC and SMT with 32K-token generations, keeping a quarter of the generated cache matches an identically trained full-KV model in accuracy and output length; the strongest published evictor at a matched budget writes far longer, and the closest trained compressor, retrained on the same data, loses 10 points. Served in vLLM, the cache is 3.5x smaller per sequence and a saturated H100 finishes a problem 3.2x faster than with full KV.
- Abstract(参考訳): 生成したトークンのKVキャッシュは、テスト時間計算に費やされるとともに線形に成長する。
キャッシュエントリの削除は、キャッシュ上でトレーニングされたことのないモデルが、多くの場合、長さ制限まで長くなるため、これを修正しない。
代わりに、モデルをトレーニングして、独自の出力を圧縮します。
AbridgeKVは生成されたテキストの各ブロックを、いくつかの学習された一時停止トークンのKVエントリに圧縮する。
これらは、出力を発生しない訓練された埋め込みを持つ追加の入力トークンである。
停止トークンがブロックをエンコードすると、AbridgeKVはブロックのオリジナルのキャッシュを解放する。
一時停止トークンのKVエントリは、置換したブロック内の位置を占め、キャッシュ予算は各レイヤのエントリの測定値に従ってレイヤ間で割り当てられ、圧縮は比較的大きなブロックで実行される。
AIME、HMMT、AMC、BRUMO、CMIMC、SMTの446の競合問題に対して、生成したキャッシュの4分の1は精度と出力長で同一に訓練されたフルKVモデルと一致する。
vLLMでは、キャッシュはシーケンス毎に3.5倍小さくなり、飽和したH100はフルKVよりも3.2倍早く問題を完成させる。
関連論文リスト
- End-to-End Context Compression at Scale [81.70601323130997]
長期コンテキスト言語モデル推論は、KVキャッシュがコンテキスト長とともに増加するにつれて、メモリによってボトルネックとなる。
KVキャッシュを圧縮する最近の技術は、モデル品質を著しく低下させるか、あるいはかなりの時間を要するか、1つの長いプロンプトを圧縮するために計算する。
既存のアプローチは、精度-効率のフロンティア上のKVキャッシュ圧縮と競合しない。
論文 参考訳(メタデータ) (2026-06-08T15:43:16Z) - MuKV: Multi-Grained KV Cache Compression for Long Streaming Video Question-Answering [75.0394545769057]
KVキャッシュは、LLMプリフィルを介して歴史的なトークンのキーバリューを格納する。
MuKV は KV キャッシュ圧縮モジュールと半階層的検索手法を特徴とする手法である。
長時間ストリーミングのVideoQAベンチマークの実験では、MKVはメモリとオンラインQA効率を犠牲にすることなく、回答の正確性を大幅に向上することが示された。
論文 参考訳(メタデータ) (2026-05-21T10:13:03Z) - VeriCache: Turning Lossy KV Cache into Lossless LLM Inference [24.571166055469508]
私たちは、フルKV-cacheデコードと同じ出力を保証する最初の推論フレームワークであるVeriCacheを紹介します。
VeriCacheはフルKVキャッシュよりも最大4倍高いスループットを実現していることを示す。
論文 参考訳(メタデータ) (2026-05-17T19:18:39Z) - DeltaKV: Residual-Based KV Cache Compression via Long-Range Similarity [50.52392445266824]
そこで本稿では,KV表現における長距離間類似性と高共有遅延成分を動機とする残差ベースのKVキャッシュ圧縮フレームワークを提案する。
DeltaKVはトークンを捨てる代わりに、検索した履歴参照に対するセマンティックな残基をエンコードし、保存を著しく削減する。
実験によると、DeltaKVは、LongBench、SCBench、AIMEでほぼロスレスの精度を維持しながら、KVキャッシュメモリを元の29%に削減している。
論文 参考訳(メタデータ) (2026-02-08T15:14:36Z) - ClusterKV: Manipulating LLM KV Cache in Semantic Space for Recallable Compression [10.003118268356017]
ロングコンテキストは推論効率に重大な課題をもたらす。
本稿では,意味クラスタの粒度でトークンをリコールするClusterKVを紹介する。
実験結果から、ClusterKVは32kのコンテキスト長を持つ様々なタスクにおいて、無視可能な精度の損失が得られることがわかった。
論文 参考訳(メタデータ) (2024-12-04T10:58:27Z) - CacheBlend: Fast Large Language Model Serving for RAG with Cached Knowledge Fusion [15.344568214955688]
大規模な言語モデル(LLM)は、必要なコンテキストを提供するために、入力に複数のテキストチャンクを組み込むことが多い。
長いLLM入力のプリフィルを高速化するため、別のLLM入力のプレフィックスとしてコンテキストが再利用された場合、テキストのKVキャッシュをプリコンプリートし、KVキャッシュを再使用することができる。
本稿では,KVキャッシュを高速に組み合わせて,高価なプリフィルと同世代の品質を実現する方法を提案する。
論文 参考訳(メタデータ) (2024-05-26T06:00:17Z) - Get More with LESS: Synthesizing Recurrence with KV Cache Compression for Efficient LLM Inference [78.65321721142624]
我々はキー値(KV)キャッシュによって課されるメモリボトルネックに焦点を当てる。
既存のKVキャッシュ手法は、比較的重要でないKVペアの大きなスワストを刈り取ったり、取り除いたりすることでこの問題に対処する。
本稿では,固定サイズキャッシュと退避型キャッシュを簡易に統合したLESSを提案する。
論文 参考訳(メタデータ) (2024-02-14T18:54:56Z) - KIVI: A Tuning-Free Asymmetric 2bit Quantization for KV Cache [67.9776980972508]
我々はKIVIというチューニング不要な2ビットKVキャッシュ量子化アルゴリズムを開発した。
KIVI は Llama, Falcon, Mistral のモデルを $mathbf2.6times$ less peak memory を使用しながらほぼ同じ品質を維持することができる。
論文 参考訳(メタデータ) (2024-02-05T06:06:47Z)
関連論文リストは本サイト内にある論文のタイトル・アブストラクトから自動的に作成しています。
指定された論文の情報です。
本サイトの運営者は本サイト(すべての情報・翻訳含む)の品質を保証せず、本サイト(すべての情報・翻訳含む)を使用して発生したあらゆる結果について一切の責任を負いません。