Flide: Accelerating Decision Inference with Exact State Forking
Bobo Li, Jiajia Song, Hao Fei, Wynne Hsu, Mong Li Lee
cs.CL
2026-10-31
把决策请求的共享上下文与所有问题压进一条 token 流、精确分叉递归状态,Flide 比 vLLM/SGLang 快至多 9.5 倍,答案最贴近 float32 参考。
Jev 这类决策模型(decision model)是 LLM agent 栈里的新角色:读一长段上下文,对若干问题各输出一份离散选项的概率分布,用于请求路由、回答评估、agent 选动作。它跑在每个请求、每一步上,下游常拿 0.5 这类阈值直接卡概率,对引擎的要求是既快又数值准。
现有引擎为文本生成设计,工作单元是一条线性序列。vLLM 和 SGLang 把每个问题当独立 prompt,共享上下文被反复渲染调度;前缀缓存(prefix cache,复用共享前缀的 KV 表示)也救不回来,hybrid 架构下它只能按块边界恢复递归状态,边界之后的 token 每个问题都要重算。SGLang 的 multi-item scoring 把所有问题拼进一条序列、以分隔符隔开,分隔符进入每个问题的输入,悄悄改掉输出概率,最多翻转 29% 的判定;动态批处理还会改变浮点求和顺序,同一请求在不同负载下给出不同答案。
根子是错配:决策请求逻辑上是树(共享主干加问题分支),物理上却被塞进线性序列执行。
Flide(Flat-Graph Inference for Decision Models)把逻辑树与物理执行拆开。
单张 H200、bfloat16,模型覆盖 Qwen3.5 0.8B–27B(hybrid)、Gemma 4 E2B–31B、Open-Jev 2B/9B,负载来自 RACE、QuALITY、MultiRC 与 Decision Index。
| 对照 | 设置 | 结果 |
| vLLM / SGLang | 多问题共享一个状态 | 延迟快至 9.5 倍,吞吐高至 8.5 倍 |
| vLLM / SGLang | 单问题、单请求 | 最小模型延迟降 3.2 倍 |
| vLLM 前缀缓存 | Gemma 4 全注意力、多问题 | 延迟降 1.1–2.9 倍 |
| vllm-jev + RadixFlow | Open-Jev 决策模型 | 短输入约 3 倍,长输入至多 1.7 倍 |
| SGLang MIS | 阈值 0.5 下判定翻转 | MIS 至多 29%,Flide 至多 2.3% |
保真度上,Flide 的 top 答案与 float32 参考的一致率在所有负载、所有尺寸上最高(RACE 平均 99.8%,vLLM 99.5%,SGLang 99.2%),准确率与 vLLM 差距不超过 1.2 个点;客户端从 1 加到 64,Flide 答案变化至多 0.2%,vLLM 到 3.2%。一个约 2 千 token 的状态挂 64 个问题时,比 vLLM 快 12.7 倍;重放 SWE-agent 轨迹,每秒步数是最强基线的 4.4–7.2 倍。消融里,去掉状态分叉,QuALITY 多问题延迟从 48.9 毫秒涨到 1868.5 毫秒;去掉 CUDA graph,短请求延迟从 8.8 涨到 76.1 毫秒。
这是服务系统工作,不是建模工作,价值在于让「快」和「准」不再是二选一。凡拿 LLM 做分类打分的场景(路由、judge、agent 的 System One 策略),只要共享上下文挂多个问题、或上下文随步数增长,收益就是数量级的;对 hybrid 架构,精确分叉补上了前缀缓存够不到递归状态的空档。MIS 翻转 29% 判定这个数字,对所有用拼接技巧省算力的人是个提醒:省下的是算力,动到的是概率。代码已开源。
边界也清楚:单问题高并发下只打平最强基线;收益随每状态问题数与历史复用度增长,纯生成负载不在射程内。
作者自述:Proposition 1 限于精确算术,浮点差异停在舍入量级,逐位模式能兜底但开销未单独报告;流尾部 padding 仍被稠密层计算;超过最大图容量的轮次拆成多次重放,单个超长请求退回无图执行。读下来另有存疑:MIS 按其配置关了 CUDA graph,吞吐对比中该基线不在最优形态;实验全部在单卡 H200 上,多卡扩展未验证,结论绑定所测引擎版本;决策模型生态还很新,这套引擎的现实影响取决于该范式能否铺开。