高技术通讯(英文版)2024,Vol.30Issue(3) :221-230.DOI:10.3772/j.issn.1006-6748.2024.03.001

Adaptive spatial-temporal graph attention network for traffic speed prediction

张玺君 ZHANG Baoqi ZHANG Hong NIE Shengyuan ZHANG Xianli
高技术通讯(英文版)2024,Vol.30Issue(3) :221-230.DOI:10.3772/j.issn.1006-6748.2024.03.001

Adaptive spatial-temporal graph attention network for traffic speed prediction

张玺君 1ZHANG Baoqi 1ZHANG Hong 1NIE Shengyuan 1ZHANG Xianli1
扫码查看

作者信息

  • 1. School of Computer and Communication,Lanzhou University of Technology,Lanzhou 730050,P.R.China
  • 折叠

Abstract

Considering the nonlinear structure and spatial-temporal correlation of traffic network,and the influence of potential correlation between nodes of traffic network on the spatial features,this paper proposes a traffic speed prediction model based on the combination of graph attention network with self-adaptive adjacency matrix(SAdpGAT)and bidirectional gated recurrent unit(BiGRU).First-ly,the model introduces graph attention network(GAT)to extract the spatial features of real road network and potential road network respectively in spatial dimension.Secondly,the spatial features are input into BiGRU to extract the time series features.Finally,the prediction results of the real road network and the potential road network are connected to generate the final prediction results of the model.The experimental results show that the prediction accuracy of the proposed model is im-proved obviously on METR-LA and PEMS-BAY datasets,which proves the advantages of the pro-posed spatial-temporal model in traffic speed prediction.

Key words

traffic speed prediction/spatial-temporal correlation/self-adaptive adjacency ma-trix/graph attention network(GAT)/bidirectional gated recurrent unit(BiGRU)

引用本文复制引用

出版年

2024
高技术通讯(英文版)
中国科学技术信息研究所(ISTIC)

高技术通讯(英文版)

影响因子:0.058
ISSN:1006-6748
段落导航相关论文