DeLS-Spec: Decoupled Long-Short Contexts for Parallel Speculative Drafting

· Source: Computation and Language · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

DeLS-Spec is a novel speculative decoding method designed to accelerate Large Language Model (LLM) inference by addressing limitations in existing block-parallel drafters. While methods like DFlash improve drafting efficiency by predicting entire blocks, they lack explicit intra-block causal conditioning. Previous attempts, such as Domino and DSpark, introduced causality but necessitated expensive training of draft models from scratch. DeLS-Spec overcomes this by decoupling long and short contexts, utilizing a fixed DFlash model as a long-context expert and integrating a lightweight, independently trainable local head as a short-context expert. This approach allows the local head to be trained with a standard next-token prediction objective, avoiding joint training and resulting in extremely low training costs. During inference, DeLS-Spec combines logits from both experts, offering modularity as the local head is not tied to a specific DFlash checkpoint. Experiments on Qwen3 models demonstrate that DeLS-Spec consistently enhances speedup and average acceptance length compared to DFlash across math, code, and dialogue benchmarks.

Key takeaway

For Machine Learning Engineers optimizing LLM inference, DeLS-Spec significantly boosts speedup and acceptance length. It achieves this without high training costs. If you use block-parallel drafters like DFlash, evaluate integrating DeLS-Spec's decoupled expert architecture. This method lets you utilize existing models while independently training a lightweight component. It streamlines deployment and improves performance across math, code, and dialogue benchmarks.

Key insights

Decoupling long-short contexts in speculative decoding with an independently trained local head significantly accelerates LLM inference with minimal training cost.

Principles

Method

DeLS-Spec employs a fixed DFlash model as a long-context expert and trains a lightweight local head as a short-context expert independently via next-token prediction. At inference, it combines long and short-context logits.

In practice

Topics

Best for: AI Scientist, Machine Learning Engineer, AI Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.