IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v10y2022i19p3507-d925481.html
   My bibliography  Save this article

An Attention and Wavelet Based Spatial-Temporal Graph Neural Network for Traffic Flow and Speed Prediction

Author

Listed:
  • Shihao Zhao

    (School of Computer Science and Mathematics, Fujian University of Technology, Fuzhou 350118, China)

  • Shuli Xing

    (School of Computer Science and Mathematics, Fujian University of Technology, Fuzhou 350118, China)

  • Guojun Mao

    (School of Computer Science and Mathematics, Fujian University of Technology, Fuzhou 350118, China
    Fujian Provincial Key Lab of Big Data Mining and Applications, Fuzhou 350118, China)

Abstract

Traffic flow prediction is essential to the intelligent transportation system (ITS). However, due to the complex spatial-temporal dependence of traffic flow data, it is insufficient in the extraction of local and global spatial-temporal correlations for the previous process on road network and traffic flow modeling. This paper proposes an attention and wavelet-based spatial-temporal graph neural network for traffic flow and speed prediction (STAGWNN). It integrated attention and graph wavelet neural networks to capture local and global spatial information. Meanwhile, we stacked a gated temporal convolutional network (gated TCN) with a temporal attention mechanism to extract the time series information. The experiment was carried out on real public transportation datasets: PEMS-BAY and PEMSD7(M). The comparison results showed that our proposed model outperformed baseline networks on these datasets, which indicated that STAGWNN could better capture the spatial-temporal correlation information.

Suggested Citation

  • Shihao Zhao & Shuli Xing & Guojun Mao, 2022. "An Attention and Wavelet Based Spatial-Temporal Graph Neural Network for Traffic Flow and Speed Prediction," Mathematics, MDPI, vol. 10(19), pages 1-15, September.
    • Handle: RePEc:gam:jmathe:v:10:y:2022:i:19:p:3507-:d:925481
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/10/19/3507/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/2227-7390/10/19/3507/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Cai, Lingru & Zhang, Zhanchang & Yang, Junjie & Yu, Yidan & Zhou, Teng & Qin, Jing, 2019. "A noise-immune Kalman filter for short-term traffic flow forecasting," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 536(C).
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Ke Zhang & Yaming Guo, 2023. "Attention-Based Residual Dilated Network for Traffic Accident Prediction," Mathematics, MDPI, vol. 11(9), pages 1-15, April.
    2. Anton Agafonov & Alexander Yumaganov & Vladislav Myasnikov, 2023. "Cooperative Control for Signalized Intersections in Intelligent Connected Vehicle Environments," Mathematics, MDPI, vol. 11(6), pages 1-19, March.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Fang, Weiwei & Zhuo, Wenhao & Yan, Jingwen & Song, Youyi & Jiang, Dazhi & Zhou, Teng, 2022. "Attention meets long short-term memory: A deep learning network for traffic flow forecasting," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 587(C).
    2. Chen, Xinqiang & Chen, Huixing & Yang, Yongsheng & Wu, Huafeng & Zhang, Wenhui & Zhao, Jiansen & Xiong, Yong, 2021. "Traffic flow prediction by an ensemble framework with data denoising and deep learning model," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 565(C).
    3. Liu, Yang & Song, Yaolun & Zhang, Yan & Liao, Zhifang, 2022. "WT-2DCNN: A convolutional neural network traffic flow prediction model based on wavelet reconstruction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 603(C).
    4. Tian, Jing & Song, Xianmin & Tao, Pengfei & Liang, Jiahui, 2022. "Pattern-adaptive generative adversarial network with sparse data for traffic state estimation," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 608(P1).
    5. Zhanzhong Wang & Ruijuan Chu & Minghang Zhang & Xiaochao Wang & Siliang Luan, 2020. "An Improved Hybrid Highway Traffic Flow Prediction Model Based on Machine Learning," Sustainability, MDPI, vol. 12(20), pages 1-22, October.
    6. Chih-Yao Chang & Kuo-Ping Lin, 2020. "Developing Support Vector Machine with New Fuzzy Selection for the Infringement of a Patent Rights Problem," Mathematics, MDPI, vol. 8(8), pages 1-16, August.
    7. Wenguang Chai & Yuexin Zheng & Lin Tian & Jing Qin & Teng Zhou, 2023. "GA-KELM: Genetic-Algorithm-Improved Kernel Extreme Learning Machine for Traffic Flow Forecasting," Mathematics, MDPI, vol. 11(16), pages 1-15, August.
    8. Yulong Pei & Songmin Ran & Wanjiao Wang & Chuntong Dong, 2023. "Bus-Passenger-Flow Prediction Model Based on WPD, Attention Mechanism, and Bi-LSTM," Sustainability, MDPI, vol. 15(20), pages 1-20, October.
    9. Huang, Haichao & Chen, Jingya & Sun, Rui & Wang, Shuang, 2022. "Short-term traffic prediction based on time series decomposition," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 585(C).

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:10:y:2022:i:19:p:3507-:d:925481. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.