arXiv Computation and Language
Oct 12

Contextual Correction of Parallel Draftsfor Speculative Decoding

The paper introduces DCOCO (DFlash Contextual Corrector), a lightweight mechanism that restores intra-block dependencies in speculative decoding by evaluating multiple candidate tokens for contextual compatibility across adjacent draft positions. DCOCO maintains parallel drafting while enabling contextually informed token selection at low additional cost. Across multiple verifier backbones and benchmarks, DCOCO achieves the highest average throughput, delivering speedups of up to 6.6× over vanilla decoding and outperforming competing methods by 6–23% under concurrent serving loads.

By Roy Eisenstadt, Harel Lidar, Tal Gonen, Omer Luxembourg, Itamar Zimerman, Lior Wolf, Eli David, Ilan Naiman
arXiv Computation and Language
1d ago

HyFlash: A Prefix-Memory Prior for On-Device Speculative Decoding

HyFlash is a compact parallel drafter designed for on-device speculative decoding. It introduces a prefix‑memory prior that separates long‑range context modeling from fine‑grained draft‑position interactions, using local attention for recent context and a recurrent memory to compress historical target features into fixed‑size states. Experiments demonstrate that HyFlash achieves higher acceptance rates than full‑context attention while requiring substantially less KV‑cache storage, especially benefiting shallow drafters and long contexts.

By Jongwoo Ko, Pashmina Cameron