LLM2D
XGrammar:大型语言模型灵活高效的结构化生成引擎
XGrammar: Flexible and Efficient Structured Generation Engine for Large Language Models
作者: Yixin Dong, Charlie F. Ruan, Yaxing Cai, Ruihang Lai, Ziyi Xu, Yilong Zhao, Tianqi Chen
发布日期: 11/28/2024
arXiv ID: oai:arXiv.org:2411.15100v2

摘要

大型语言模型 (LLM) 代理的应用日益复杂和多样化,对可解析为代码、结构化函数调用和具体代理命令的结构化输出的需求也日益增长。这些发展对 LLM 推理中的结构化生成提出了重大需求。上下文无关文法是一种通过约束解码实现结构化生成的灵活方法。然而,执行上下文无关文法需要在运行时遍历词汇表中所有标记的多个堆栈状态,这给结构化生成带来了不可忽略的开销。在本文中,我们提出了 XGrammar,这是一种用于大型语言模型的灵活高效的结构生成引擎。XGrammar 通过将词汇表划分为可以预先检查的上下文无关标记和需要在运行时解释的上下文相关标记来加速上下文无关文法的执行。我们进一步构建转换以扩展语法上下文并减少上下文无关标记的数量。此外,我们构建了一个高效的持久堆栈来加速上下文相关标记的检查。最后,我们将语法引擎与 LLM 推理引擎共同设计,以使语法计算与 GPU 执行重叠。评估结果表明,XGrammar 可以比现有解决方案快 100 倍。结合 LLM 推理引擎,它可以在端到端低 LLM 服务中实现接近零开销的结构化生成。