通过多重霍夫变换提升线性RNN状态追踪能力
DeltaProduct: Improving State-Tracking in Linear RNNs via Householder Products
- 用多步梯度更新替代单步,构建可调的高阶状态转移矩阵
- 实验显示在状态追踪和语言建模上均优于DeltaNet,长序列外推更强
- 适合需要高效且强记忆能力的序列建模任务
线性循环神经网络(linear RNNs)作为Transformer的高效替代方案,具备训练快速、推理线性时间的优势。然而,现有架构在表达能力和效率间存在根本权衡,源于状态转移矩阵的结构限制。采用对角矩阵的模型如Mamba、GLA或mLSTM虽运行快,但表达能力有限。近期的DeltaNet和RWKV-7引入了对角加秩-1结构,实现同时的令牌与通道混合,提升了关联回忆能力,并在允许负特征值时改善了状态追踪。基于将DeltaNet的递归解释为每令牌一次在线梯度下降,本文提出DeltaProduct,每令牌执行n_h步更新。这自然导出对角加秩-n_h的状态转移矩阵,由n_h个广义霍夫变换的乘积构成,提供可调节的表达力与效率平衡机制。我们详细分析了有限精度下DeltaProduct的状态追踪能力,表明其随n_h增大而提升。大量实验表明,DeltaProduct在状态追踪与语言建模上均优于DeltaNet,且显著提升长度外推能力。
原文摘要 · Abstract (English)
Linear Recurrent Neural Networks (linear RNNs) have emerged as competitive alternatives to Transformers for sequence modeling, offering efficient training and linear-time inference. However, existing architectures face a fundamental trade-off between expressivity and efficiency, dictated by the structure of their state-transition matrices. Diagonal matrices, used in models such as Mamba, GLA, or mLSTM, yield fast runtime but have limited expressivity. To address this, recent architectures such as DeltaNet and RWKV-7 adopted a diagonal plus rank--1 structure, which allows simultaneous token and channel mixing, improving associative recall and, as recently shown, state-tracking when allowing state-transition matrices to have negative eigenvalues. Building on the interpretation of DeltaNet's recurrence as performing one step of online gradient descent per token on an associative recall loss, we introduce DeltaProduct, which instead takes multiple ($n_h$) steps per token. This naturally leads to diagonal plus rank--$n_h$ state-transition matrices, formed as products of $n_h$ generalized Householder transformations, providing a tunable mechanism to balance expressivity and efficiency. We provide a detailed theoretical characterization of the state-tracking capability of DeltaProduct in finite precision, showing how it improves by increasing $n_h$. Our extensive experiments demonstrate that DeltaProduct outperforms DeltaNet in both state-tracking and language modeling, while also showing significantly improved length extrapolation capabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。