A Spatiotemporal Convolutional Neural Network Model Based on Dual Attention Mechanism for Passenger Flow Prediction

Establishing a high-precision passenger flow prediction model is a critical and complex task for the optimization of urban rail transit systems. With the development of artificial intelligence technology, the data-driven technology has been widely studied in the intelligent transportation system. In this study, a neural network model based on the data-driven technology is established for the prediction of passenger flow in multiple urban rail transit stations to enable smart perception for optimizing urban railway transportation. The integration of network units with different specialities in the proposed model allows the network to capture passenger flow data, temporal correlation, spatial correlation, and spatiotemporal correlation with the dual attention mechanism, further improving the prediction accuracy. Experiments based on the actual passenger flow data of Beijing Metro Line 13 are conducted to compare the prediction performance of the proposed data-driven model with the other baseline models. The experimental results demonstrate that the proposed prediction model achieves lower MAE and RMSE in passenger flow prediction, and its fitted curve more closely aligns with the actual passenger flow data. This demonstrates the model’s practical potential to enhance intelligent transportation system management through more accurate passenger flow forecasting.