paper.dinq.meAI 与前沿科学顶会 · 学术新星雷达
登录 DINQ
返回新星榜
KB

Karim Bouyarmane

COLM 2026ICLR 2026效率与系统可解释性数据与预训练推理强化学习与后训练
新星分资深
19
一作贡献0
核心/参与5
方向广度13
合作网络3
0 一作 · 2 篇 · COLM 2026 / ICLR 2026

获取完整研究档案

首次获取该候选人的完整档案及 StartLux 评估共 20 积分;每位用户单独解锁,之后查看免费。主动刷新 20 积分。

联系方式

实时查找电子邮箱,费用不超过20积分;未找到邮箱不扣费。

2026 顶会论文

2
作者位次 #2/2COLM 2026Poster· Poster Session 4· Imperial Ballroom· #73

Trie Automata for Constrained Decoding over Large Finite Sets

Xingzi Xu · Karim Bouyarmane

We propose a trie automaton for constrained decoding over finite sets uses Aho-Corasick matching to precompute token masks, breaking the cardinality wall of grammar engines with 29x throughput over XGrammar, scaling to 100K+ values in sub-100ms.

摘要

Large language models increasingly need to generate structured outputs that conform to predefined schemas, with one common constraint being selection from a finite set of valid strings. Current constrained decoding systems handle this through general-purpose grammar compilation, which becomes prohibitively slow as the number of valid values grows into the thousands, a \emph{cardinality wall}. We introduce the \emph{trie automaton}, a specialized mechanism that exploits finite-set structure (shared prefixes, bounded depth, known cardinality) via Aho-Corasick multi-pattern matching to precompute per-node token masks. The trie achieves 7$\times$ faster per-step valid-token computation (0.65$\mu$s vs.\ 5.8$\mu$s) compared to XGrammar, one of the primary backends in vLLM and SGLang, and 2--6.5$\times$ faster compilation at $K \geq 300$. Because precomputed masks enable a stateless serving path that bypasses the guided decoding pipeline, this advantage compounds in batch serving: end-to-end vLLM throughput reaches 219 req/s vs.\ XGrammar's 7.5 req/s at batch size 256 (29$\times$). The 29$\times$ combines the algorithmic speedup with integration-path savings that only precomputed masks can unlock. Across seven tokenizer families (32K--262K vocabulary), the trie maintains sub-100ms compilation up to $K = 10{,}000$ and flat per-step cost regardless of set size, while guaranteeing 100\% output validity.

可解释性效率与系统数据与预训练TrieConstrained DecodingFinite State MachineOpenReview

常见合作者