A Light Weight Traffic Volume Prediction Approach Based on Finite Traffic Volume Data

Abstract

As one of the key technologies of intelligent transportation systems, short-term traffic volume prediction plays an increasingly important role in solving urban traffic problems. In the last decade, many approaches were proposed for the traffic volume prediction from different perspectives. However, most of these approaches are based on a large amount of historical data. When there are only finite collected traffic data, they cannot be well trained, so the prediction accuracy of these approaches will be poor. In this paper, a tensor model is proposed to capture the change patterns of continuous traffic volumes. From collected traffic volume data, the element data are extracted to update the corresponding elements of the tensor model. Then, a tucker decomposition and gradient descent based algorithm is employed to impute the missing elements of the tensor model. After missing element imputation, the tensor model can be directly applied to the short-term traffic volume prediction through searching the corresponding elements of the model and the storage cost of the model is low. Our model is evaluated on real traffic volume data from PeMS dataset, which indicates that our model has higher traffic volume prediction accuracy than other approaches in the situation of finite traffic volume data.

Keywords

Short-term traffic volume prediction / tensor / Tucker decomposition / finite traffic volume data

References

[1] Acar E, Dunlavy D M, Kolda T G, Merup M. Scalable tensor factorizations for incomplete data. Chemometrics & Intelligent Laboratory Systems, 2010, 106(1): 41-56.

[2] Akash P S, Chang K C (2022). Exploring variational graph auto-encoders for extract class refactoring recommendation. arXiv Preprint arXiv: 2203.08787.

[3] Akhtar M, Moridpour S. A review of traffic congestion prediction using artificial intelligence. Journal of Advanced Transportation, 2021, 2021: 1-18.

[4] Aldegheishem A, Yasmeen H, Maryam H, Shah M A, Mehmood A, Alrajeh N, Song H. Smart road traffic accidents reduction strategy based on intelligent transportation systems (tars). Sensors, 2018, 18(7): 1983.

[5] Caltrans (2023). PeMS. http://pems.dot.ca.gov.

[6] Castro-Neto M, Jeong Y S, Jeong M K, Han L D. Online-SVR for short-term traffic flow prediction under typical and atypical traffic conditions. Expert Systems with Applications, 2009, 36(3): 6164-6173.

[7] Chen W, Zhao Z, Liu J, Chen P W. LSTM network: A deep learning approach for short-term traffic forecast. IET Intelligent Transport Systems, 2017, 11(2): 68-75.

[8] Duan P, Mao G, Zhang C, Wang S (2016). STARIMA-based traffic prediction with time-varying lags. 2016 IEEE 19th International Conference on Intelligent Transportation Systems (ITSC). Rio de Janeiro, Brazil, November 01–04, 2016.

[9] Ferreira M, d’Orey P M. On the impact of virtual traffic lights on carbon emissions mitigation. IEEE Transactions on Intelligent Transportation Systems, 2012, 13(1): 284-295.

[10] Fu R, Zhang Z, Li L (2016). Using LSTM and GRU neural network methods for traffic flow prediction. 2016 31st Youth Academic Annual Conference of Chinese Association of Automation (YAC). Wuhan, China, November 11–13, 2016.

[11] Habtemichael F G, Cetin M. Short-term traffic flow rate forecasting based on identifying similar traffic patterns. Transportation Research Part C: Emerging Technologies, 2016, 66: 61-78.

[12] Hamed M M, Al-Masaeid H R, Said Z M B. Short-term prediction of traffic volume in urban arterials. Journal of Transportation Engineering, 1995, 121(3): 249-254.

[13] Hou Z, Li X. Repeatability and similarity of freeway traffic flow and long-term prediction under big data. IEEE Transactions on Intelligent Transportation Systems, 2016, 17(6): 1786-1796.

[14] Kim Y, Wang P, Mihaylova L (2019). Structural recurrent neural network for traffic speed prediction. ICASSP 2019-2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP). Brighton, United Kingdom, May 12–17, 2019.

[15] Kolda T G, Bader B W. Tensor decompositions and applications. SIAM Review, 2009, 51(3): 455-500.

[16] Li Y, Li Z, Li L. Missing traffic data: Comparison of imputation methods. IET Intelligent Transport Systems, 2014, 8(1): 51-57.

[17] Li Y, Shahabi C. A brief overview of machine learning methods for short-term traffic forecasting and future directions. Sigspatial Special, 2018, 10(1): 3-9.

[18] Liao J, Tang J, Zeng W, Zhao X. Efficient and accurate traffic flow prediction via incremental tensor completion. IEEE Access, 2018, 6: 36897-36905.

[19] Ma J, Meng Y (2008). Research of traffic flow forecasting based on neural network. IEEE 2008 Second International Symposium on Intelligent Information Technology Application. Shanghai, China, December 20–22, 2008.

[20] Miyazaki. Book Review: Faraway, Julian J (2006). Extending the linear model with R: Generalized linear, mixed effects and nonparametric regression models. Applied Psychological Measurement, 2011, 35(4): 330-333.

