mlsanityData to 14 Oct 2026

Osprey: Target-agnostic Pre-training Makes Stronger Drafters in Speculative Decoding

arXiv 2609.09338 ↗First posted 8 Sep 2026cs.CLcited by 0

On the map: Efficient LLM inference and serving → Speculative decoding

Abstract (arXiv)

Speculative decoding is critical for accelerating LLM inference. However, the speedup is fragile: drafters are typically trained against a narrow distribution for a single target model, and their acceptance rate collapses under workload shifts. This is a striking inversion of modern LLM development, where target models are valued precisely for the broad generalization they acquire through large-scale pretraining. We argue that the natural remedy, pretraining, has been hard to apply to drafters because existing recipes are target-specific: the drafter consumes the target's hidden states and is distilled on the target's logits, so pretraining must be repeated for each target. We introduce Osprey, which instead bootstraps drafters from off-the-shelf pretrained small language models, treating broad pretraining as a reusable, target-agnostic asset and reducing per-target work to a lightweight adaptation step. Realizing this requires overcoming two challenges: small LMs are far deeper than a latency-bound drafter can afford, and their pretrained computation must remain intact while the drafter learns to ingest target hidden states and emit tokens in the target's vocabulary. Osprey addresses both by pruning to a shallow backbone, restoring its language-modeling capability with target-agnostic next-token pretraining, and adapting it to each target through vocabulary alignment, zero-initialized QKV expansion, and distillation from the target model's output distribution. Empirically, a single pretrained Osprey backbone transfers across targets and improves mean acceptance length by 16.1% for Qwen3-8B, 21.2% for Llama-3.3-70B-Instruct, and 22.7% for the 229B MiniMax-M2.5 (with 17.5% higher tokens per second), with the largest gains on out-of-domain and multilingual data. Our code is available at https://github.com/LeanModels/Osprey.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Speculative decoding, one of 12 directions in Efficient LLM inference and serving. Builds on 48 of its 63 references that are in the map. First posted on arXiv 8 Sep 2026.

Builds on

48 of 63 references.

Closest ancestors · 23

Same direction, newest first

  1. 2026DFlash: Block Diffusion for Flash Speculative Decoding5 Feb
  2. 2025Steering Pretrained Drafters during Speculative Decoding13 Nov
  3. 2025The Disparate Impacts of Speculative Decoding2 Oct
  4. 2025OmniDraft: A Cross-vocabulary, Online Adaptive Drafter for On-device Speculative Decoding3 Jul
  5. 2025Training Domain Draft Models for Speculative Decoding: Best Practices and Insights10 Mar
  6. 2025EAGLE-3: Scaling up Inference Acceleration of Large Language Models via Training-Time Test3 Mar
  7. 2025Accelerating LLM Inference with Lossless Speculative Decoding Algorithms for Heterogeneous Vocabularies31 Jan
  8. 2024Falcon: Faster and Parallel Inference of Large Language Models through Enhanced Semi-Autoregressive Drafting and Custom-Designed Decoding Tree17 Dec
  9. 2024SWIFT: On-the-Fly Self-Speculative Decoding for LLM Inference Acceleration9 Oct
  10. 2024Learning Harmonized Representations for Speculative Sampling28 Aug
  11. 2024Clover-2: Accurate Inference for Regressive Lightweight Speculative Decoding1 Aug
  12. 2024EAGLE-2: Faster Inference of Language Models with Dynamic Draft Trees24 Jun
11 more
  1. 2024Clover: Regressive Lightweight Speculative Decoding with Sequential Knowledge1 May
  2. 2024Kangaroo: Lossless Self-Speculative Decoding via Double Early Exiting29 Apr
  3. 2024Speculative Streaming: Fast LLM Inference without Auxiliary Models16 Feb
  4. 2024Break the Sequential Dependency of LLM Inference Using Lookahead Decoding3 Feb
  5. 2024EAGLE: Speculative Sampling Requires Rethinking Feature Uncertainty26 Jan
  6. 2024Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Heads19 Jan
  7. 2024Unlocking Efficiency in Large Language Model Inference: A Comprehensive Survey of Speculative Decoding15 Jan
  8. 2023Accelerating Transformer Inference for Translation via Parallel Decoding17 May
  9. 2023Accelerating Large Language Model Decoding with Speculative Sampling2 Feb
  10. 2022Fast Inference from Transformers via Speculative Decoding30 Nov
  11. 2022Speculative Decoding: Exploiting Speculative Execution for Accelerating Seq2seq Generation30 Mar

Other references · 25

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2015Distilling the Knowledge in a Neural Networkcited by 10,324
  3. 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
  4. 2021Evaluating Large Language Models Trained on Codecited by 8,320
  5. 2020Scaling Laws for Neural Language Modelscited by 6,586
  6. 2021Measuring Mathematical Problem Solving With the MATH Datasetcited by 5,371
  7. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  8. 2024LiveCodeBench: Holistic and Contamination Free Evaluation of Large Language Models for Codecited by 2,004
  9. 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,565
  10. 2024The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scalecited by 1,160
  11. 2023SGLang: Efficient Execution of Structured Language Model Programscited by 1,141
  12. 2023On-Policy Distillation of Language Models: Learning from Self-Generated Mistakescited by 939
13 more
  1. 2016Sequence-Level Knowledge Distillationcited by 801
  2. 2024TinyLlama: An Open-Source Small Language Modelcited by 618
  3. 2023Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruningcited by 415
  4. 2024ShortGPT: Layers in Large Language Models are More Redundant Than You Expectcited by 383
  5. 2024Better & Faster Large Language Models via Multi-token Predictioncited by 360
  6. 2024LayerSkip: Enabling Early Exit Inference and Self-Speculative Decodingcited by 307
  7. 2025Small Language Models are the Future of Agentic AIcited by 270
  8. 2024The Unreasonable Ineffectiveness of the Deeper Layerscited by 226
  9. 2025NVIDIA Nemotron Nano 2: An Accurate and Efficient Hybrid Mamba-Transformer Reasoning Modelcited by 108
  10. 2021WECHSEL: Effective initialization of subword embeddings for cross-lingual transfer of monolingual language modelscited by 89
  11. 2024The Perfect Blend: Redefining RLHF with Mixture of Judgescited by 43
  12. 2024Fast Vocabulary Transfer for Language Model Compressioncited by 42
  13. 2024An Empirical Comparison of Vocabulary Expansion and Initialization Approaches for Language Modelscited by 20

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024Towards Fast Multilingual LLM Inference: Speculative Decoding and Specialized Drafters
  2. 2026Carryover Drafting: Recycling Rejected States for Speculative Decoding
  3. 2026Variational Speculative Decoding: Rethinking Draft Training from Token Likelihood to Sequence Acceptance
  4. 2026Learning to Draft: Adaptive Speculative Decoding with Reinforcement Learning
  5. 2025READER: Retrieval-Assisted Drafter for Efficient LLM Inference
  6. 2026Margins, Not Windows: Training-Free Per-Step Lossy Speculative Decoding
  7. 2026Faster LLM Inference via Sequential Monte Carlo
  8. 2023Cascade Speculative Drafting for Even Faster LLM Inference