STKAN

STKAN架构利用Kolmogorov-Arnold网络增强时空预测能力

研究人员推出了一种新颖的时空预测架构STKAN，该架构集成了泰勒多项式Kolmogorov-Arnold网络模块。该方法旨在改进对复杂现实世界数据的建模，例如同时表现出空间相关性和非线性时间动态的交通模式。STKAN通过软节点分组分配机制构建空间表示，然后对时间依赖性进行建模，并纳入自注意力层以实现长距离交互。在五个交通预测基准上的实验表明，STKAN表现具有竞争力，并且优于基于MLP的变体。