CoolFace
Modelpublic

inference-optimization/Qwen3-8B-from-Qwen3-8B_regen-speculators.eagle3-qwen3arch-ckpt1

sourceHugging Faceapache-2.0updated 4mo agoView on Hugging Face
0likes8kdownloads
Model Card

Qwen3-8B Eagle3 Drafter (Qwen3 Architecture)

Eagle3 speculative decoding drafter for Qwen/Qwen3-8B using the Qwen3 draft architecture (--draft-arch qwen3).

The Qwen3 architecture uses additional q_norm/k_norm layers in attention, which stabilize training at higher learning rates compared to the default Llama draft architecture. See speculators#563 for the RFC and experimental results.

Benchmark Results

Evaluated with sampling params temp=0.6, topp=0.95, topk=20.

Acceptance lengths

Use Casek=1k=2k=3k=4k=5k=6k=7
Coding (HumanEval, 164 samples)1.812.402.803.063.223.303.33
Math Reasoning (gsm8k, 80 samples)1.832.502.953.283.453.613.68
Text Summarization (CNN/DM, 80 samples)1.692.112.352.462.502.522.52

Comparison vs Llama-arch baseline

Trained on the same dataset with the same hyperparameters (except LR). Format: delta vs llamaarch-ckpt1.

Use Casek=1k=2k=3k=4k=5k=6k=7
Coding+0.02+0.05+0.09+0.18+0.22+0.24+0.23
Math Reasoning+0.02+0.08+0.08+0.16+0.15+0.19+0.23
Text Summarization+0.02+0.06+0.10+0.13+0.13+0.14+0.13

Outperforms the Llama-arch baseline by ~3-7% across all benchmarks and all k values, with the gap widening at higher k (math k=7: 3.68 vs 3.45, +6.7%).

Training

ParameterValue
Target modelQwen/Qwen3-8B
Draft architectureQwen3 (--draft-arch qwen3)
Learning rate5e-4
Epochs2
Draft vocab size32000
Sequence length8192
Training modeOnline (hidden states generated on-the-fly)
Datasetinference-optimization/Qwen3-8B-Regenerated-Collection (magpie + ultrachat subsets, 508k samples)
Hardware7x H200 (2 vLLM DP=2, 5 training FSDP)
Training libraryspeculators

Usage

Requires vLLM with Qwen3 Eagle3 support (vllm#43132) and the architecture resolution fix.
bash
vllm serve Qwen/Qwen3-8B \
  --speculative-config '{
    "model": "inference-optimization/Qwen3-8B-from-Qwen3-8B_regen-speculators.eagle3-qwen3arch-ckpt1",
    "num_speculative_tokens": 3,
    "method": "eagle3"
  }'

Related