跳到正文
arXiv cs.AI· Wenyu Du, Stephen Chung·· 13 小时前AI 评分31

CopyLoop:通过上下文复制与浅层筛选加速循环 Transformer 解码

CopyLoop: Fast Looped Transformer Decoding via Context Copying and Shallow Screening

AI 导读

CopyLoop 是一种免训练解码算法,用重复上下文提出多 token 后缀,并在浅层循环深度提前筛选错误分支,同时让新块与仍在向全深度验证推进的旧块并发处理。

正文 · AI 翻译

查看 PDF

摘要:循环 Transformer 通过重复应用共享模块来提升计算容量,但其循环深度使得传统的全深度解码代价高昂。现有的投机式流水线通过在不同深度处理连续 token 位置来加速生成,但通常受限于顺序解码的开销。我们提出 CopyLoop,一种无需训练的解码算法,它将低成本的上下文复制提议与浅层筛选和连续循环调度相结合。CopyLoop 并非逐步生成,而是从重复的上下文中提出多 token 后缀。随后,它在浅层循环深度上对这些后缀进行早期评估,以高效过滤掉错误分支,并在较早的块仍在向全深度验证推进时,并发处理更新的块。我们在来自 Ouro、Raven 和 Huginn 系列的六个模型检查点上评估了 CopyLoop,涵盖代码生成和多步数学推理。在 Base 模型上,CopyLoop 相比全深度自回归解码实现了 2.98 倍至 10.70 倍的解码加速,在 Thinking 模型上实现了 2.74 倍至 3.52 倍的加速,在各评估任务上始终优于或接近所有竞争基线。此外,消融研究证实了浅层筛选与连续调度之间的强协同效应,而我们的分析表明,CopyLoop 仅需轻量级的模型校准,并在具有可复用上下文的较长生成中带来更大的加速。
评论: 预印本
主题: 计算与语言 (cs.CL);人工智能 (cs.AI)
引用为: arXiv:2610.38822 [cs.CL]
  (或 arXiv:2610.38822v1 [cs.CL] 为此版本)
  https://doi.org/10.48550/arXiv.2610.38822

由 DataCite 颁发的 arXiv DOI

提交历史

来自:Wenyu Du [查看电子邮件]
[v1] 2026 年 10 月 28 日星期三 20:16:54 UTC(105 KB)

来源:arXiv cs.AI · arxiv.org