arXiv:2602.12021cs.LG2026-02被引 1

提升线性循环网络的表达能力,同时保持高效计算。

Improved state mixing in higher-order and block diagonal linear recurrent networks

  • 提出高阶和分块对角两种结构化线性循环单元,增强状态跨时间与通道的混合。
  • 分块对角架构在序列建模中性能媲美甚至超越Mamba和LSTM,高阶架构压缩任务最省参数。
  • 通过归一化稳定训练,支持更大窗口或分块规模,适合长序列高效建模场景。

线性循环网络(LRNNs)和线性状态空间模型(SSMs)在长序列建模中具有计算与内存效率优势,但其对角状态转移限制了表达能力。而密集非线性架构(如LSTM)虽表达能力强,但计算成本高。本文通过增强时间与通道间的状态混合,在保持效率的同时提升LRNN表达力。提出两种结构化LRNN:(i) 高阶线性循环单元(H-LRU),将一阶递推推广至高阶,融合多个历史状态;(ii) 分块对角线性循环单元(BD-LRU),实现块内通道的稠密混合。采用逐通道(H-LRU)或逐行(BD-LRU)L1归一化的选择门,稳定训练并支持扩展窗口/块大小。并行扫描实现使吞吐量与对角LRNN相当。在合成序列建模任务中,BD-LRU性能匹配或超过线性SSMs(Mamba)、低秩LRNN(DeltaNet)及LSTM基线;H-LRU在压缩任务中参数效率最高。结果表明,状态混合结构而非宽度本身决定表达力,为缩小线性序列模型的效率-表达力差距提供可行路径。

原文摘要 · Abstract (English)

Linear recurrent networks (LRNNs) and linear state space models (SSMs) promise computational and memory efficiency on long-sequence modeling tasks, yet their diagonal state transitions limit expressivity. Dense and nonlinear architectures (e.g., LSTMs) on the other hand are provably more expressive, but computationally costly. Here, we explore how expressivity in LRNNs can be increased via richer state mixing across time and channels while maintaining competitive efficiency. Specifically, we introduce two structured LRNN architectures: (i) Higher-order Linear Recurrent Units (H-LRU), which generalize first-order recurrence to higher order, mixing multiple past states, and (ii) Block-Diagonal LRUs (BD-LRU), which enable dense intra-block channel mixing. Per-channel (H-LRU) or per-row (BD-LRU) L1-normalization of selective gates stabilizes training and allows for scaling window/block sizes. A parallel-scan implementation of the proposed architectures keeps the throughput competitive with diagonal LRNNs for moderate orders (H-LRU) and block sizes (BD-LRU). In synthetic sequence modeling tasks, the performance of BD-LRU matches or exceeds those of linear SSMs (Mamba), low-rank LRNNs (DeltaNet) and LSTM baselines, while H-LRU is found to be the most parameter-efficient in compression task. In both synthetic sequence modeling and language modeling, our results indicate that the structure of state mixing rather than width alone shapes expressivity of LRNNs, offering a practical route to closing the efficiency-expressivity gap in linear sequence models.

线性循环状态空间序列建模高效模型

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