English

Dynamic Depth Decoding: Faster Speculative Decoding for LLMs

Computation and Language 2024-09-04 v1 Artificial Intelligence

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%44\%, giving DDD an average speedup of 3.163.16x.

Keywords

Cite

@article{arxiv.2409.00142,
  title  = {Dynamic Depth Decoding: Faster Speculative Decoding for LLMs},
  author = {Oscar Brown and Zhengjie Wang and Andrea Do and Nikhil Mathew and Cheng Yu},
  journal= {arXiv preprint arXiv:2409.00142},
  year   = {2024}
}