面向交通流量预测的时空超关系图卷积网络

摘要：

交通流量预测是智能交通系统中的重要研究课题,然而,交通对象(如站点、传感器)之间存在的复杂局部时空关系使得这项研究颇具挑战.尽管以往的一些研究将流量预测问题转化为一个时空图预测问题从而取得了较大的进展,但是它们忽略了交通对象们跨时空维度的直接关联性.目前仍缺乏一种全面建模局部时空关系的方法.针对这一问题,首先提出一种新颖的时空超图建模方案,通过构造一种时空超关系来全面地建模复杂的局部时空关系;然后提出一种时空超关系图卷积网络(STHGCN)预测模型来捕获这些关系用于交通流量预测.在四个公开交通数据集上进行了大量对比实验,结果表明,相比ASTGCN、时空同步图卷积网络(STSGCN)等时空预测模型,STHGCN在均方根误差(RMSE)、平均绝对误差(MAE)、平均绝对百分比误差(MAPE)这三个评价指标上均取得了更优的结果,不同模型运行时间的对比结果也表明,STHGCN有着更高的推理速度.

关键词：交通流量预测;局部时空关系;时空图预测;超图;时空超关系

所属期刊栏目：41

分类号：TP302.1(计算技术、计算机技术)

资助基金：中央高校基本科研业务费专项资金资助项目2019JBM023

在线出版日期：2022-01-04（万方平台首次上网日期，不代表论文的发表时间）

页数：共7页

页码：3578-3584