Simplified State Space Layers for Sequence Modeling

论文地址：

• https://arxiv.org/abs/2208.04933

代码地址：

• https://github.com/lindermanlab/S5

整体思路以及计算方式

对State Space里面的矩阵做可对角化假设，然后使用PrefixSum算法代替FFT。

简评

可以尝试使用PrefixSum算法。