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

Transformerに3つの投影は必要か?QKV変種の系統的研究

記事の要約

ICML 2026採択の研究が、Q-K=V、Q=K-V、Q=K=Vという3つの投影共有制約を系統的に評価した。Transformerは同等以上の性能を示し、Q-K=VはKVキャッシュを半減しつつ困惑度の劣化は3.1%にとどまる一方、Q=K-Vは注意の方向性を損なう。投影共有はヘッド共有(GQA/MQA)と相補的で、組み合わせにより最大96.9%のキャッシュ削減が可能となり、エッジ展開に適する。

ソースarXiv Machine Learning著者: Ali Kayyam, Anusha Madan Gopal, M Anthony Lewis
Transformerに3つの投影は必要か?QKV変種の系統的研究
誤りを報告

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

訂正案内
本文へ

TransformerはさまざまなAIタスクにおける標準的な手法となり、その中核をなすのがクエリ・キー・バリュー(QKV) attentionの定式化である。しかし、これら三つの射影がそれぞれどれだけ寄与しているのか、また一部を省略した場合に何が起きるのかは、十分に理解されていない。論文「Do Transformers Need Three Projections? Systematic Study of QKV Variants」はこの問いを体系的に検討したもので、Ali Kayyam氏ら3名による研究成果であり、ICML 2026に採択され(PMLR第306巻)、本文26ページ、図12点、表16点からなり、コードもGitHubで公開されている。

研究では三つの射影共有制約を評価している。第一はQ-K=V、すなわちキーとバリューで射影を共有する方式(共有キー・バリュー)。第二はQ=K-V、すなわちクエリとキーで射影を共有する方式(共有クエリ・キー)。第三はQ=K=V、すなわち単一射影で三つすべてを共有する方式である。著者らは、後ろ二つの変種が対称なattentionマップを生み出す点を指摘し、これに対処するため二次元位置エンコーディングを用いた非対称attentionの可能性も探索している。

実験は広範に及ぶ。合成タスクに加え、視覚タスク(MNIST、CIFAR、TinyImageNet、異常検知)と、3億パラメータおよび12億パラメータのモデルを100億トークンで学習する言語モデリングを対象とした。その結果、射影共有を用いたTransformerは標準的なQKV Transformerと同等、あるいは時に上回る性能を示した。言語モデリングでは、Q-K=Vの射影共有によりKVキャッシュを50%削減しつつ、perplexityの劣化はわずか3.1%にとどまった。

さらに重要な点として、射影共有はヘッド共有(GQA/MQA)と補完関係にある。Q-K=VをGQA-4と組み合わせるとキャッシュ削減率は87.5%に達し、Q-K=VとMQAの組み合わせでは96.9%に達するため、オンデバイス推論が現実的になる。著者らは、Q-K=Vが品質を維持できる理由を、キーとバリューが類似した表現空間を占め得ること、そしてattentionが低ランク領域で動作することに求めている。一方、Q=K-Vはattentionの方向性を損なうため性能が低下するという。

総じて本研究は、射影共有をattentionにおける未探索の重み共有(weight tying)の一形態として体系的に位置づけ、特にエッジ展開に有用な、定量的に測定可能な推論メモリ削減効果を示した。論文は2026年6月1日にarXivへ投稿され(番号2606.04032)、機械学習(cs.LG)に分類されつつ、人工知能、計算と言語、性能といった分野にもまたがっている。

要点と分析を開く

記事インテリジェンス

エンジニア上級

要点

  • Q-K=V共有は言語モデリングでKVキャッシュを50%削減し、困惑度の劣化は3.1%のみ。
  • Q=K=Vは対称な注意マップを生むが、2D位置エンコーディングで非対称性を導入できる。
  • Q-K=VとGQA-4またはMQAの組み合わせで、それぞれ87.5%と96.9%のキャッシュ削減を実現。
  • コードは公開済みで、論文はICML 2026に採択された。

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

Transformerに3つの投影は必要か?QKV変種の系統的研究 | AI News Hub