A Novel Hybrid Model for Short-Term Traffic Flow Prediction Based on Spatio-Temporal Deep Learning With Considering Associated Factors Selection

Yingping Tang,Qiang Shang,Longjiao Yin
DOI: https://doi.org/10.1109/access.2024.3456295
IF: 3.9
2024-09-21
IEEE Access
Abstract:Effective predicting of traffic flow is the key to fully utilizing the carrying capacity of roads and improving the travel experience. In order to overcome the randomness and volatility of traffic flow, a novel hybrid model for short-term traffic flow prediction based on spatio-temporal deep learning with considering associated factors selection is proposed. In this model, traffic flow with high spatial relevance is identified firstly with the use of the Pearson Product-Moment Correlation Coefficient (PPMCC), then associated factors affecting traffic flow are screened with the use of Random Forest (RF), finally, the results of the two steps mentioned above and the historical traffic flow data are used as input, and the Convolutional Neural Network (CNN) and Bidirectional Gated Recurrent Unit (BiGRU) are used to perceive the spatiotemporal characteristics of traffic flow and the hidden relationships between various factors and traffic flow to predict traffic flow. To test the performance of the proposed model, seven baseline models proposed in the existing literature are compared on publicly available datasets using four indexes to evaluate the performance of the models. In addition, we conducted an ablation study. The results showed that the proposed model has a 39% decrease in root mean square error and a 37% decrease in mean absolute error compared to the baseline models with the best performance.
computer science, information systems,telecommunications,engineering, electrical & electronic
What problem does this paper attempt to address?