Quail 博客:用 trie 树让 LLM 分类快 1.8 倍,挑战 BigQuery 式自由生成

sh_reya · x · 2026-10-13

Shreya Shankar 与 Charles Frye 发布博文,介绍其 AI-SQL 执行引擎 Quail 如何执行 AI.CLASSIFY(输入文档与标签列表、返回最匹配标签),比基线快约 1.8 倍。

问题:主流做法(BigQuery 的 AI.CLASSIFY、Snowflake 的 AICLASSIFY)让 LLM 自由生成再解析标签,既浪费又脆弱——decode 逐步生成是内存瓶颈、GPU 大量闲置,输出还可能不含合法标签。

方法:Quail 在自己的推理引擎内运行 LLM,可以把生成约束到标签上,文章比较三种执行方式并估算各自成本、自动选最便宜的:

分类是 AI-SQL 最常见操作之一(如客服消息分到 billing、refund request、shipping delay 等),全文用 Qwen3-4B-fp8 作示例模型。还讨论了 decision models(为从候选答案中挑选而微调的 LLM)这一旁支。配套推文补充:标签预先存入 trie、约束解码到 trie 路径、只剩一个候选即停、把少量解码步骤打包进其他文档的 prefill 使 GPU 保持忙碌。

所属事件:Quail 用 trie 约束解码做 LLM 分类,比 vLLM 快约 1.8 倍(4 条相关)→

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →