Industrial Engineering Journal ›› 2023, Vol. 26 ›› Issue (5): 159-167.doi: 10.3969/j.issn.1007-7375.2023.05.018

• System Modeling & Optimization Algorithm • Previous Articles    

Traffic Flow Prediction for Large-scale Road Network Based on Graph Transformer

DONG Chengxiang1, WEI Xin1, ZHANG Kunpeng2, WANG Yongchao3   

  1. 1. School of Electromechanical Engineering, Guangdong University of Technology, Guangzhou 510006, China;
    2. College of Electrical Engineering, Henan University of Technology, Zhengzhou 450001, China;
    3. School of Intelligent Manufacturing, Guangzhou Panyu Polytechnic, Guangzhou 511483, China
  • Received:2022-12-26 Published:2023-10-25

Abstract: To accurately predict traffic flow in large-scale road networks, a traffic flow prediction model is proposed based on Graph Transformer to capture the complex and dynamic spatiotemporal characteristics of traffic flows. Gate recurrent unit (GRU) module is adopted in the model to extract temporal features of historical traffic flow data in a road network. According to the connections among sensors distributed in a network, traffic graphs are then established with historical traffic flows as nodes and connections of sensors as edges. On this basis, spatiotemporal characteristics are captured using Graph Transformer-based deep learning technologies. To verify the effectiveness of the proposed model, comparisons with six baseline models using the PeMS highway dataset are conducted. Experiments show that the proposed prediction model results in the best performance.

Key words: large-scale road network, gate recurrent unit (GRU), spatiotemporal characteristics, traffic flow

CLC Number: