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

Bowen Wang and Jingsheng Wang

Physica A: Statistical Mechanics and its Applications, 2022, vol. 603, issue C

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.

Keywords: Traffic engineering; Traffic flow prediction model; GAT model; GLU model; Traffic flow (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (3)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0378437122005076
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

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:phsmap:v:603:y:2022:i:c:s0378437122005076

DOI: 10.1016/j.physa.2022.127762

Access Statistics for this article

Physica A: Statistical Mechanics and its Applications is currently edited by K. A. Dawson, J. O. Indekeu, H.E. Stanley and C. Tsallis

More articles in Physica A: Statistical Mechanics and its Applications from Elsevier
Bibliographic data for series maintained by Catherine Liu ().