AI・機械学習
上級
Hydra投機的デコーディング(ハイドラ トウキテキ デコーディング)
Medusaを拡張した高度な投機的デコーディング手法。マルチヘッドデコードにドラフト検証の改良アルゴリズムを組み合わせ、Medusaの2〜3倍を上回る3〜4倍の推論高速化を単一モデルで実現。
0 回閲覧
0 いいね
2026/6/7 更新
関連タグ
投機的デコーディング
LLM推論高速化
Medusa拡張
マルチヘッドデコード
単一モデル
Hydra投機的デコーディング
概要
Hydra(Hydra: Sequentially-Dependent Draft Heads)はGhaffari et al.(2024)が発表したMedusa後継の投機的デコーディング手法です。Medusaが独立して動作する複数デコーダヘッドを使うのに対し、Hydraはヘッド間の依存関係をモデル化することで予測精度と受理率を向上させます。単一モデル設計を維持しつつ、LLaMA2-70Bで3.5倍の高速化を実現します。
Medusaとの根本的な違い
Medusaの独立ヘッド
各ヘッドがLLM最終隠れ状態 h_t のみを参照して独立予測。ヘッド間の一貫性が弱い。
Hydraの逐次依存ヘッド
h_t → Head-1: P(t+1) → 出力e_1
→ Head-2: P(t+2) ← h_t + e_1
→ Head-K: P(t+K) ← h_t + e_1 + ... + e_(K-1)
前のヘッドの出力を次のヘッドが参照することでトークン列の一貫性が向上し受理率が上がります。
パフォーマンス比較
| 手法 | LLaMA2-7B | LLaMA2-70B | CodeLlama-34B |
|---|---|---|---|
| Medusa-1 | 2.2× | 2.8× | 3.0× |
| EAGLE | 3.0× | 3.5× | 4.0× |
| Hydra | 2.8× | 3.5× | 4.2× |
EAGLEと競合するパフォーマンスで、特にコード生成タスクで優位性があります。
訓練の効率性
HydraはMedusa-1と同様に元モデルを凍結した上での軽量訓練です:
- 訓練データ: ShareGPT/SlimOrca等の指示チューニングデータ(5万〜50万サンプル)
- 訓練時間: A100 1枚で数時間〜半日
- 追加パラメータ: 元モデルの1〜3%程度(7Bモデルで約0.2B)
EAGLEとの比較
| 観点 | EAGLE | Hydra |
|---|---|---|
| アーキテクチャ | 自己回帰ドラフトモデル(小) | 逐次依存マルチヘッド |
| コードタスク | 優 | 最優 |
| 汎用タスク | 最優 | 優 |
よくある質問(FAQ)
Q1: MedusaとHydraのどちらを選ぶべきですか? A: コード生成が主なユースケースならHydra(4倍超の高速化)、汎用チャットならEAGLEかMedusa-2が現状の本番実績が豊富です。
Q2: 出力品質は保証されますか? A: 投機的デコーディングの検証機構により元モデルと等価な出力分布を保証します。
Q3: 70Bモデルに適用できますか? A: はい。70Bモデルではバッチ並列性の恩恵が大きく、3.5倍の高速化効果が実測されています。
まとめ
- Medusaの逐次依存ヘッドへの拡張でトークン予測一貫性を向上
- LLaMA2-70Bで3.5倍、コードタスクで4.2倍の高速化
- 単一モデル・元モデル凍結での軽量訓練
- 2025年現在、本番導入はMedusa/EAGLEが先行