arXiv:2505.21024cs.LGcs.CL2025-05NeurIPS被引 13

pause tokens让浅层Transformer表达能力显著提升

Pause Tokens Strictly Increase the Expressivity of Constant-Depth Transformers

  • 在固定深度的Transformer中引入停顿符号,可严格增强其计算能力
  • 有停顿符号时,模型能表达完整AC⁰类函数,无则只能表达子集
  • 适合研究模型表达力、推理机制与注意力架构的学者参考

停顿符号(如'...')虽简单,却在语言和数学任务中持续提升Transformer性能,但其理论作用尚未明确。本文首次提供形式化分离结果:在常数深度、对数宽度的Transformer中,加入多项式数量的停顿符号可严格提升其计算表达能力。在有界精度下,无停顿符号的Transformer仅能计算AC⁰函数的真子集;加入停顿符号后,可表达整个AC⁰类。对于对数精度模型,停顿符号使表达能力达到TC⁰,匹配已知上界。实验证明,两层因果掩码Transformer在引入停顿符号后可学习奇偶性函数,而无停顿符号时无法学习。本研究为先前经验发现提供严谨理论解释,阐明停顿符号与宽度、深度、精度的交互机制,将其定位为一种独立于思维链提示的增强推理能力的新机制。

原文摘要 · Abstract (English)

Pause tokens, simple filler symbols such as "...", consistently improve Transformer performance on both language and mathematical tasks, yet their theoretical effect remains unexplained. We provide the first formal separation result, proving that adding pause tokens to constant-depth, logarithmic-width Transformers strictly increases their computational expressivity. With bounded-precision activations, Transformers without pause tokens compute only a strict subset of $\mathsf{AC}^0$ functions, while adding a polynomial number of pause tokens allows them to express the entire class. For logarithmic-precision Transformers, we show that adding pause tokens achieves expressivity equivalent to $\mathsf{TC}^0$, matching known upper bounds. Empirically, we demonstrate that two-layer causally masked Transformers can learn parity when supplied with pause tokens, a function that they appear unable to learn without them. Our results provide a rigorous theoretical explanation for prior empirical findings, clarify how pause tokens interact with width, depth, and numeric precision, and position them as a distinct mechanism, complementary to chain-of-thought prompting, for enhancing Transformer reasoning.

Transformer表达力停顿符号理论分析

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