Large language models (LLMs) have achieved impressive success across natural language processing tasks but face persistent challenges in inference efficiency due to their autoregressive nature. While speculative decoding and beam sampling offer improvements, traditional methods verify draft sequences sequentially and uniformly, causing unnecessary overhead. This work proposes the Hierarchical Verification Tree (HVT), a framework that restructures speculative beam decoding by prioritizing high-likelihood drafts and pruning suboptimal candidates early. A formal verification-pruning algorithm ensures correctness and efficiency. HVT integrates with standard LLM pipelines without retraining. Experiments demonstrate that HVT outperforms existing methods, achieving significant reductions in inference time and energy consumption while maintaining output quality. These results highlight the promise of hierarchical verification for accelerating LLM inference.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Hierarchical Verification of Speculative Beams for Accelerating LLM Inference

  • Jaydip Sen,
  • Harshitha Puvvala,
  • Subhasis Dasgupta

摘要

Large language models (LLMs) have achieved impressive success across natural language processing tasks but face persistent challenges in inference efficiency due to their autoregressive nature. While speculative decoding and beam sampling offer improvements, traditional methods verify draft sequences sequentially and uniformly, causing unnecessary overhead. This work proposes the Hierarchical Verification Tree (HVT), a framework that restructures speculative beam decoding by prioritizing high-likelihood drafts and pruning suboptimal candidates early. A formal verification-pruning algorithm ensures correctness and efficiency. HVT integrates with standard LLM pipelines without retraining. Experiments demonstrate that HVT outperforms existing methods, achieving significant reductions in inference time and energy consumption while maintaining output quality. These results highlight the promise of hierarchical verification for accelerating LLM inference.