arXiv:2603.03612cs.LGcs.CC2026-03被引 6

解释线性RNN为何比传统RNN更易并行,揭示其理论优势。

Why Are Linear RNNs More Parallelizable?

  • 将线性RNN视为对数深度算术电路,接近Transformer的并行复杂度
  • 非线性RNN可解L-完全问题,存在根本性并行障碍
  • 不同线性RNN变体表达能力有差异,适合不同任务场景

学界日益关注线性RNN(LRNNs)作为语言模型,因其表达力强且易于并行化。尽管已有研究证明LRNNs在表达力上优于Transformer,但尚不清楚为何LRNNs而非传统非线性RNNs能像Transformer一样高效并行。本文通过建立RNN类型与标准复杂度类之间的紧密联系,发现LRNNs可被视为对数深度(有界扇入)算术电路,其深度仅略高于Transformer所对应的对数深度布尔电路。此外,我们证明非线性RNN可求解$ℏ$-完全问题(甚至在多项式精度下可解$ℎ$-完全问题),揭示了其难以高效并行的根本障碍。我们的理论还区分了近期流行的几种LRNN变体:置换-对角型LRNN为$ℋ℁^1$-完全,而对角加低秩型LRNN更强大($ℎ℁^1$-完全)。通过关联每种RNN与其可模拟的自动机模型,我们揭示了非线性RNN与各类线性RNN间的根本权衡,为设计兼具表达力与并行性的大模型架构提供了理论基础。

原文摘要 · Abstract (English)

The community is increasingly exploring linear RNNs (LRNNs) as language models, motivated by their expressive power and parallelizability. While prior work establishes the expressivity benefits of LRNNs over transformers, it is unclear what makes LRNNs -- but not traditional, nonlinear RNNs -- as easy to parallelize in practice as transformers. We answer this question by providing a tight connection between types of RNNs and standard complexity classes. We show that LRNNs can be viewed as log-depth (bounded fan-in) arithmetic circuits, which represents only a slight depth overhead relative to log-depth boolean circuits that transformers admit. Furthermore, we show that nonlinear RNNs can solve $\mathsf{L}$-complete problems (and even $\mathsf{P}$-complete ones, under polynomial precision), revealing a fundamental barrier to parallelizing them as efficiently as transformers. Our theory also identifies fine-grained expressivity differences between recent popular LRNN variants: permutation-diagonal LRNNs are $\mathsf{NC}^1$-complete whereas diagonal-plus-low-rank LRNNs are more expressive ($\mathsf{PNC}^1$-complete). We provide further insight by associating each type of RNN with a corresponding automata-theoretic model that it can simulate. Together, our results reveal fundamental tradeoffs between nonlinear RNNs and different variants of LRNNs, providing a foundation for designing LLM architectures that achieve an optimal balance between expressivity and parallelism.

RNN并行计算复杂度理论模型架构

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