基于对齐反馈的多起草者推测解码 / Multi-Drafter Speculative Decoding with Alignment Feedback
1️⃣ 一句话总结
这篇论文提出了一个名为MetaSD的统一框架,通过整合多个擅长不同任务的‘起草者’模型并利用反馈动态选择它们,来显著提升大语言模型的推理速度,同时保证生成质量。
Speculative decoding (SD) accelerates large language model (LLM) inference by using a smaller model to draft future tokens, which are then verified by the target LLM. This preserves generation quality by accepting only aligned tokens. However, individual drafters, often trained for specific tasks or domains, exhibit limited effectiveness across diverse applications. To address this, we introduce \textsc{MetaSD}, a unified framework that integrates multiple drafters into the SD process. MetaSD dynamically allocates computational resources to heterogeneous drafters by leveraging alignment feedback and framing drafter selection as a multi-armed bandit problem. Extensive experiments show MetaSD consistently outperforms single-drafter approaches.
基于对齐反馈的多起草者推测解码 / Multi-Drafter Speculative Decoding with Alignment Feedback
这篇论文提出了一个名为MetaSD的统一框架,通过整合多个擅长不同任务的‘起草者’模型并利用反馈动态选择它们,来显著提升大语言模型的推理速度,同时保证生成质量。
源自 arXiv: 2604.05417