FlashMap 流图推测解码:单次前向起草整块 token,Qwen3-8B 提速超 6 倍

kalyan_kpl · x · 2026-10-20

普林斯顿/USC 等机构发布 FlashMap,一种基于流图(flow map)的推测解码算法,在 Qwen3-8B 上实现超 6 倍推理加速,且输出与目标模型完全一致。工作原理:- 每轮从目标模型若干层读取 hidden states,用一次流图评估把高斯噪声直接映射为下 K 个 token;- 起草器是 DiT,交叉注意力目标模型 hidden states,单次前向同时起草 K 个位置;- 目标模型一次前向验证全部 K 个 token,保留最长一致前缀。与自回归起草器(准确但慢)和 DFlash 等离散扩散并行起草器(快但忽略 token 间依赖)不同,FlashMap 既并行又自洽,能建模块内联合分布。Qwen3-8B 与 Qwen3-4B 的起草器已在 Hugging Face 发布,支持 Transformers/vLLM/SGLang。单卡 H100、greedy 解码、并发 1 测试。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →