arXiv:2507.08567cs.LG2025-07被引 8

AbbIE通过递归迭代提升序列建模效率,测试时可动态调算力。

AbbIE: Autoregressive Block-Based Iterative Encoder for Efficient Sequence Modeling

  • 在隐空间递归迭代,仅需2次训练即可泛化到任意迭代长度。
  • 零样本上下文学习任务提升12%,语言困惑度降低5%。
  • 适合追求高效推理与动态计算的模型部署场景。

我们提出自回归块式迭代编码器(AbbIE),一种新型的仅编码器Transformer递归推广结构,其困惑度优于标准Transformer,且可在测试时动态调节计算资源。该方法在隐空间执行迭代,无需专用数据集或训练协议。实验表明,仅用2次训练迭代,AbbIE即可在测试时向上泛化至任意迭代长度,显著优于其他迭代方法。基于任务复杂度动态调整计算量,使零样本上下文学习任务性能最高提升12%,语言困惑度改善达5%。所有评估均在最大3.5亿参数的模型上进行,为Transformer性能扩展开辟新路径。

原文摘要 · Abstract (English)

We introduce the Autoregressive Block-Based Iterative Encoder (AbbIE), a novel recursive generalization of the encoder-only Transformer architecture, which achieves better perplexity than a standard Transformer and allows for the dynamic scaling of compute resources at test time. This simple, recursive approach is a complement to scaling large language model (LLM) performance through parameter and token counts. AbbIE performs its iterations in latent space, but unlike latent reasoning models, does not require a specialized dataset or training protocol. We show that AbbIE upward generalizes (ability to generalize to arbitrary iteration lengths) at test time by only using 2 iterations during train time, far outperforming alternative iterative methods. AbbIE's ability to scale its computational expenditure based on the complexity of the task gives it an up to \textbf{12\%} improvement in zero-shot in-context learning tasks versus other iterative and standard methods and up to 5\% improvement in language perplexity. The results from this study open a new avenue to Transformer performance scaling. We perform all of our evaluations on model sizes up to 350M parameters.

序列建模递归结构动态计算Transformer

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。