《计算机应用》唯一官方网站 ›› 2021, Vol. 41 ›› Issue (12): 3578-3584.DOI: 10.11772/j.issn.1001-9081.2021060956

• 第十八届中国机器学习会议(CCML 2021) • 上一篇    

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

张永凯1,2, 武志昊1,2,3, 林友芳1,2,3, 赵苡积1,2()   

  1. 1.北京交通大学 计算机与信息技术学院,北京 100044
    2.交通数据分析与挖掘北京市重点实验室(北京交通大学),北京 100044
    3.中国民用航空局 民航旅客服务智能化应用技术重点实验室,北京 100105
  • 收稿日期:2021-05-12 修回日期:2021-06-15 接受日期:2021-06-29 发布日期:2021-12-28 出版日期:2021-12-10
  • 通讯作者: 赵苡积
  • 作者简介:张永凯(1997—),男,河南信阳人,硕士研究生,主要研究方向:交通数据挖掘、机器学习
    武志昊(1984—),男,山西大同人,副教授,博士,主要研究方向:社交网络分析、数据挖掘、机器学习
    林友芳(1971—),男,福建武平人,教授,博士,主要研究方向:智能系统、复杂网络、交通数据挖掘;
  • 基金资助:
    中央高校基本科研业务费专项资金资助项目(2019JBM023)

Spatio-temporal hyper-relationship graph convolutional network for traffic flow forecasting

Yongkai ZHANG1,2, Zhihao WU1,2,3, Youfang LIN1,2,3, Yiji ZHAO1,2()   

  1. 1.School of Computer and Information Technology,Beijing Jiaotong University,Beijing 100044,China
    2.Beijing Key Lab of Traffic Data Analysis and Mining (Beijing Jiaotong University),Beijing 100044,China
    3.Key Laboratory of Intelligent Application Technology for Civil Aviation Passenger Services,Civil Aviation Administration of China,Beijing 100105,China
  • Received:2021-05-12 Revised:2021-06-15 Accepted:2021-06-29 Online:2021-12-28 Published:2021-12-10
  • Contact: Yiji ZHAO
  • About author:ZHANG Yongkai, born in 1997, M. S. candidate. His research interests include traffic data mining, machine learning.
    WU Zhihao, born in 1984, Ph. D., associate professor. His research interests include social network analysis, data mining, machine learning.
    LIN Youfang, born in 1971, Ph. D., professor. His research interests include intelligent system, complex network, traffic data mining.
  • Supported by:
    the Fundamental Research Funds for the Central Universities(2019JBM023)

摘要:

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

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

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.

Key words: traffic flow forecasting, local spatio-temporal relationship, spatio-temporal graph forecasting, hypergraph, spatio-temporal hyper-relationship

中图分类号: