GRIFFIN: Effective Token Alignment for Faster Speculative Decoding

Speculative decoding accelerates inference in large language models (LLMs) by generating multiple draft tokens simultaneously. However, existing methods often struggle with token misalignment between the training and decoding phases, limiting their performance. To address this, we propose GRIFFIN, a novel framework that incorporates a token-alignable training strategy and a token-alignable draft model to mitigate misalignment. The training strategy employs a loss masking mechanism to exclude highly misaligned tokens during training, preventing them from negatively impacting the draft model's optimization. The token-alignable draft model introduces input tokens to correct inconsistencies in generated features. Experiments on LLaMA, Vicuna, Qwen and Mixtral models demonstrate that GRIFFIN achieves an average acceptance length improvement of over 8\% and a speedup ratio exceeding 7\%, outperforming current speculative decoding state-of-the-art methods. Our code and GRIFFIN's draft models will be released publicly in https: //github. com/hsj576/GRIFFIN.

Authors

• Shijing Hu
• Jingyang Li Xidian University
• Xingyu Xie
• Zhihui Lu
• Kim-Chuan Toh National University of Singapore Institute of Operations Research and Analytics, National University of Singapore
• Pan Zhou Huazhong University of Science and Technology

Keywords

No keywords are indexed for this paper.