Search papers, labs, and topics across Lattice.
This paper investigates the phenomenon of alignment drift in single-model speculative decoding for Automatic Speech Recognition (ASR), where a lightweight draft model proposes tokens that the target model verifies. The authors find that while the draft can access the entire audio input, its proposals deteriorate over time due to misalignment with the changing audio position, leading to significant errors in later proposals. They introduce two corrective methods, one utilizing verification attention to guide the draft and another, AnchorDraft, which trains the draft to better track audio position, both of which enhance end-to-end processing speed without compromising accuracy.
Misalignment in speculative decoding can lead to a staggering 21-frame error in ASR, but innovative tracking methods can significantly boost efficiency and accuracy.
Speculative decoding speeds up generation by letting a cheap draft propose several tokens that a target model checks in one pass. In the single-model form, the draft is a lightweight module attached to the target rather than a separate model. Applying this design to Automatic Speech Recognition (ASR) introduces an extra problem. The draft can read the whole audio at every step, yet its proposals get worse as it runs on its own. Access is not localization. The accepted text keeps the transcript position explicit, but the draft must also track the changing audio position. In the primary matched comparison, per-step audio access changes the first proposal modestly but roughly doubles later-proposal acceptance. Fixed-width windows show that the audio position explains part of this gap. A correctly placed window recovers continuation, while an equally narrow window at the wrong position reduces it. Late-draft median error reaches 21 frames in the hardest reported condition, while target attention during verification stays within a 2-frame median. We test two ways to reduce this drift. The first reads the audio position from verification attention and uses it to guide the next draft round. It saves time only when the extra accepted tokens offset the readout cost. The second is AnchorDraft, which teaches the draft to track the audio position during training without changing the inference graph. The trained draft improves end-to-end speed at both tested target scales. These results show that ASR self-speculation depends on token prediction, audio-position tracking, and draft cost.