Skip to main content
SpringerLink
Log in

A Short-term Traffic Speed Prediction Model Based on LSTM Networks

  • Published:
International Journal of Intelligent Transportation Systems Research Aims and scope Submit manuscript

Abstract

To successfully deploy an intelligent transportation system, it is essential to construct an effective method of traffic speed prediction. Recently, due to the advancements in sensor technology, traffic data have experienced explosive growth. It is therefore a challenge to construct an efficient model with highly accurate predictions. To improve the accuracy and the efficiency of short-term traffic predictions, we propose a prediction model based on deep learning approaches. We use a long short-term memory (LSTM) network to analyze sequential sensor data to predict the car speed of the next time interval on the freeway. Unlike the traditional model that only considers the changes in traffic speed which is used to derive the temporal and spatial features from the prediction road section, we mainly consider the features of the number of the most representative car types and the traffic speed variation of the front road segment that is ahead of the prediction road segment in addition to the number of cars, the road occupancy, and the traffic speed latency to successfully learn and capture the hidden patterns from the sensor data so as to improve the prediction accuracy. To the best of our knowledge, very few investigations have been conducted to consider the correlation between car speed and car type for a prediction model. Moreover, our extensive experiments demonstrate that the proposed method for traffic speed prediction has achieved high accuracy.

This is a preview of subscription content, log in via an institution to check access.

Access this article

Log in via an institution

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Institutional subscriptions

Fig. 1
Fig. 2
Fig. 3
Fig. 4
Fig. 5
Fig. 6
Fig. 7
Fig. 8
Fig. 9
Fig. 10
Fig. 11
Fig. 12
Fig. 13
Fig. 14
Fig. 15
Fig. 16
Fig. 17
Fig. 18
Fig. 19
Fig. 20
Fig. 21
Fig. 22
Fig. 23

Similar content being viewed by others

