IDEAS home Printed from https://ideas.repec.org/a/eee/phsmap/v603y2022ics0378437122005076.html
   My bibliography  Save this article

ST-MGAT:Spatio-temporal multi-head graph attention network for Traffic prediction

Author

Listed:
  • Wang, Bowen
  • Wang, Jingsheng

Abstract

Traffic flow prediction is an important part of the intelligent transportation system, and accurate traffic flow prediction can better perform the data management and control of the urban road network. Traditional methods often ignore the interaction between traffic flow factors and the spatiotemporal dependence of traffic networks. In this paper, a spatiotemporal multi-head graph attention network (ST-MGAT) is proposed for the traffic flow prediction task. In the input layer, multiple traffic flow variables are used as input to learn the nonlinearity and complexity existing in it. In terms of modeling, the structure of linear gating units is transformed using the full volume and applied to correlation learning in the temporal dimension. The spatial dimension features are then captured using a multi-head attention map network. Ablation experiments are designed to verify the use effect of each module. At the same time, several sets of parameter experiments are designed to obtain the best model parameters. The experimental results show that compared with the baselines, the RMSE of the ST-MGAT model is decreased by almost 2.340%∼22.471%, and the MAE is decreased by almost 7.417%∼26.142%. It is proved that the ST-MGAT model has high performance on the traffic flow prediction task.

Suggested Citation

  • Wang, Bowen & Wang, Jingsheng, 2022. "ST-MGAT:Spatio-temporal multi-head graph attention network for Traffic prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 603(C).
    • Handle: RePEc:eee:phsmap:v:603:y:2022:i:c:s0378437122005076
    DOI: 10.1016/j.physa.2022.127762
    as

    Download full text from publisher

    File URL: http://www.sciencedirect.com/science/article/pii/S0378437122005076
    Download Restriction: Full text for ScienceDirect subscribers only. Journal offers the option of making the article available online on Science direct for a fee of $3,000
    File URL: https://libkey.io/10.1016/j.physa.2022.127762?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    References listed on IDEAS

    as
    1. Alessandro Crivellari & Euro Beinat, 2020. "Forecasting Spatially-Distributed Urban Traffic Volumes via Multi-Target LSTM-Based Neural Network Regressor," Mathematics, MDPI, vol. 8(12), pages 1-16, December.
    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. Liu, Shuai & Feng, Xiaoyuan & Ren, Yilong & Jiang, Han & Yu, Haiyang, 2023. "DCENet: A dynamic correlation evolve network for short-term traffic prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 614(C).
    2. Qian, Jun-Hao & Zhao, Yi-Xin & Huang, Wei, 2023. "Model improvement and scheduling optimization for multi-vehicle charging planning in IoV," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 621(C).
    3. Ma, Changxi & Zhao, Mingxi, 2023. "Spatio-temporal multi-graph convolutional network based on wavelet analysis for vehicle speed prediction," Physica A: Statistical Mechanics and its Applications, Elsevier, vol. 630(C).

    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. Ke Zhang & Yaming Guo, 2023. "Attention-Based Residual Dilated Network for Traffic Accident Prediction," Mathematics, MDPI, vol. 11(9), pages 1-15, April.

    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:eee:phsmap:v:603:y:2022:i:c:s0378437122005076. 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: Catherine Liu (email available below). General contact details of provider: http://www.journals.elsevier.com/physica-a-statistical-mechpplications/ .

    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.