長文対話のKVキャッシュを二段階で共有するHySparse2
HySparse2: Hybrid Sparse Attention with Two-Level KV Sharing
この論文をやさしく読む
ひとことで言うと
長い対話を処理するモデルで、注意機構のKV情報を二段階で共有して計算と記憶を減らす。
何に役立つ?
道具の長い出力を読むエージェントの推論速度やKVキャッシュ量を改善する設計の参考になる。
この研究の面白いところ
クロスデコーダーのKVを自己デコーダーから作ることで、事前計算ではクロスデコーダー自体を省ける。
どこまで分かった?
比較結果は要旨に示された80B-A3Bのモデルと課題に関するもので、改善の具体的な倍率は記載されていない。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
長時間にわたり複数回対話するエージェントは、通常、短い行動を生成し、道具や環境から長い観測を処理する。増大する文脈に対応するには、効率的な事前計算、コンパクトなKVキャッシュ、正確な長文検索が必要となる。これに向け、二段階のKV共有を持つ混合疎注意機構HySparse2を導入する。外側の段階では、KV BridgingがYOCO型の自己デコーダーとクロスデコーダーを用いるが、全注意層だけを橋渡しする。自己デコーダーは混合スライディングウィンドウ注意(SWA)を使い、クロスデコーダーは混合疎注意を使う。クロスデコーダーの全注意層のKVキャッシュは、自己デコーダーの全注意層の隠れ状態から生成される。内側の段階では、HySparseの中核であるKV再利用設計を保ちながら、二点を改良する。第一に、ブロック単位の疎性をトークン単位へ変え、長文検索を細かくする。第二に、疎層から別個のSWA分枝を除き、代わりに最近のトークンのスライディングウィンドウを疎な選択に必ず含める。この二段階共有によって、クロスデコーダーのすべてのKVキャッシュを自己デコーダーの隠れ状態から構築できる。そのため、事前計算は自己デコーダーの終了時点で抜けられ、クロスデコーダーの全層を省略できる。80B-A3Bの専門家混合モデルでは、HySparse2は長文検索と複数回のエージェント課題でHySparseおよび混合SWAを上回り、事前計算量とKVキャッシュ容量も大幅に減らした。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-22(UTC)
- 最新改訂
- 2026-09-22 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-22 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Long-horizon and multi-turn agents typically generate short actions and process long observations from tools and environments. This growing context demands efficient prefill, compact KV-cache storage, and accurate long-context retrieval. To meet these demands, we introduce HySparse2, a hybrid sparse attention architecture with two-level KV sharing. At the outer level, KV Bridging adopts a YOCO-style self-decoder and cross-decoder structure, but bridges only full-attention layers. The self-decoder uses hybrid sliding-window attention (SWA), while the cross-decoder uses hybrid sparse attention. The KV caches for full-attention layers in the cross-decoder are generated from the hidden states of full-attention layers in the self-decoder. At the inner level, HySparse2 retains HySparse's core KV Reuse design with two refinements. First, it replaces block-level sparsity with token-level sparsity for finer long-context retrieval. Second, it removes the separate SWA branch from sparse layers and instead forces a sliding window of recent tokens into the sparse selection. This two-level KV sharing allows all cross-decoder KV caches to be constructed from self-decoder hidden states. Prefill can therefore exit after the self-decoder, skipping all cross-decoder layers. On an 80B-A3B MoE model, HySparse2 outperforms HySparse and Hybrid SWA on long-context retrieval and multi-turn agentic tasks, while substantially reducing prefill computation and KV-cache storage.
arXiv ID: 2609.26368 / 要約の誤りについて