Prediction of traffic flow based on deep learning

  • Yuheng zhou CSU
  • Zuping Zhang
Keywords: deep learning, traffic flow prediction, neural network, Random Forests


Deep neural networks (DNNs) have recently demonstrated the capability to predict traffic flow with big data. Although existing DNN models can provide better performance than shallow models, it is still an open question to make full use of the spatio-temporal characteristics of traffic flows to improve performance. We propose a novel deep architecture combining CNN and LSTM for traffic flow (RCF) predictio. The model uses CNN to explore temporal correlation and LSTM to explore spatial correlation . Factors such as weather and historical period data are also added to the feature. Its advantage lies in making full use of the spatial-temporal correlation of traffic data and more comprehensively considered the impact of multiple related factors. Aiming at the difficult problem of obtaining spatial features, a feature selection method based on Random Forests is proposed. We use the gini score to represent the spatial connection between intersections to form a network graph constructed based on data.  The experimental results show that based on the random forest feature selection and RCF model, the accuracy of traffic prediction reaches 90% .


Download data is not yet available.


[1] Yuan Y, Tian C, Lu X. Auxiliary Loss Multimodal GRU Model in Audio-visual Speech Recognition[J]. IEEE Access, 2018, PP(99): 1–1.
[2] Jin X, Xu C, Feng J, et al. Deep Learning with S-shaped Rectified Linear Activation Units[J]. Computer Science, 2015, 3: 1–8.
[3] Li L, Li Y, Li Z. Efficient missing data imputing for traffic flow by considering temporal and spatial dependence[J]. Transportation Research Part C Emerging Technologies, 2013, 34(9): 108–120.
[4] Tan H, Wu Y, Shen B, et al. Short-Term Traffic Prediction Based on Dynamic Tensor Completion[J]. IEEE Transactions on Intelligent Transportation Systems, 2016, 17(8): 1–11.
[5] Okutani I, Stephanedes Y J. Dynamic prediction of traffic volume through Kalman filtering theory[J]. Transportation Research Part B Methodological, 1984, 18(1): 1–11.
[6] Smith B L, Demetsky M J. Traffic flow forecasting: comparison of modeling approaches[J]. Journal of transportation engineering, 1997, 123(4): 261–266.
[7] Kim, E.Y. MRF model based real-time traffic flow prediction with support vector regression[J]. Electronics Letters, 2017, 53(4): 243–245.
[8] Luo X, Niu L, Zhang S. An algorithm for traffic flow prediction based on improved SARIMA and GA[J]. KSCE Journal of Civil Engineering, 2018, 22(10): 4107–4115.
[9] Haofan Yang, Tharam S. Dillon, Elizabeth Chang, et al. Optimized Configuration of Exponential Smoothing and Extreme Learning Machine for Traffic Flow Forecasting[J]. IEEE Transactions on Industrial Informatics, 2019, 15(1): 23–34.
[10] Jiang X, Adeli H. Dynamic wavelet neural network model for traffic flow forecasting[J]. Journal of transportation engineering, 2005, 131(10): 771–779.
[11] Lv Y, Duan Y, Kang W, et al. Traffic Flow Prediction With Big Data: A Deep Learning Approach[J]. IEEE Transactions on Intelligent Transportation Systems, 2015, 16(2): 865–873.
[12] Yang H F, Dillon T S, Chen Y P P. Optimized Structure of the Traffic Flow Forecasting Model With a Deep Learning Approach[J]. IEEE Transactions on Neural Networks & Learning Systems, 2017, 28(10): 2371–2381.
[13] Polson N G, Sokolov V O. Deep learning for short-term traffic flow prediction[J]. Transportation Research Part C: Emerging Technologies, 2017, 79: 1–17.
[14] Zhang J, Zheng Y, Qi D, et al. DNN-based prediction model for spatio-temporal data[A]. Proceedings of the 24th ACM SIGSPATIAL International Conference on Advances in Geographic Information Systems[C]. ACM, 2016: 92.
[15] Ma X, Tao Z, Wang Y, et al. Long short-term memory neural network for traffic speed prediction using remote microwave sensor data[J]. Transportation Research Part C, 2015, 54: 187–197.
[16] Zhang D, Kabuka M R. Combining weather condition data to predict traffic flow: a GRU-based deep learning approach[J]. IET Intelligent Transport Systems, 2018, 12(7): 578–585.
[17] Huang W, Song G, Hong H, et al. Deep Architecture for Traffic Flow Prediction: Deep Belief Networks With Multitask Learning[J]. IEEE Transactions on Intelligent Transportation Systems, 2014, 15(5): 2191–2201.
[18] Yu Z, Liu K, Zhao C, et al. Combined Forecasting Model of Traffic Flow Based on DGM (1, 1) and GRNN[A]. 2017 International Conference on Computing Intelligence and Information System (CIIS)[C]. IEEE, 2017: 58–61.
[19] Xu H, Jiang C. Deep belief network-based support vector regression method for traffic flow forecasting[J]. Neural Computing and Applications, 2019: 1–10.
[20] Tang F, Ishwaran H. Random Forests missing data algorithms[J]. Statistical Analysis and Data Mining: The ASA Data Science Journal, 2017, 10(6): 363–377.
How to Cite
zhou, Y., & Zhang, Z. (2020). Prediction of traffic flow based on deep learning. International Journal of Advanced Computer Technology, 9(2), 05-11. Retrieved from