[21] Smith B L, Demetsky M J. Short-term traffic flow prediction: Neural network approach. Transportation Research Record, 1994, 1453(1453): 98-104.

[22] Smith B L, Williams B M, Oswald R. Comparison of parametric and nonparametric models for traffic flow forecasting. Transportation Research Part C: Emerging Technologies, 2002, 10(4): 303-321.

[23] Song X, Guan F, Yang Z, Yao B. K-nearest neighbor model for multiple-time-step prediction of short-term traffic condition. Journal of Transportation Engineering, 2016, 142(6): 4016018.

[24] Su X, Fan M, Zhang M, Liang Y, Guo L. An innovative approach for the short-term traffic flow prediction. Journal of Systems Science and Systems Engineering, 2021, 30(5): 519-532.

[25] Sun B, Cheng W, Goswami P, Bai G. Short-term traffic forecasting using self-adjusting k-nearest neighbours. IET Intelligent Transport Systems, 2018, 12(1): 41-48.

[26] Tan H, Feng G, Feng J, Wang W, Zhang Y J, Li F. A tensor-based method for missing traffic data completion. Transportation Research Part C: Emerging Technologies, 2013, 28: 15-27.

[27] Tan H, Wu Y, Shen B, Jin P J, Ran B. Short-term traffic prediction based on dynamic tensor completion. IEEE Transactions on Intelligent Transportation Systems, 2016, 17(8): 1-11.

[28] Tchrakian T T, Basu B, O’Mahony M. Real-time traffic flow forecasting using spectral analysis. IEEE Transactions on Intelligent Transportation Systems, 2012, 13(2): 519-526.

[29] Tomasi G, Bro R. A comparison of algorithms for fitting the PARAFAC model. Computational Statistics & Data Analysis, 2006, 50(7): 1700-1734.

[30] Vlahogianni E I, Golias J C, Karlaftis M G. Short-term traffic forecasting: Overview of objectives and methods. Transport Reviews, 2004, 24(5): 533-557.

[31] Vlahogianni E I, Karlaftis M G, Golias J C. Spatio-temporal short-term urban traffic volume forecasting using genetically optimized modular networks. Computer-Aided Civil and Infrastructure Engineering, 2007, 22(5): 317-325.

[32] Vlahogianni E I, Karlaftis M G, Golias J C. Short-term traffic forecasting: Where we are and where were going. Transportation Research Part C: Emerging Technologies, 2014, 43: 3-19.

[33] Wang Y, Li L, Xu X (2017). A piecewise hybrid of ARIMA and SVMs for short-term traffic flow prediction. Neural Information Processing: 24th International Conference (ICONIP 2017). Guangzhou, China, November 14–18, 2017.

[34] Wang Z, Su X, Ding Z. Long-term traffic prediction based on LSTM encoder-decoder architecture. IEEE Transactions on Intelligent Transportation Systems, 2020, 22(10): 6561-6571.

[35] Wang Y, Zheng Y, Xue Y (2014). Travel time estimation of a path using sparse trajectories. Proceedings of the 20th SIGKDD Conference on Knowledge Discovery and Data Mining (KDD 2014). New York, USA, August 24–27, 2014.

[36] Wei W, Wu H, Ma H. An autoencoder and LSTM-based traffic flow prediction method. Sensors, 2019, 19(13): 2946.

[37] Williams B M, Hoel L A. Modeling and forecasting vehicular traffic flow as a seasonal ARIMA process: Theoretical basis and empirical results. Journal of Transportation Engineering, 2003, 129(6): 664-672.

[38] Wu C H, Ho J M, Lee D. Travel-time prediction with support vector regression. IEEE Transactions on Intelligent Transportation Systems, 2005, 5(4): 276-281.

[39] Xia D, Zhang M, Yan X, Bai Y, Li H. A distributed WND-LSTM model on mapreduce for short-term traffic flow prediction. Neural Computing and Applications, 2021, 33: 2393-2410.

[40] Xu D W, Wang Y D, Jia L M, Qin Y, Dong H H. Real-time road traffic state prediction based on ARIMA and Kalman filter. Frontiers of Information Technology and Electronic Engineering, 2017, 18: 287-302.

[41] Zhang M, Zhen Y, Hui G, Chen G. Accurate multi-steps traffic flow prediction based on SVM. Mathematical Problems in Engineering, 2013, 2013(6): 91-109.

[42] Zhao Q, Zhang L, Cichocki A. Bayesian CP factorization of incomplete tensors with automatic rank determination. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2015, 37(9): 1751-1763.

[43] Zonoozi A, Kim J J, Li X L, Cong G (2018). Periodic-CRN: A convolutional recurrent model for crowd density prediction with recurring periodic patterns. Twenty-Seventh International Joint Conference on Artificial Intelligence IJCAI-18. Stockholm, Sweden, July 13–19, 2018.