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

ART:効率的な大規模言語モデルデコーディングのための注意力ランタイム終了機構

記事の要約

本論文は、カーネル実行中に累積注意力出力を追跡し、寄与が無視できるようになった時点で後続のKVブロックアクセスを終了する軽量ランタイム機構「Attention Run-time Termination(ART)」を提案する。大バッチサイズでスループットを20%向上させ、精度を維持する。

ソースarXiv Computational Linguistics著者: Chen Qiu, Guozhong Li, Panos Kalnis
ART:効率的な大規模言語モデルデコーディングのための注意力ランタイム終了機構
誤りを報告

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

訂正案内
本文へ

長いコンテキストでの大規模言語モデル(LLM)のデコーディングは、大規模なキーバリュー(KV)キャッシュを取得するために必要なメモリ帯域幅によって深刻に制約されている。既存のKV管理手法の多くは、デコーディング前にキーのみのプルーニングに依存しているが、注意力出力がキーと値の両方に依存するという証拠があるにもかかわらず、値の組み込みは prohibitive な追加オーバーヘッドを伴うためである。本論文では、Chen Qiu、Guozhong Li、Panos Kalnisの研究者らが「Attention Run-time Termination(ART)」と呼ばれる軽量ランタイム機構を提案する。

ARTの核心は、注意力カーネルの実行中に累積注意力出力をリアルタイムで追跡することにある。後続のKVブロックの寄与が無視できるほど小さくなると、ARTはそれらのブロックへのアクセスを終了する。この設計により、ARTは既存のキーベースのKVキャッシュ管理手法と直交し、競合や追加負担なしにシームレスに統合できる。

研究チームはLongBenchベンチマークでARTを評価した。実験結果は、大バッチサイズにおいて、ARTが最先端のベースライン手法と比較して生成スループットを20%向上させながら、同等の精度を維持することを示している。この成果は、ARTが長コンテキストデコーディングにおけるメモリ帯域幅のボトルネックを効果的に緩和し、LLMの効率的な推論に新たな道を開くことを示唆している。

本論文はarXivに掲載され、番号は2606.00024、計算と言語(cs.CL)分野に属する。投稿日は2026年4月15日である。ARTの提案はKVキャッシュ管理に新たな方向性をもたらし、今後さらに多くの最適化技術と組み合わせることで、大規模言語モデルの実用的な展開効率を一層推進することが期待される。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • ARTは既存のキーベースのKVキャッシュ管理手法と直交し、シームレスに統合可能である。
  • LongBenchベンチマークで、大バッチサイズ時に最先端ベースライン比で生成スループットが20%向上し、精度は同等に保たれる。
  • キーのみのプルーニングの限界に対し、値の影響をランタイムで考慮しつつ追加オーバーヘッドを回避する。

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

ART:効率的な大規模言語モデルデコーディングのための注意力ランタイム終了機構 | AI News Hub