A Short-term Traffic Flow Prediction Model for Urban Road Network Based on Graph Transformer Network

Feng ZHOU , Shi-pu WANG , Kun-peng ZHANG

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (10) : 4307 -4316.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (10) : 4307-4316 DOI: 10.12404/j.issn.1671-1815.2303913
research-article
A Short-term Traffic Flow Prediction Model for Urban Road Network Based on Graph Transformer Network
Author information +
History +
PDF

Abstract

A prediction method based on graph transformer (Graformer) was proposed for short-term traffic flow prediction in urban road networks, considering the spatiotemporal correlation of road network traffic status. This method transformed the traffic state prediction problem of multiple road sections into a graph node state prediction problem. For distinguishing spatial road network structures with the same structure, a graph isomorphism network with edges (GINE) and a Transformer network to model the spatiotemporal correlation of traffic states at the road network level were combined, thereby achieving short-term traffic flow prediction for urban road networks. Specifically, the Graformer model first used the long short term memory (LSTM) network to preprocess the temporal information of the traffic data, then used the global attention mechanism based on GINE and Transformer to extract the spatial characteristics of the traffic data, and finally realized the synchronous prediction of the traffic flow of each road section of the road network. Through experimental verification using the PeMS dataset, the results show that the proposed Graformer model outperformed the comparative model in various performance indicators, demonstrating its effectiveness as a reliable and efficient short-term traffic flow prediction method for road networks.

Cite this article

Download citation ▾
Feng ZHOU, Shi-pu WANG, Kun-peng ZHANG. A Short-term Traffic Flow Prediction Model for Urban Road Network Based on Graph Transformer Network. Science Technology and Engineering, 2024, 24(10): 4307-4316 DOI:10.12404/j.issn.1671-1815.2303913

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (2785KB)

16

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