SAD-ARGRU: A Metro Passenger Flow Prediction Model for Graph Residual Gated Recurrent Networks

Abstract

This paper proposes a graph residual gated recurrent network subway passenger flow prediction model considering the flat-peak characteristics, which firstly proposes the use of an adaptive density clustering method, which is capable of dynamically dividing the flat-peak time period of subway passenger flow. Secondly, this paper proposes graph residual gated recurrent network, which uses a graph convolutional network fused with a residual network and combined with a gated recurrent network, to simultaneously learn the temporal and spatial characteristics of passenger flow. Finally, this paper proposes to use the spatial attention mechanism to learn the spatial features around the subway stations, construct the spatial local feature components, and fully learn the spatial features around the stations to realize the local quantization of the spatial features around the subway stations. The experimental results show that the graph residual gated recurrent network considering the flat-peak characteristics can effectively improve the prediction performance of the model, and the method proposed in this paper has the highest prediction accuracy when compared with the traditional prediction model.

Keywords

• spatial attention mechanism
• local character component
• graph convolutional network
• gate recurrent unit