Speculative Sampling via Exponential Races
Speculative decoding accelerates large language model inference using a smaller draft model.In this paper, we establish a surprising connection between speculative sampling and the concept of channel simulation from information theory, which aims at simulating a noisy channel using as few bits as possible.This connection allows us to provide an informationtheoretic analysis of the speed up that can be achieved by speculative sampling.Leveraging this link, we derive an explicit relation between generation speed-up and the number of tokens k generated by the draft model for large k, which serves as an upper bound for all k.We also propose a novel speculative sampling method via exponential races called ERSS that matches state-of-the-art performance.
Authors
- Denız Gündüz (ORCID: https://orcid.org/0000-0002-7725-395X)
- Szymon Kobus (ORCID: https://orcid.org/0009-0006-3605-9073)
Institutions
- NIHR Imperial Biomedical Research Centre (GB)
- Imperial College London (GB)
Publication Details
- Published
- 2025-01-01
- DOI
- https://doi.org/10.18653/v1/2025.findings-acl.936
- Primary Topic
- Machine Learning and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 0.00