3 个月前

时空图混合变换器用于交通预测

时空图混合变换器用于交通预测

摘要

交通预测在智能交通系统(ITS)中具有重要意义。由于交通行为本身具有高度复杂性,且交通数据具有非欧几里得特性,实现精准的交通预测面临巨大挑战。尽管以往研究已关注不同节点之间的关联关系,但大多数方法仍依赖于静态表征,难以捕捉节点间随时间动态演变的交互关系。此外,先前的研究多采用基于循环神经网络(RNN)的模型来建模时间依赖性。虽然RNN在预测任务中广受青睐,但其通常具有较高的内存消耗,且训练速度较慢。近年来,部分研究开始引入相似性算法,以更有效地刻画节点之间的内在关联。然而,据我们所知,目前尚无研究系统探讨节点间动态交互对交通预测的贡献。

基准测试

基准方法指标
traffic-prediction-on-metr-laSTGM
12 steps MAE: 3.229
12 steps MAPE: 9.39
12 steps RMSE: 7.099
MAE @ 12 step: 3.229
traffic-prediction-on-pems-baySTGM
MAE @ 12 step: 1.857
RMSE : 4.369
traffic-prediction-on-pemsd7-mSTGM
12 steps MAE: 3.002
12 steps MAPE: 8.01
12 steps RMSE: 6.331

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
时空图混合变换器用于交通预测 | 论文 | HyperAI超神经