[R] TriForce: Lossless Acceleration of Long Sequence Generation with Hierarchical Speculative Decoding

**Paper**: [https://arxiv.org/abs/2404.11912](https://arxiv.org/abs/2404.11912)

**Code**: [https://github.com/Infini-AI-Lab/TriForce](https://github.com/Infini-AI-Lab/TriForce)

**Project page**: [https://infini-ai-lab.github.io/TriForce/](https://infini-ai-lab.github.io/TriForce/)

**Abstract**:

>With large language models (LLMs) widely deployed in long content generation recently, there has emerged an increasing demand for efficient long-sequence inference support. However, key-value (KV) cache, which is stored to avoid re-computation, has emerged as a critical bottleneck by growing linearly in size with the sequence length. Due to the auto-regressive nature of LLMs, the entire KV cache will be loaded for every generated token, resulting in low utilization of computational …

abstract machinelearning