LoopAhead 提出循环语言模型推测解码框架,解码最高提速 7.82 倍

论文AI 评分 68/100arXiv cs.AI待复核
AI 聚合

本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。

一篇投稿于 arXiv cs.AI 的论文提出 LoopAhead,面向循环 Transformer 的推测解码框架。循环 Transformer 通过反复复用同一批层来提升参数效率,但逐 token 生成仍需顺序计算并反复读取权重,解码延迟高。已有流水线自推测解码方法从早期循环预测中起草 token,并在同一批次内让不同循环深度的 token 异步推进,但未充分利用早期循环特征做更长前瞻,且仍会花算力验证错误的推测分支。LoopAhead 让轻量前瞻起草器用早期循环状态预测多个未来 token,深层循环反馈决定哪些草稿分支继续、哪些暂停,最终由全深度模型做 token 决策。论文称在保持目标模型输出分布的前提下,相对自回归解码最高取得 7.82 倍解码加速。

全文梳理

AI 摘要依据下方信源原文自动整理,非原文转载

循环模型的解码瓶颈

循环 Transformer 的思路是让同一批 Transformer 层被反复调用,以此在参数量受限的情况下获得更强的表达能力,参数效率因此提升。但论文指出,这种结构在生成阶段并没有摆脱逐 token 的顺序计算:每生成一个 token 都要再次访问模型权重,重复的权重读取与串行推进叠加,使解码延迟居高不下。这一矛盾正是该工作要处理的问题——参数省下来了,推理速度却成为新的代价。论文把降低这一延迟作为目标,而不是去改动循环结构本身的表达能力。

已有自推测方案的不足

论文回顾了此前的流水线自推测解码方法。这类方法从早期循环的预测结果中起草 token,并让处于不同循环深度的 token 在同一批次里异步向前推进,从而在一定程度上压缩等待时间。论文认为它们存在两点未做足的地方:一是没有把早期循环特征充分用于更长的前瞻,也就是没有尽可能多地提前猜出后续 token;二是仍然会把算力花在验证那些最终被证明是错误的推测分支上。这两点共同限制了加速上限,也成为 LoopAhead 设计的出发点。

分工式推测解码设计

LoopAhead 的做法是给不同循环深度分配互补的角色。一个轻量的前瞻起草器(LAD)利用早期循环状态一次性预测多个未来 token,把前瞻长度拉长;更深循环产生的反馈则用来判断哪些草稿分支值得继续推进、哪些应当暂停,避免在明显无望的分支上继续投入;最后由全深度模型给出最终的 token 决策,保证结果由完整模型拍板。论文强调,这一流程在取得加速的同时保持了目标模型的输出分布,即不改变原本会生成的结果分布。

加速幅度与适用对象

论文给出的结果是,LoopAhead 相对自回归解码最高可实现 7.82 倍的解码加速,同时保持目标模型的输出分布不变。这一数字是该工作最直接的可比指标,衡量的是解码阶段的速度提升,而非训练或其他环节。论文将其定位为面向循环 Transformer 的推测解码框架,因此其适用对象是采用循环层复用结构的语言模型,而非所有 Transformer 变体。论文同时标注了 cs.AI 与 cs.LG 两个方向,并提供了 PDF、HTML 实验版本与音频摘要等访问入口。

为什么值得看

循环 Transformer 推理慢是落地瓶颈,该工作给出保持输出分布的提速方案,并公开了具体加速比。

算力论文

信源1 家

  1. [1]arXiv cs.AI一手信源LoopAhead: Speculating Ahead for Faster Decoding in Looped Language Models

关键事实

  • 论文标题为 LoopAhead: Speculating Ahead for Faster Decoding in Looped Language Models,提交于 2026 年 10 月 15 日,编号 arXiv:2610.215…[1]

  • 循环 Transformer 反复应用同一批 Transformer 层以提升参数效率,但逐 token 生成仍涉及顺序计算与反复访问模型权重,解码延迟高[1]

  • 已有流水线自推测解码方法从早期循环预测起草 token,并让不同循环深度的 token 在同一批次内异步推进,但未充分利用早期循环特征做更长前瞻,且仍会验证错误的推测分支[1]

  • LoopAhead 的轻量前瞻起草器(LAD)用早期循环状态预测多个未来 token,深层循环反馈决定草稿分支继续或暂停,全深度模型做最终 token 决策[1]

  • LoopAhead 在保持目标模型输出分布的同时,相对自回归解码最高实现 7.82 倍解码加速[1]

相关 · 论文

本页内容由 AI 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。