跳到正文
arXiv cs.AI· Zhi-Kai Chen, Song-Yan Li, De-Chuan Zhan, Han-Jia Ye·· 10 天前AI 评分29

SchemaFill:通过槽位并行推测解码实现高效 LLM 工具调用

SchemaFill: Efficient LLM Tool Calling via Slot-Parallel Speculative Decoding

AI 导读

SchemaFill 是一个通过槽位并行推测解码实现高效 LLM 工具调用的框架,可并发生成未来槽位值作为候选,无需预先知道实际调用序列或参数值,候选经目标模型在实际输出前缀下验证后才提交。在 Glaive 和 BFCL 上,SchemaFill 相比自回归解码端到端吞吐量最高提升 4.05 倍。

正文 · AI 翻译

查看 PDF HTML(实验性)

摘要:LLM 智能体通过生成结构化工具调用来与外部系统交互。给定用户请求、对话上下文和工具模式目录,工具调用模型必须选择工具并生成其参数,可能在单次响应中产生多个调用。标准的自回归解码逐 token 生成这些调用,对于涉及多个调用或大量参数字段的请求会造成显著延迟。显式的参数结构为并行生成提供了机会,但后续参数值可能依赖于前面的字段和调用,因此独立生成的值可能与目标模型的输出不同。我们提出 SchemaFill,一个通过槽位并行推测解码实现高效 LLM 工具调用的框架。SchemaFill 并发地生成未来槽位值作为候选,无需预先知道实际调用序列或参数值。跨多个字段和调用的候选被拼接起来,由目标模型在实际输出前缀下进行验证。只有通过验证的 token 才会被提交,当候选不一致时由目标模型提供修正。这在利用槽位和调用间并行性的同时应用了目标验证。在 Glaive 和 BFCL 上,SchemaFill 相比自回归解码实现了高达 4.05$\times$ 的端到端吞吐量提升。代码可在 此 https URL 获取。
学科: 机器学习(cs.LG);人工智能(cs.AI)
引用为: arXiv:2610.07086 [cs.LG]
  (或 arXiv:2610.07086v1 [cs.LG] 用于此版本)
  https://doi.org/10.48550/arXiv.2610.07086

由 DataCite 颁发的 arXiv DOI(待注册)

提交历史

来自:Zhikai Chen [查看邮箱]
[v1] 2026年10月5日星期一 12:22:34 UTC(1,388 KB)

来源:arXiv cs.AI · arxiv.org