Back to Search Start Over

Dynamic Depth Decoding: Faster Speculative Decoding for LLMs

Authors :
Brown, Oscar
Wang, Zhengjie
Do, Andrea
Mathew, Nikhil
Yu, Cheng
Publication Year :
2024

Abstract

The acceleration of Large Language Models (LLMs) with speculative decoding provides a significant runtime improvement without any loss of accuracy. Currently, EAGLE-2 is the state-of-the-art speculative decoding method, improving on EAGLE with a dynamic draft tree. We introduce Dynamic Depth Decoding (DDD), which optimises EAGLE-2's tree drafting method using a dynamic depth. This extends the average speedup that EAGLE-2 achieves over EAGLE by $44\%$, giving DDD an average speedup of $3.16$x.

Details

Database :
arXiv
Publication Type :
Report
Accession number :
edsarx.2409.00142
Document Type :
Working Paper