DSpark Multidomain — Gemma-4 26B-A4B Speculative Decoding Draft
Multidomain DSpark draft for google/gemma-4-26B-A4B-it. Trained for broad coverage across domains (coding, math, chat, RAG, and mixed traffic) so acceptance stays more uniform instead of peaking on a single workload. Speeds up Gemma 4 26B-A4B inference with DSpark speculative decoding in vLLM — up to ~2× output throughput vs the same target without a draft.
This checkpoint is a draft head only (not a standalone LLM). Pair it with the Gemma-4 26B-A4B-it target.
| Target | google/gemma-4-26B-A4B-it |
| Method | DSpark (num_speculative_tokens=7) |
| Peak speedup | 1.99× (math, concurrency 1) |
| Typical AL | ~2.9–3.3 accepted tokens / draft step |
| Hardware for numbers below | 8×H200, vLLM 0.28, TP=1 DP=4 |
Quick start
vllm serve google/gemma-4-26B-A4B-it \
--speculative-config '{"method": "dspark", "model": "systalyze/gemma4-26b-a4b-it-dspark-multidomain", "num_speculative_tokens": 7}'
Model details
| Property | Value |
|---|---|
| Target model | google/gemma-4-26B-A4B-it |
| Draft parameters | 1.99B (BF16) |
| Draft layers | 5 full-attention |
| Hidden size | 2816 |
| Block size / γ | 7 speculative tokens |
| Target feature layers | [3, 10, 18, 25, 28] |
| Markov head | rank 256 |
| Confidence head | enabled |
Benchmarks vs base (no draft)
Same GPUs, same prompts: base = target alone; +DSpark = this draft + target.
AL = (accepted + num_drafts) / num_drafts. aiperf 0.12, 512 requests, temperature 1.0, ignore_eos=true.
Coding throughput
ISL ~U[3500,10000], OSL ~U[200,400].
| Concurrency | Base tok/s | +DSpark tok/s | Speedup | AL |
|---|---|---|---|---|
| 1 | 172.5 | 307.7 | 1.78× | 2.89 |
| 2 | 325.0 | 561.8 | 1.73× | 2.91 |
| 4 | 555.8 | 889.6 | 1.60× | 2.84 |
| 8 | 872.3 | 1388.2 | 1.59× | 2.91 |
Speedup by domain
ISL/OSL held fixed; only prompt content changes.
| Domain | c=1 | c=2 | c=4 | c=8 |
|---|---|---|---|---|
| Math | 1.99× | 1.90× | 1.75× | 1.70× |
| Coding | 1.93× | 1.86× | 1.75× | 1.67× |
| RAG | 1.88× | 1.91× | 1.71× | 1.63× |
| Chat | 1.76× | 1.72× | 1.64× | 1.60× |
Acceptance length by domain
| Domain | c=1 | c=2 | c=4 | c=8 |
|---|---|---|---|---|
| Math | 3.31 | 3.30 | 3.30 | 3.29 |
| Coding | 3.21 | 3.21 | 3.27 | 3.21 |
| RAG | 2.99 | 3.14 | 3.01 | 2.99 |
| Chat | 2.95 | 2.90 | 2.95 | 2.91 |
Context length (coding, OSL fixed)
AL stays flat as context grows; speedup falls mainly from target verify cost.
Speedup vs base
| ISL | c=1 | c=2 | c=4 | c=8 |
|---|---|---|---|---|
| 1k | 1.89× | 1.89× | 1.81× | 1.92× |
| 4k | 1.86× | 1.79× | 1.71× | 1.70× |
| 8k | 1.75× | 1.67× | 1.56× | 1.51× |
| 16k | 1.64× | 1.56× | 1.41× | 1.37× |
Acceptance length
| ISL | c=1 | c=2 | c=4 | c=8 |
|---|---|---|---|---|
| 1k | 2.95 | 2.96 | 2.93 | 2.95 |
| 4k | 2.92 | 2.91 | 2.91 | 2.88 |
| 8k | 2.88 | 2.92 | 2.87 | 2.85 |
| 16k | 2.88 | 2.90 | 2.89 | 2.90 |
What is DSpark?
DSpark is a speculative-decoding method that combines a parallel draft backbone with a lightweight Markov head and a confidence head for scheduled verification. Training/reference code: DeepSpec.
Citation
@article{dspark2026,
title={DSpark: Confidence-Scheduled Speculative Decoding with Semi-Autoregressive Generation},
year={2026},
url={https://arxiv.org/abs/2607.05147}
}