Papers by Sungsoo Ha
BASS: Batched Attention-optimized Speculative Sampling (2024.findings-acl)
Copied to clipboard
Haifeng Qian, Sujan Kumar Gonugondla, Sungsoo Ha, Mingyue Shang, Sanjay Krishna Gouda, Ramesh Nallapati, Sudipta Sengupta, Xiaofei Ma, Anoop Deoras
| Challenge: | Speculative decoding has emerged as a powerful method to improve latency and throughput in hosting large language models. |
| Approach: | They propose a batched speculative decoding system that generates sequences at an average speed of 5.8ms per token and a batch size of 8 at a 2.15 speed-up over optimized regular decoding. |
| Outcome: | The proposed system achieves state-of-the-art latency and speed-up over optimized regular decoding. |