arXiv論文メモ
新着一覧
cs.CL · 査読状況未確認

長文対話のKVキャッシュを二段階で共有するHySparse2

HySparse2: Hybrid Sparse Attention with Two-Level KV Sharing

Jianyu Wei, Yizhao Gao, Qihao Zhang, Shimao Chen, Zhengju Tang, Yu Cheng, Shengjie Zhou, Zihan Jiang, Yifan Song, Hailin Zhang, Liang Zhao, Bo Yang, Gang Wang, Shijie Cao, Fuli Luo

この論文をやさしく読む

ひとことで言うと

長い対話を処理するモデルで、注意機構の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
査読・掲載
査読状況未確認
arXivで読むPDF

更新履歴

取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。

原文の要旨

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 / 要約の誤りについて