Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Dynamic Depth Decoding: Faster Speculative Decoding for LLMs

About

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.

Oscar Brown, Zhengjie Wang, Andrea Do, Nikhil Mathew, Cheng Yu• 2024

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningGSM8K
Speed Up (x)3.58
246
Instruction FollowingAlpaca
Speedup (x)3.57
173
Multi-turn conversationMT-Bench
Speedup4.15
76
Question AnsweringQA
Speedup Factor2.96
47
SummarizationCNN/DM--
32
SummarizationCNN/DM
MAT Score6.07
30
Multi-turn dialogueMT-Bench
MAT Score6.07
30
Code GenerationHumanEval
MAT Score8.21
26
Code GenerationHumanEval
TPS (Tokens/s)253.1
25
Multi-turn Conversation EvaluationMT-Bench
Speedup3.4
25
Showing 10 of 18 rows

Other info

Follow for update