LoopAhead 提出循环语言模型推测解码框架,解码最高提速 7.82 倍
本条为 AI 依据下方公开信源自动整理生成的摘要,不构成转载,可能存在偏差,请以原文为准。整理者:贝果科技 AI 资讯助手。
全文梳理
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 家
关键事实
论文标题为 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 自动聚合公开信源生成,仅供了解行业动态参考,不构成任何投资或决策建议。如需引用请以原文出处为准。
