基于生物神经网络的振荡机制,提出高效长序列建模的新模型。
Oscillatory State-Space Models
- 用受迫谐振子系统构建状态空间模型,实现稳定离散化。
- 在50k长度序列上性能接近Mamba和LRU的2倍,且支持长期预测。
- 适用于需要稳定长期建模的时序任务,如金融、医疗时间序列。
我们提出线性振荡状态空间模型(LinOSS),用于高效处理长序列学习。受生物神经网络皮层动态启发,模型基于受迫谐振子系统构建。通过快速关联并行扫描实现时间上的积分离散化,保证了稳定动态。我们证明,仅需非负对角状态矩阵,即可确保动态稳定,这与许多依赖严格参数化的旧模型形成鲜明对比。此外,我们严格证明了LinOSS具有通用性,可任意精度逼近任意连续因果算子。进一步地,其隐式-显式离散化能完美保持底层动力学的时间可逆对称性。这些特性使模型能高效建模长程依赖,同时确保长期预测的稳定与准确。实证结果覆盖中等至超长序列的分类、回归及长期预测任务,表明该模型持续优于当前主流序列模型。值得注意的是,在50,000长度序列建模任务中,其性能接近Mamba和LRU的两倍。
原文摘要 · Abstract (English)
We propose Linear Oscillatory State-Space models (LinOSS) for efficiently learning on long sequences. Inspired by cortical dynamics of biological neural networks, we base our proposed LinOSS model on a system of forced harmonic oscillators. A stable discretization, integrated over time using fast associative parallel scans, yields the proposed state-space model. We prove that LinOSS produces stable dynamics only requiring nonnegative diagonal state matrix. This is in stark contrast to many previous state-space models relying heavily on restrictive parameterizations. Moreover, we rigorously show that LinOSS is universal, i.e., it can approximate any continuous and causal operator mapping between time-varying functions, to desired accuracy. In addition, we show that an implicit-explicit discretization of LinOSS perfectly conserves the symmetry of time reversibility of the underlying dynamics. Together, these properties enable efficient modeling of long-range interactions, while ensuring stable and accurate long-horizon forecasting. Finally, our empirical results, spanning a wide range of time-series tasks from mid-range to very long-range classification and regression, as well as long-horizon forecasting, demonstrate that our proposed LinOSS model consistently outperforms state-of-the-art sequence models. Notably, LinOSS outperforms Mamba and LRU by nearly 2x on a sequence modeling task with sequences of length 50k.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。