Oossa

Joint Subspace Sparse Attention、長文脈LLMのデコードを高速化

新手法JSSAは、精度をフルアテンションに近く保ちながら、128Kトークンの文脈で最大6.4倍高速にデコードできる。

短信著者: Oossa公開日: 1 分で読める

Jiwon Song氏らの研究チームが、Joint Subspace Sparse Attention(JSSA)と呼ばれる手法を発表した。トークンを事前にまとめず、個別に選択することで、KVキャッシュのデータ転送量を抑える。NVIDIA B200 GPUでの測定では、128Kトークンの文脈において、フルKVアテンションと比べて最大6.43倍のデコード速度を達成した。精度はフルKVアテンションや、モデル本来の密・疎アテンションに近い水準を維持し、最大700Bパラメーターのモデルでも同様だった。

なぜ重要か

長文脈のデコードが高速化すれば、開発者は非常に大きな文書を読み取って応答できるAIアシスタントを、より低コストで機敏に構築できる。

この記事は役に立ちましたか?
共有

次に読む

Oossa · ニュースレター

今週のAIを、わかりやすく

毎週月曜日、知っておきたいニュースをやさしい言葉で。無料・スパムなし。