arXiv:2601.09495cs.LG2026-01被引 2

提出可并行计算的新型循环神经网络,兼具持久记忆与高效推理能力。

Parallelizable memory recurrent units

  • 利用多稳态机制实现持久记忆,同时避免瞬态动态影响效率
  • 设计双稳态记忆单元(BMRU),支持并行扫描算法,训练推理更高效
  • 适用于长序列任务,可与状态空间模型结合,兼顾记忆与速度

随着大规模并行计算单元的出现,并行化成为新序列模型的重要特性。Transformer 架构因可在训练时并行处理序列长度而兴起,但生成阶段效率低,需重复处理所有历史步。近期状态空间模型(SSMs)作为替代方案,虽具备高效训练和并行性,却受限于单稳态,无法实现持久记忆。本文提出一类新式循环神经网络——记忆循环单元(MRUs),融合非线性 RNN 的持久记忆能力与 SSM 的并行计算优势。通过多稳态机制实现长期信息保持,同时消除瞬态动态以提升效率。进一步推导出具体实现:双稳态记忆循环单元(BMRU),其兼容并行扫描算法。实验表明,BMRU 在长依赖任务中表现优异,且可与 SSM 结合构建兼具瞬态动态、持久记忆与并行性的混合网络。

原文摘要 · Abstract (English)

With the emergence of massively parallel processing units, parallelization has become a desirable property for new sequence models. The ability to parallelize the processing of sequences with respect to the sequence length during training is one of the main factors behind the uprising of the Transformer architecture. However, Transformers lack efficiency at sequence generation, as they need to reprocess all past timesteps at every generation step. Recently, state-space models (SSMs) emerged as a more efficient alternative. These new kinds of recurrent neural networks (RNNs) keep the efficient update of the RNNs while gaining parallelization by getting rid of nonlinear dynamics (or recurrence). SSMs can reach state-of-the art performance through the efficient training of potentially very large networks, but still suffer from limited representation capabilities. In particular, SSMs cannot exhibit persistent memory, or the capacity of retaining information for an infinite duration, because of their monostability. In this paper, we introduce a new family of RNNs, the memory recurrent units (MRUs), that combine the persistent memory capabilities of nonlinear RNNs with the parallelizable computations of SSMs. These units leverage multistability as a source of persistent memory, while getting rid of transient dynamics for efficient computations. We then derive a specific implementation as proof-of-concept: the bistable memory recurrent unit (BMRU). This new RNN is compatible with the parallel scan algorithm. We show that BMRU achieves good results in tasks with long-term dependencies, and can be combined with state-space models to create hybrid networks that are parallelizable and have transient dynamics as well as persistent memory.

循环神经网络并行计算持久记忆状态空间模型

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