Journal of Beijing University of Posts and Telecommunications

  • EI核心期刊

Journal of Beijing University of Posts and Telecommunications ›› 2020, Vol. 43 ›› Issue (5): 9-14.doi: 10.13190/j.jbupt.2019-243

• PAPERS • Previous Articles     Next Articles

Urban Short-Term Traffic Flow Prediction Algorithm Based on CNN-ResNet-LSTM Model

PU Yue-yi1, WANG Wen-han1, ZHU Qiang1, CHEN Peng-peng1,2   

  1. 1. School of Computer Science and Technology, China University of Mining and Technology, Xuzhou 221116, China;
    2. Mine Digitization Engineering Research Center of the Ministry of Education, China University of Mining and Technology, Xuzhou 221116, China
  • Received:2019-11-26 Published:2021-03-11

Abstract: With the continuous advancement of smart city construction, urban short-term traffic flow forecasting becomes more and more important. According to the influence of traffic flow characteristics and external factors on traffic flow forecast results, the model CNN-ResNet-LSTM for urban short-term traffic flow forecasting is proposed. The model integrates convolutional neural networks(CNN), residual neural units(ResNet) and long-short-term memory networks(LSTM) into an end-to-end network framework. The convolutional neural network is used to capture the local spatial characteristics of traffic flow, and multiple residual neural units are added to deepen the network depth and improve the prediction accuracy. On the other hand, the long short-term memory-cycle neural network is used to capture temporal characteristics of traffic flow data. The output results of the two networks are combined by the corresponding weights to obtain the predicted results through the trajectory data. Finally, the traffic flow prediction values of the urban areas are obtained by fusing with external factors. Through the verification of the CNN-ResNet-LSTM model by Beijing data, the model is not only higher in accuracy than the traditional model, but also has fewer parameters in the case of ensuring the accuracy of prediction.

Key words: traffic flow forecasting, deep learning, long short-term memory network, convolutional neural network

CLC Number: