A dynamical spatial-temporal graph neural network for traffic demand prediction

被引:59
|
作者
Huang, Feihu [1 ]
Yi, Peiyu [1 ]
Wang, Jince [1 ,2 ]
Li, Mengshi [1 ]
Peng, Jian [1 ]
Xiong, Xi [3 ]
机构
[1] Sichuan Univ, Coll Comp Sci, Chengdu 610065, Peoples R China
[2] Shanxi Inst Energy, Dept Comp & Informat Engn, Jinzhong 030600, Peoples R China
[3] Chengdu Univ Informat Technol, Sch Cybersecur, Chengdu 610225, Peoples R China
基金
山西省青年科学基金;
关键词
Traffic demand prediction; Graph neural network; Spatial-temporal dependence; Inhomogeneous Poisson process; TIME-SERIES; ALGORITHM;
D O I
10.1016/j.ins.2022.02.031
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Traffic demand prediction is significant and practical in the resource scheduling of transportation application systems. Meanwhile, it remains a challenging topic due to the complexities of contextual effects and the highly dynamic nature of demand. Many works based on graph neural network (GNN) have recently been proposed to cope with this task. However, most previous studies treat the spatial dependence as a static graph, and their inference mechanism lacks interpretability. To address the issues, a Dynamical Spatial-Temporal Graph Neural Network model (DSTGNN) is proposed in this paper. DSTGNN has two critical phases: (1) Creating a spatial dependence graph. To capture the dynamical relationship, we propose building a spatial graph based on the stability of node's spatial dependence. (2) Inferring intensity. We model the changing demand process using the inhomogeneous Poisson process, which addresses the interpretability issue, and build a spatial-temporal embedding network to infer the intensity. Specifically, the spatial-temporal embedding network integrates the diffusion convolution neural network (DCNN) and a modified transformer. Extensive experiments are carried out on two real data sets, and the experimental results demonstrate that the performance of DSTGNN outperforms the state-of-the-art models on traffic demand prediction. (C) 2022 Elsevier Inc. All rights reserved.
引用
收藏
页码:286 / 304
页数:19
相关论文
共 50 条
  • [1] Spatial-Temporal Dynamic Graph Convolutional Neural Network for Traffic Prediction
    Xiao, Wenjuan
    Wang, Xiaoming
    [J]. IEEE ACCESS, 2023, 11 : 97920 - 97929
  • [2] STAGNN: a spatial-temporal attention graph neural network for network traffic prediction
    Luo, Yonghua
    Ning, Qian
    Chen, Bingcai
    Zhou, Xinzhi
    Huang, Linyu
    [J]. INTERNATIONAL JOURNAL OF COMMUNICATION NETWORKS AND DISTRIBUTED SYSTEMS, 2024, 30 (04) : 413 - 432
  • [3] Graph Spatial-Temporal Transformer Network for Traffic Prediction
    Zhao, Zhenzhen
    Shen, Guojiang
    Wang, Lei
    Kong, Xiangjie
    [J]. BIG DATA RESEARCH, 2024, 36
  • [4] Spatial-Temporal Multiscale Fusion Graph Neural Network for Traffic Flow Prediction
    Hou, Hongxin
    Ning, Nianwen
    Shi, Huaguang
    Zhou, Yi
    [J]. 2022 IEEE 7th International Conference on Intelligent Transportation Engineering, ICITE 2022, 2022, : 272 - 277
  • [5] Traffic Flow Prediction Based on Dynamic Graph Spatial-Temporal Neural Network
    Jiang, Ming
    Liu, Zhiwei
    [J]. MATHEMATICS, 2023, 11 (11)
  • [6] Multi-View Spatial-Temporal Graph Neural Network for Traffic Prediction
    Li, He
    Jin, Duo
    Li, XueJiao
    Huang, HongJie
    Yun, JinPeng
    Huang, LongJi
    [J]. COMPUTER JOURNAL, 2023, 66 (10): : 2393 - 2408
  • [7] Spatial-Temporal Multiscale Fusion Graph Neural Network for Traffic Flow Prediction
    Hou, Hongxin
    Ning, Nianwen
    Shi, Huaguang
    Zhou, Yi
    [J]. 2022 IEEE 7TH INTERNATIONAL CONFERENCE ON INTELLIGENT TRANSPORTATION ENGINEERING, ICITE, 2022, : 272 - 277
  • [8] Adaptive Hybrid Spatial-Temporal Graph Neural Network for Cellular Traffic Prediction
    Wang, Xing
    Yang, Kexin
    Wang, Zhendong
    Feng, Junlan
    Zhu, Lin
    Zhao, Juan
    Deng, Chao
    [J]. ICC 2023-IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS, 2023, : 4026 - 4032
  • [9] Spatial-Temporal Dilated and Graph Convolutional Network for traffic prediction
    Yang, Guoliang
    Wen, Junlin
    Yu, Dinglin
    Zhang, Shuo
    [J]. 2020 CHINESE AUTOMATION CONGRESS (CAC 2020), 2020, : 802 - 806
  • [10] Based Matrix Fusion Spatial-Temporal Graph Neural Network for Traffic Flow Prediction
    Jing, Xin
    Zhu, Hai
    [J]. PROCEEDINGS OF 2023 7TH INTERNATIONAL CONFERENCE ON ELECTRONIC INFORMATION TECHNOLOGY AND COMPUTER ENGINEERING, EITCE 2023, 2023, : 1171 - 1175