首页 | 官方网站   微博 | 高级检索  
     

面向交通流量预测的时空超关系图卷积网络
引用本文:张永凯,武志昊,林友芳,赵苡积.面向交通流量预测的时空超关系图卷积网络[J].计算机应用,2021,41(12):3578-3584.
作者姓名:张永凯  武志昊  林友芳  赵苡积
作者单位:北京交通大学 计算机与信息技术学院,北京 100044
交通数据分析与挖掘北京市重点实验室(北京交通大学),北京 100044
中国民用航空局 民航旅客服务智能化应用技术重点实验室,北京 100105
基金项目:中央高校基本科研业务费专项资金资助项目(2019JBM023)
摘    要:交通流量预测是智能交通系统中的重要研究课题,然而,交通对象(如站点、传感器)之间存在的复杂局部时空关系使得这项研究颇具挑战。尽管以往的一些研究将流量预测问题转化为一个时空图预测问题从而取得了较大的进展,但是它们忽略了交通对象们跨时空维度的直接关联性。目前仍缺乏一种全面建模局部时空关系的方法。针对这一问题,首先提出一种新颖的时空超图建模方案,通过构造一种时空超关系来全面地建模复杂的局部时空关系;然后提出一种时空超关系图卷积网络(STHGCN)预测模型来捕获这些关系用于交通流量预测。在四个公开交通数据集上进行了大量对比实验,结果表明,相比ASTGCN、时空同步图卷积网络(STSGCN)等时空预测模型,STHGCN在均方根误差(RMSE)、平均绝对误差(MAE)、平均绝对百分比误差(MAPE)这三个评价指标上均取得了更优的结果,不同模型运行时间的对比结果也表明,STHGCN有着更高的推理速度。

关 键 词:交通流量预测  局部时空关系  时空图预测  超图  时空超关系  
收稿时间:2021-05-12
修稿时间:2021-06-15

Spatio-temporal hyper-relationship graph convolutional network for traffic flow forecasting
ZHANG Yongkai,WU Zhihao,LIN Youfang,ZHAO Yiji.Spatio-temporal hyper-relationship graph convolutional network for traffic flow forecasting[J].journal of Computer Applications,2021,41(12):3578-3584.
Authors:ZHANG Yongkai  WU Zhihao  LIN Youfang  ZHAO Yiji
Affiliation:School of Computer and Information Technology,Beijing Jiaotong University,Beijing 100044,China
Beijing Key Lab of Traffic Data Analysis and Mining (Beijing Jiaotong University),Beijing 100044,China
Key Laboratory of Intelligent Application Technology for Civil Aviation Passenger Services,Civil Aviation Administration of China,Beijing 100105,China
Abstract:Traffic flow forecasting is an important research topic for the intelligent transportation system, however, this research is very challenging because of the complex local spatio-temporal relationships among traffic objects such as stations and sensors. Although some previous studies have made great progress by transforming the traffic flow forecasting problem into a spatio-temporal graph forecasting problem, in which the direct correlations across spatio-temporal dimensions among traffic objects are ignored. At present, there is still lack of a comprehensive modeling approach for the local spatio-temporal relationships. A novel spatio-temporal hypergraph modeling scheme was first proposed to address this problem by constructing a kind of spatio-temporal hyper-relationships to comprehensively model the complex local spatio-temporal relationships. Then, a Spatio-Temporal Hyper-Relationship Graph Convolutional Network (STHGCN) forecasting model was proposed to capture these relationships for traffic flow forecasting. Extensive comparative experiments were conducted on four public traffic datasets. Experimental results show that compared with the spatio-temporal forecasting models such as Attention based Spatial-Temporal Graph Convolutional Network (ASTGCN) and Spatial-Temporal Synchronous Graph Convolutional Network (STSGCN), STHGCN achieves better results in Root Mean Square Error (RMSE), Mean Absolute Error (MAE) and Mean Absolute Percentage Error (MAPE); and the comparison of the running time of different models also shows that STHGCN has higher inference speed.
Keywords:traffic flow forecasting  local spatio-temporal relationship  spatio-temporal graph forecasting  hypergraph  spatio-temporal hyper-relationship  
本文献已被 万方数据 等数据库收录!
点击此处可从《计算机应用》浏览原始摘要信息
点击此处可从《计算机应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司    京ICP备09084417号-23

京公网安备 11010802026262号