Dec, 2023

稀疏交叉注意力图卷积网络与辅助信息感知的交通流预测

TL;DR提出了一种名为 AIMSAN 的深度编码器 - 解码器模型,包含了一个辅助信息感知模块 (AIM) 和基于稀疏交叉注意力的图卷积网络 (SAN),在交通数据中挖掘丰富的时空动态,并考虑并利用交通节点的空间稀疏性以减少二次计算复杂度。实验证明,该方法在各种性能指标上胜过其他算法,同时平均节省了 35.74%的 GPU 内存使用量、42.25%的训练时间和 45.51%的验证时间。