ハードウェア・推論モデル 🇷🇺 31.07.2026 16:01

負荷下のハイブリッドTransformer:Gated DeltaNetがフルアテンションに劣るケース

エンジニアが、フルアテンションを採用したQwenモデルと、Gated DeltaNetを用いたハイブリッドアーキテクチャの2つのモデルを、RTX 3090上でエージェント的な負荷下でテストしました。ハイブリッドモデルは、長いコンテキストと高い並行処理では優れていますが、共有システムプロンプトを用いた短い対話では劣ります。主な問題は、キャッシュブロックサイズが528トークンに跳ね上がることで、プレフィックスキャッシュのヒット率が低下し、プロンプトの3分の1の再計算を余儀なくされることです。
ハイブリッドモデルQwen3.5-4Bは、32層のうち8層のみでフルアテンションを使用し、残りの24層はリカレントGated DeltaNet層で、成長するKVキャッシュの代わりに固定サイズの状態を保持する。単一のRTX 3090上でvLLM 0.26.0を用いた合成エージェント負荷では、ハイブリッドモデルはプレフィックスキャッシュヒット率が84%で、フルアテンションのQwen3-4Bの94%に対して低く、プロンプトトークンの再計算が3倍多かった。根本原因は、リカレント状態ページとアテンションKVページが共通のプールを共有しており、アテンションページサイズを状態ページサイズに合わせる必要があるため、ブロックサイズが16トークンから528トークンに増加したことにある。これによりプレフィックス一致の粒度が低下し、再計算が増加する。ハイブリッドモデルは有効キャッシュ容量が3.2倍大きい(297,720トークン対93,408トークン)が、キャッシュ用メモリが2.6 GiB少なく、アクティベーションが2.3倍大きいというコストがかかる。また、ブロックサイズが大きいため、フルスナップショットが有効な場合、状態スナップショットのトークンあたりのメモリコストも増加する。
出典: Habr — хаб ИИ — 原文
関連記事 ↓
新着ニュース