游客发表

阿里巴巴AMAP(高德地图)机器学习团队最新研究成果——BlockPilot,阿里以预印本形式于2026年6月30日发布于arXiv平台(论文编号:arXiv:2606.31315)。团队I提态换该研究通过引入“动态换挡”机制,挡让大模解决了大语言模型推理中的速新速度瓶颈,在Qwen3-4B模型上实现了高达4.20倍的招动推理加速。
大语言模型传统的型推运作方式称为自回归解码(Autoregressive Decoding),即模型必须逐字生成答案。理快这种方式如同打字缓慢的阿里秘书,每次只能确认一个字,团队I提态换构成了推理速度的挡让大模主要瓶颈。
为突破此限制,速新业界引入了推测解码(Speculative Decoding)技术,招动其核心逻辑为“猜题+批改”:
1. 草稿模型(轻量、型推快速):预先猜测接下来的理快一串字符。
2. 目标模型(精准、阿里缓慢):一次性验证草稿的正确性。
* 若全部猜对,直接采用;
* 若出错,则在错误处截断。
近年来,块级扩散模型(如DFlash)进一步激进地让草稿模型一次性生成整个“块”(Block)的字符,显著提升了草稿质量。然而,现有方法普遍存在一个盲点:块大小(Block Size)是固定的。
BlockPilot团队指出,固定块大小如同出租车司机无论路况如何都使用同一挡位:
* 简单任务(如数学计算):答案确定,适合大块猜测,固定小挡位导致效率低下。
* 复杂任务(如创意写作):不确定性高,适合小块谨慎猜测,固定大挡位导致大量猜测被拒绝,造成算力浪费。
实验数据显示,在Alpaca数据集上,94%的样本最佳块大小并非训练时的默认值;在GSM8K数据集上,这一比例也高达66%。固定块大小策略在大多数场景下并非最优解。
BlockPilot的核心创新在于让AI根据输入内容,自动选择当前最合适的“挡位”(块大小),在“猜得多”与“猜得准”之间取得平衡。
研究团队通过穷举实验发现,虽然不同样本的最优块大小各异,但它们高度集中在训练默认块大小 $B$ 的附近。具体而言,最优值几乎必然落在区间 $[B-3, B+3]$ 内。这一发现将原本无边界的搜索问题,压缩为一个仅含7个候选值的分类问题。
BlockPilot利用大模型推理中必然发生的预填充(Prefilling)阶段信息:
* 状态提取:提取预填充结束后,模型对下一个词的概率分布。该分布反映了任务的确定性(集中度高表示确定,分散度高表示开放)。
* 轻量分类:将概率分布输入一个仅两层的轻量级多层感知机(MLP),从5个候选块大小 ${B-2, B-1, B, B+1, B+2}$ 中选出预计验收长度最高的选项。
* 固定执行:预测仅在预填充后执行一次,随后在整个推理过程中保持块大小不变。
关键洞察:直接使用原始概率分布效果最佳。若仅使用Top-k值或进行归一化/Softmax处理,会丢失绝对概率尺度信息,导致预测准确率大幅下降(从80%跌至10%)。
研究团队在Qwen3-4B、Qwen3-8B、Llama-3.1-8B-Instruct及Qwen3-Coder-30B-A3B四款模型上进行了广泛测试,涵盖数学、代码、对话三大类任务。
| 模型 | 设置 | BlockPilot加速比 | 对比DFlash(16)最佳表现 | 平均验收长度 |
|---|---|---|---|---|
| Qwen3-4B | T=0 (确定性) | 4.17x | 3.99x | 6.59 |
| Qwen3-4B | T=1 (随机) | 4.20x | 3.80x | 5.92 |
| Qwen3-8B | T=0 | 4.66x | - | - |
| Qwen3-8B | T=1 | 3.94x | - | - |
| Llama-3.1-8B | T=0 | 3.25x | - | - |
| Llama-3.1-8B | T=1 | 2.40x | - | - |
| Qwen3-Coder | T=0 | 4.12x | - | - |
| Qwen3-Coder | T=1 | 3.95x | - | - |
增加预测器带来的额外开销微乎其微:
* 延迟:Qwen3-4B推理延迟约183ms,预测器仅增加7.34ms(占比~4%)。
* 显存:预测器额外占用约0.62GB显存(总显存8.62GB)。
* 参数量:预测器仅约0.32亿参数,远小于主干模型。
由于预测仅在预填充后执行一次,这7毫秒的开销被整个推理过程摊薄,性价比极高。
研究团队从“前缀存活过程”角度证明:
* 验收长度取决于每位字符的接受概率连乘。
* 接受概率由任务可预测性($\gamma$)和块大小匹配度($r(b,B)$)共同决定。
* 最优块大小必然集中在训练块大小 $B$ 附近,且随 $\gamma$ 变化。这从数学上证实了动态调整的必要性及局部搜索的有效性。
Q1:BlockPilot与普通大模型推理有何区别?
A:普通推理逐字生成;推测解码用小模型批量猜测、大模型验证。BlockPilot在此基础上,根据输入内容动态调整每次猜测的字符数量(块大小),而非固定不变,从而在Qwen3-4B上实现4.20倍加速。
Q2:BlockPilot如何决定块大小?
A:基于预填充后的预测概率分布。确定性高的任务(概率集中)选择大块,开放性任务(概率分散)选择小块。通过轻量级神经网络从5个候选值中选出最优解,耗时仅约7毫秒。
Q3:BlockPilot会影响输出质量吗?
A:不会。BlockPilot仅优化生成策略,最终输出仍由大模型严格验证。不符合大模型判断的草稿会被截断,因此输出质量与准确性无损,加速是完全无损的。
随机阅读
热门排行
友情链接