Oossa

RoLaQ、2ビットKVキャッシュの精度と速度を向上

新手法RoLaQにより、マルチヘッド潜在注意モデルの低ビット鍵・値キャッシュで、精度とデコード処理量が向上する。

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

Shiyu Ji氏らの研究チームは、マルチヘッド潜在注意で使われる2ビットの鍵・値キャッシュ向けキャリブレーション手法「RoLaQ」を公開した。モデルの重みを固定したまま、回転と量子化範囲を同時に調整する。3つのモデルと6つのベンチマークでのテストでは、2ビット手法の中で平均精度が最も高かった。DeepSeek‑V2‑Liteでは、SGLangによる実装で256Kトークンのコンテキストを使い、BF16キャッシュの3.96倍の容量と2.51倍のデコード速度を実現した。

なぜ重要か

長文コンテキストを扱う言語モデルの開発者は、追加のハードウェアなしで、より長いプロンプトをより速く処理できる。

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

次に読む

Oossa · ニュースレター

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

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