计算机应用

作者：张永凯, 武志昊, 林友芳, 赵苡积

单位：1.北京交通大学 计算机与信息技术学院，北京 100044；2.交通数据分析与挖掘北京市重点实验室（北京交通大学），北京 100044；3.中国民用航空局 民航旅客服务智能化应用技术重点实验室，北京 100105

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

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

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

来源：2021年第12期

《计算机应用》期刊编辑部

查看计算机应用杂志2021年第12期