Attention Drift: What Autoregressive Speculative Decoding Models Learn
Speculative decoding accelerates LLM inference by drafting future tokens with a small model, but drafter models degrade sharply under template perturbation and long-context inputs. We identify a previously-unreported phenomenon we call \textbf{attention drift}: as the drafter generates successive tokens within a speculation chain, attention progressively moves from the prompt onto its own recently-generated tokens. We observe this across both \emph{EAGLE3} drafters and \emph{MTP heads}, suggesting drift is a property of drafter designs. We trace this to the un-normalized residual path between chain steps: the drafter's hidden state magnitude grows monotonically with chain depth, which exh...
| Authors | Doğaç Eldenk, Payal Mohapatra, Yigitcan Comlek et al. |
|---|---|
| Primary category | cs.LG |
| Published | 11 May 2026 |
| Last updated at source | 11 May 2026 |
Connected in the knowledge graph
- lightseekorg/kimi-k3-eagle3.1-mlaModel · cited by
Models and datasets referencing this paper
Silytics shows public metadata from arXiv and links to the original. Information may be incomplete or out of date. Last refreshed 17 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.