表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

ツール定義を個別に再利用し、AIエージェントの応答待ちとキャッシュ量を減らす

ReCache

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

ReCacheの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

(a) Causal attention among resources (relative-position bins). Orange boxes mark cross-resource blocks.

出典:ReCache / arXiv ↗

概要

  • ツール定義を個別に符号化して保存し、候補の順番や組み合わせが変わっても再利用する。
  • 主な評価では呼び出し精度が通常方式の82.4%に対して82.3%で、保存情報への命中時に最初の出力までの時間が3.655倍改善した。
  • 層・ヘッド群・保持項目を絞る全構成では、著者らはKV用メモリ92.43%削減を報告した。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、ReCacheの仕組み

左から右へ読む
  1. 01依頼とツール定義
  2. 02定義ごとに符号化
  3. 03参照経路を選別
  4. 04重要項目を保持
  5. 05呼び出しを生成

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

ReCacheは、AIエージェントが繰り返し読むツールやスキルの定義を、依頼をまたいで使い回す手法だ。同じ定義でも、依頼ごとに候補の組み合わせや順番が変わると、通常の先頭部分のキャッシュは使いにくい。

KVキャッシュは、文章を読む途中で作られる注意機構用の計算結果を保存する仕組みである。定義が増えるほど再読込の待ち時間と保存量が増えるため、呼び出しの正確さを保ちながら再利用できるかが重要になる。

手法

まず、各ツール・スキルの定義内だけで注意を計算し、位置番号も定義ごとに付け直す。これにより、別の定義が前後に置かれても同じ定義のKVキャッシュを再利用できる。例えば検索ツールと集計ツールの順番が入れ替わっても、両者の保存済み表現を個別に呼び出す。

次に、正しいツール呼び出しへの寄与を測り、重要なモデルの層とKVヘッド群だけが定義を参照するようにする。ヘッド群は、入力のどこを見るかを分担する注意機構の単位だ。最後に定義名、引数名、引数説明、末尾の要素を残し、呼び出しに必要な情報を保ちながら保存量を減らす。

新規性

ReCacheの要点は、ツールやスキルの定義を、並び順や隣接する定義に依存しない形で個別に記憶することだ。資源間の参照をなくし、各定義内の位置番号を付け直すことで、組み合わせが変わっても同じ定義の計算結果を使える。

さらに、呼び出しへの寄与が大きい層とKVヘッド群だけに定義を見せ、名前・引数名・引数説明などを残して保持量を減らす。再利用可能性と圧縮を一続きの設計として扱う。

従来手法との違い

標準的な先頭部分のキャッシュは、入力の冒頭が同一のときに再利用しやすい。ツール定義の組み合わせや順序が変わる場面では、その条件が崩れる。比較対象のDenseは、複数の定義を通常どおり一緒に読み込む方式である。

ReCacheは定義間の参照を遮断して個別の計算結果を作るため、同じ定義を再計算せずに済む。著者らの評価では、この段階の呼び出し精度はDenseとほぼ同じだった。圧縮を追加した段階の効果は、再利用だけの効果と分けて読む必要がある。

実験結果

著者らは公開の7データセットから学習49,424例を構成し、既知の定義と学習時に見ない定義について各1,000例で評価した。Inv-F1は、必要なツール名と全引数が一致した呼び出しを測る指標である。

主なQwen3-4Bの評価では、個別符号化後のInv-F1は82.3%で、通常方式の82.4%に近かった。保存済み情報が使える場合、最初の出力までの時間は26.319ミリ秒から7.200ミリ秒となり、3.655倍の高速化と報告した。全構成ではKV用メモリ92.43%削減、注意計算1.423倍高速化を報告している。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:社内の申請業務で、AIエージェントが複数の社内ツールから必要なものを選ぶ画面に使う。入力は利用者の依頼と、その時点で候補となるツール定義で、出力は選択したツール名と引数である。

定義が更新されたら対応する保存済み情報を作り直し、通常の依頼では再利用する。担当者はツール名と引数が正しいかを確認し、呼び出し前の待ち時間を記録して、運用上の効果を判断する。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:実際に使うツール定義と、正しいツール名・引数が分かる依頼を小規模に集める。通常の読み込み方式を基準に、定義を再利用する方式、さらに圧縮した方式を同じ依頼で比較する。

ツール名と全引数の一致率、無効なツール名の割合、最初の出力までの時間、KV用メモリを測る。既知の定義と新しい定義を分け、定義の更新後に古い保存情報が使われないことも確認する。

限界と課題

著者らの効果測定は、主にQwen3-4Bと構成したツール・スキル利用データでの結果である。最初の出力までの時間は保存済み情報に当たる条件で測り、保存情報を事前に作る費用は含めない。

編集上の確認点は、定義が頻繁に更新される環境で再作成費用がどの程度か、必要な引数の制約を削りすぎないかである。未知のツールを含む依頼では、呼び出しの正確さと無効なツール名の発生率を別に追いたい。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

ReCache: Efficient KV Cache Reuse and Compression for Tool-Augmented LLM Agents

Agentic language models repeatedly encode tool and skill schemas that recur across requests in different combinations and orders, preventing standard prefix caching from reusing their key--value (KV) states. We introduce \textbf{ReCache}, a framework for independently caching resource representations while reducing their inference-time computational and memory overhead. Resource-wise attention removes cross-resource interactions and assigns resource-local positions, producing composition-invariant KV blocks. ReCache then restricts resource visibility to contribution-selected layer--KV-head-group routes and retains only invocation-critical fields through structural and semantic pruning. We evaluate ReCache on a benchmark assembled from seven public tool- and skill-use datasets, including resource-disjoint tests. Resource-wise attention matches dense invocation performance (82.3\% versus 82.4\% Inv-F1) while providing a 3.655$\times$ time-to-first-token speedup. The complete framework reduces allocated KV-tensor memory by 92.43\% and accelerates attention by 1.423$\times$. These results show that separating reusable schema encoding from selective resource access substantially reduces agentic inference costs with limited effectiveness loss. The code is available at https://github.com/EIT-NLP/ReCache.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-30