跳到主要内容
AI News HubLIVE
站内改写2 分钟阅读

上下文就绪Transformer:一种新型循环神经网络架构

文章摘要

该论文提出上下文就绪Transformer,通过修正网络在令牌进入变换器块之前预上下文化,将顺序推理变为循环神经网络,训练时展开修正过程K次并行处理。实验表明,D=5模型在A100上生成速度比12层Transformer快1.7倍且性能更优,单层模型(D=1)在K=10时优于6层Transformer且推理速度提升2.6倍。该架构在宽表示和长上下文下受益最大,在指针追踪任务中单层模型可解决全部10个组合层次。

来源arXiv Computational Linguistics作者: Mahesh Godavarti
上下文就绪Transformer:一种新型循环神经网络架构
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明
直接读正文

这篇题为《The Context-Ready Transformer》(上下文就绪的Transformer)的论文由Mahesh Godavarti撰写,于2026年6月25日提交至arXiv,编号为arXiv:2606.27538,归类于计算与语言(cs.CL)和人工智能(cs.AI)方向,MSC分类为68T07,ACM分类为I.2.6与I.5.1,并注明为NeurIPS论文、全文共22页。

论文提出了一种名为“上下文就绪Transformer”的全新循环神经网络架构。它的核心构成是一个D层Transformer块,但与标准做法不同的是,每个词元在进入该块之前就已经被预先上下文化。具体来说,在从左到右的生成过程中,一个校正网络会把前一个位置的块输出——即对过往上下文所作的缓存摘要——与当前词元的嵌入向量结合起来。这样一来,词元进入Transformer块时携带的已经是经过上下文化的表示,而不是原始的嵌入向量。由于在顺序推理时这种校正链条会不断传递,该架构实际上转变成了一个循环神经网络。

在训练方面,作者采用了对校正过程展开(unroll)K次的方式,每次都在整条序列上对所有位置进行并行处理,从而在保留循环式推理特性的同时维持训练效率。论文还指出,已有的预训练Transformer可以通过增加一个零初始化的校正前馈网络(FFN)并进行微调,转换成上下文就绪模型,这为该架构的落地提供了一条低成本的迁移路径。

实验部分,作者在两种数据集上、跨不同宽度、深度和块大小进行了评估,所有比较都对照标准Transformer、其他变体以及消融实验展开。结果显示,一个D=5的模型在A100上以1.7倍的生成速度超过了12层Transformer。当K=10时,单层模型(D=1)不仅击败了6层Transformer,还带来2.6倍的推理加速;而且顺序推理的结果与并行的K=10版本在困惑度(PPL)上相差不超过0.01。作者观察到,该架构从宽表示和长上下文中获益最为明显。

在指针追踪(pointer-chasing)任务上,使用BPTT训练的单层模型(D=1)成功解决了全部10个组合层级,而标准Transformer则表现出类似阶梯状的深度依赖,即需要逐层增加深度才能解决更高的组合层级。总体而言,这项工作试图通过让模型在输入阶段就具备上下文感知能力,在不牺牲推理速度的前提下提升计算效率与长程组合能力。

展开要点与分析

文章情报

工程师进阶

要点

  • 上下文就绪Transformer在令牌进入变换器块前进行预上下文化,顺序推理时形成循环神经网络
  • 训练时展开修正过程K次,每步并行处理所有位置,支持从预训练Transformer转换
  • D=5模型性能超越12层Transformer,生成速度快1.7倍;单层模型(D=1)在K=10时超越6层Transformer,推理速度快2.6倍
  • 该架构在宽表示和长上下文下优势明显,在指针追踪任务中单层模型可解决全部10个组合层次

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。