A Novel STFSA-CNN-GRU Hybrid Model for Short-Term Traffic Speed Prediction

Short-term traffic speed prediction is fundamental to intelligent transportation systems (ITS), and the accuracy of the model largely determines the performance of real-time traffic control and management. In this study, a short-term traffic speed prediction method based on the spatial-temporal analysis of traffic flow and a combined deep-learning model, and a hybrid spatial-temporal feature selection algorithm (STFSA) of a convolutional neural network–gated recurrent unit (CNN-GRU)) is initially developed. Specifically, the STFSA is firstly employed to reconstruct the spatial-temporal matrix of traffic speed based on temporal continuity and spatial characteristics, and then this matrix is considered as the input feature of the prediction model. After this, the nonlinear fitting ability of the CNN is adopted to extract deep features from the convolutional and pooling layers for model training. Finally, by combining the timing and long-range dependence of the captured data with the forward GRU and the reverse GRU, the accuracy of the prediction result is further improved. The validity of the proposed model can be verified by comparing the prediction results with the actual traffic data. Accordingly, in the case study, the performance is compared with various benchmark methods under the same prediction scenario, verifying the superiority of the proposed model.