Traffic & Transportation Engineering

Passenger Flow Forecast of Urban Rail Transit Based on Graph Convolution and Recurrent Neural Network

Expand
  • 1. Key Laboratory of Road Traffic Engineering of the Ministry of Education, Tongji University, Shanghai 201804, China;
    2. CAD Research Center of Tongji University, Shanghai 201804, China;
    3. Shanghai Shentong Metro Group Co., Ltd., Shanghai 201102, China
刘晓磊(1997-),女,博士生,主要从事交通大数据研究。E-mail:liuxiaolei_2021@163.com

Received date: 2021-05-20

  Revised date: 2021-08-11

  Online published: 2021-08-22

Supported by

Supported by Shanghai Science and Technology Commission Research Project (20dz1202903)

Abstract

Passenger flow forecast is of great significance to the organization and management of urban rail transit. This paper constructed a graph convolution and recurrent neural network (GCGRU model) by combining graph convolutional network with recurrent neural network. The graph convolutional network was used to learn the complex topological structure of an urban rail network and capture spatial correlation characteristics. Then one of the recurrent neural network variants called gated recurrent unit was used to learn the variation of multi-characteristics of traffic trends and to capture the temporal characteristics. An experiment was carried out with the passenger flow data obtained from the entire network with all subway cross-sections in Shanghai in a whole year, and the mean decrease impurity method provided by random forest was used for feature selection. The experimental results show that the GCGRU model can well capture the temporal and spatial correlation in the prediction of large-scale urban rail transit passenger flow, with a prediction accuracy of 89%. The prediction results can provide a basis for managers to manage and organize rail transit passenger flow as well as provide travelers with early warning information, ensuring the safe and efficient operation of the urban rail transit network.

Cite this article

LIU Xiaolei, DUAN Zhengyu, YU Qing, et al . Passenger Flow Forecast of Urban Rail Transit Based on Graph Convolution and Recurrent Neural Network[J]. Journal of South China University of Technology(Natural Science), 2022 , 50(3) : 21 -27 . DOI: 10.12141/j.issn.1000-565X.210320

Outlines

/