A dynamic multigraph and multidimensional attention neural network model for metro passenger flow prediction

Abstract

SummaryPassenger flow prediction is an important part of daily metro operation, and its accuracy affects the deployment of train resources management. Due to the complex spatiotemporal correlation characteristics of metro passenger flow, it is necessary to describe it to improve the accuracy of passenger flow prediction. However, the existing models mainly construct the weight matrix based on the static graph and the similarity between stations when describing the spatial correlation of station passenger flow but ignore the time‐varying characteristics of the spatial correlation of station passenger flow. To address this problem, this study introduces a dynamic multi‐graph and multidimensional attention spatiotemporal model. Specifically, the Graph Convolutional Neural Network combined with dynamic multigraph extracts spatial features and the Gated Recurrent Unit extracts temporal features of passenger flow. The multidimensional attention can obtain the spatiotemporal correlation of passenger flow data by assigning weights to them. Finally, this model has been used to conduct experiments on Beijing metro passenger flow datasets with time granularity of 10 and 15 min. The result indicates that the DGMANN model outperforms state‐of‐the‐art other deep learning methods in passenger flow prediction. In addition, the effectiveness of its key submodules has been verified through ablation experiments.