References

  1. Keras: https://keras.io/ (2021)

  2. Theano: https://github.com/Theano/Theano (2021)

  3. Ahmed, M.S., Cook, A.R.: Analysis of freeway traffic time-series data by using Box–Jenkins techniques. Transportation Research Record (722):1–9 (1979)

  4. Bezuglov, A., Comert, G.: Short-term freeway traffic parameter prediction: Application of grey system theory models. Expert Syst. Appl. 62, 284–292 (2016)

    Article  Google Scholar 

  5. Freeway Bureau: Ministry of transportation and communications http://tisvcloud.freeway.gov.tw (2021)

  6. Wang, Y., Lu, G., Chen, P., Ding, C., Sun, J., Cai, P.: A spatiotemporal cor- relative k-nearest neighbor model for short-term traffic multistep forecasting. Transportation Research Part C Emerging Technologies 62, 21–34 (2016)

    Article  Google Scholar 

  7. Jeong, Y.S., Jeong, M.K., Castro-Neto, M.: Online-svr for short-term traffic flow prediction under typical and atypical traffic conditions. Expert Syst. Appl. 36(3), 6164–6173 (2009)

    Article  Google Scholar 

  8. Chung, J., Gulcehre, C., Cho, K., Bengio, Y.: Empirical evaluation of gated recurrent neural networks on sequence modeling. In: NIPS Workshop on Deep Learning (2014)

  9. Davis, G.A., Nihan, N.L.: Nonparametric regression and short-term freeway traffic forecasting. J. Transp. Eng. 117(2), 178–1888 (1991)

    Article  Google Scholar 

  10. Dharia, A., Adeli, H.: Neural network model for rapid forecasting of freeway link travel time, vol. 16 (2003)

  11. Karlaftis, M.G., Vlahogianni, E.I., Golias, J.C.: Optimized and metaoptimized neural networks for short-term traffic flow prediction: A genetic approach. Transp. Res. C, Emerging Technol. 13(3), 211–234 (2005)

    Article  Google Scholar 

  12. Liu, F., Yang, H., Yin, Z., Ran, B.: Online recursive algorithm for shortterm traffic prediction. Transp. Res Rec. 1879, 1–8 (2004)

    Article  Google Scholar 

  13. Hochreiter, S, Schmidhuber, J.: Long short-term memory. Neural Comput. 9(8), 1735–1780 (1997)

    Article  Google Scholar 

  14. Song, G., Hong, H., Huang, W.: Deep architecture for traffic flow prediction: deep belief networks with multitask learning. IEEE Trans. Intell. Transport. Syst. 15(5), 2191–2201 (2014)

    Article  Google Scholar 

  15. Kamarianakis, Y., Prastacos, P.: Forecasting traffic flow conditions in an urban network—comparison of multivariate and univariate approaches. Transp. Res. Rec. 1857, 74–84 (2003)

    Article  Google Scholar 

  16. Lee, S., Fambro, D: Application of subset autoregressive integrated moving average model for short-term freeway traffic volume forecasting. Transp. Res. Rec. 1678, 179–188 (1999)

    Article  Google Scholar 

  17. Levin, M., Tsao, Y.-D.: On forecasting freeway occupancies and volumes. Transp. Res. Rec. 773, 47–49 (1980)

    Google Scholar 

  18. Sharma, S., Zhong, M., Lingras, P.: Prediction of recreational travel using genetically designed regression and time-delay neural network models. Transport. Res. Record 13(1), 435–446 (2002)

    Google Scholar 

  19. Duan, Y., Kang, W., Lv, Y.: Traffic flow prediction with big data: A deep learning approach. IEEE Trans. Intell. Trans.. Syst. 16(2), 865–873 (2015)

    Google Scholar 

  20. Dougherty, M., vander Voort, M., Watson, S.: Combining kohonen maps with arima time series models to forecast traffic flow. Transp. Res. C, Emerging Technol. 4(5), 307–318 (1996)

    Article  Google Scholar 

  21. Tao, Z., Wang, Y., Ma, X.: Long short-term memory neural network for traffic speed prediction using remote microwave sensor data. Transport. Res. C Emerging Technol. 54, 187–197 (2015)

    Article  Google Scholar 

  22. Yu, H., Wang, Y., Ma, X.: Large-scale transportation network congestion evolution prediction using deep learning theory. PLoS One 10(3), 1–17 (2015)

    Google Scholar 

  23. Urbanik II, T., Messer, C.: Short-term freeway traffic volume forecasting using radial basis function neural network. Transport. Res. Record 1651(1), 39–47 (1998)

    Article  Google Scholar 

  24. Okutani, I., Stephanedes, Y.J.: Dynamic prediction of traffic volume through kalman filtering theory. Trans. Res. B Methodol. 18(1), 1–11 (1984)

    Article  Google Scholar 

  25. Park, B.: Hybrid neuro-fuzzy application in short-term freeway traffic volume forecasting. Transp. Res. Rec. 1802, 190–196 (2002)

    Article  Google Scholar 

  26. Rilett Laurence, R., Han, G., Park, D.: Spectral basis neural networks for real-time travel time forecasting. J. Transp. Eng. 125(125), 515–523 (1999)

    Google Scholar 

  27. Rilett Laurence, R., Park, D.: Forecasting freeway link travel times with a multilayer feedforward neural network. Comput.-Aided Infrastruct. Eng. 14(5), 357–367 (1999)

    Article  Google Scholar 

  28. Pascanu, R., Mikolov, T., Bengio, Y.: On the difficulty of training recurrent neural networks. In: Proceedings of the 30th International Conference on International Conference on Machine Learning - Volume 28, ICML’13, page III–1310–III–1318. JMLR.org (2013)

  29. Ishak, S., Qi, Y.: A hidden markov model for short term prediction of traffic conditions on freeways. Transp. Res. Part C: Emerg. Technol. 43, 95–111 (2014)

    Article  Google Scholar 

  30. Qi, Y., Ishak, S.: Stochastic approach for short-term freeway traffic prediction during peak periods. IEEE Trans. Intell Transp. Syst. 14(2), 660–672 (2013)

    Article  Google Scholar 

  31. Zhang S. , Sun, C., Guoqiang, Y.: A bayesian network approach to traffic flow forecasting. IEEE Intell. Transp. Syst. Mag. 7(1), 124–132 (2006)

    Article  Google Scholar 

  32. Smith, B.L. , Demetsky, M.J.: Short-term traffic flow prediction: Neural network approach. Transp. Res. Rec 1453, 98–104 (1994)

    Google Scholar 

  33. Research Board Transportation: Highway capacity manual, transp (2000)

  34. Lee, D.H., Zheng, W.Z., Shi, Q.X.: Short-term freeway traffic flow prediction: Bayesian combined neural network approach. J. Transp. Eng. 132(2), 114–121 (2006)

    Article  Google Scholar 

  35. Wang, J., Shi, Q.: Short-term traffic speed forecasting hybrid model based on chaos–wavelet analysis-support vector machine theory. Transp. Res. Part C: Emerg. Technol. 27, 219–232 (2013)

    Article  Google Scholar 

  36. Williams, B.M.: Multivariate vehicular traffic flow prediction— evaluation of arimax modeling. Transp. Res. Rec. 1776, 194–200 (2001)

    Article  Google Scholar 

  37. Williams, B.M., Hoel, L.A.: Modeling and forecasting vehicular traffic flow as a seasonal arima process: Theoretical basis and empirical results. J. Transp. Eng. 129(6), 664–672 (2003)

    Article  Google Scholar 

  38. Zhang, Y., Ye Z, Xie, Y.: Short-term traffic volume forecasting using kalman filter with discrete wavelet decomposition (2006)

  39. Castro-Neto, M.M., Jeong, Y.S., Byon, Y.J., Easa, S.M.: Supervised weighting-online learning algorithm for short-term traffic flow prediction. IEEE Trans. Intell Transp. Syst. 14(4), 1700–1707 (2013)

    Article  Google Scholar 

  40. Liu, Y., Zhang, Y.: Traffic forecasting using least squares support vector machines. Transportmetrica 5(3), 193–213 (2009)

    Article  MathSciNet  Google Scholar 

  41. Chen, W., Wu, X., Chen, P.C., Liu, J., Zhao, Z.: Lstm network: a deep learning approach for short-term traffic forecast. IET Intel. Transp. Syst. 11, 68–75 (2017)

    Article  Google Scholar 

Download references

Author information

Authors and Affiliations

  1. Department of Computer Science & Information Engineering, National Chung Cheng University, Chiayi County, Taiwan

    Yu-Ling Hsueh & Yu-Ren Yang

  2. Center for Innovative Research on Aging Society (CIRAS), National Chung Cheng University, Chiayi County, Taiwan

    Yu-Ling Hsueh

  3. Advanced Institute of Manufacturing with High-tech Innovations (AIM-HI), National Chung Cheng University, Chiayi County, Taiwan

    Yu-Ling Hsueh

Authors
  1. Yu-Ling Hsueh
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Yu-Ren Yang
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Yu-Ling Hsueh.

Additional information

Publisher’s Note

Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Hsueh, YL., Yang, YR. A Short-term Traffic Speed Prediction Model Based on LSTM Networks. Int. J. ITS Res. 19, 510–524 (2021). https://doi.org/10.1007/s13177-021-00260-7

Download citation

  • Received:

  • Revised:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1007/s13177-021-00260-7

Keywords

Search

Navigation