A Long-Term Traffic Flow Prediction Model Based on Variational Mode Decomposition and Auto-Correlation Mechanism
Source abstract
Abstract
Traffic flow forecasting, as an integral part of intelligent transportation systems, plays a critical part in traffic planning. Previous studies have primarily focused on short-term traffic flow prediction, paying insufficient attention to long-term prediction. In this study, we propose a hybrid model that utilizes variational mode decomposition (VMD) and the auto-correlation mechanism for long-term prediction. In view of the periodic and stochastic characteristics of traffic flow, VMD is able to decompose the data into intrinsic mode functions with different frequencies, which in turn helps the model extract the internal features of the data and better capture the changes of traffic flow data in the cycle. Additionally, we improve the residual structure by adding a convolutional layer to propose a correction module and use it together with the auto-correlation mechanism to jointly build an encoder and decoder to extract features from different data components (intrinsic mode functions) and fuse the extracted features for output. To meet the requirements of long-term forecasting, we set the traffic flow forecast length to 4 levels: 96, 192, 336, and 720. We validated our model using the departure statistics dataset of a taxi parking lot at Beijing Capital International Airport and achieved the best prediction performance in terms of mean squared error and mean absolute error, compared to the baseline model.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Traffic Prediction and Management Techniques · Traffic control and management · Traffic and Road Safety
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
School of Urban Rail Transportation and Logistics, Beijing Union University, Beijing 100101, China
Funding
R&D Program of Beijing Municipal Education Commission
References (41)
Show source-provided references
1. A novel fuzzy-based convolutional neural network method to traffic flow prediction with uncertain traffic accident information10.1109/ACCESS.2019.2896913 ↗
2. A novel hybrid model combining a fuzzy inference system and a deep learning method for short-term traffic flow prediction10.1016/j.knosys.2022.109760 ↗
3. Short-term traffic flow prediction using seasonal ARIMA model with limited input data10.1007/s12544-015-0170-8 ↗
4. Chen, C., Hu, J., Meng, Q., and Zhang, Y. (2011, January 5–9). Short-time traffic flow prediction with ARIMA-GARCH model. Proceedings of the 2011 4nd IEEE Conference on Intelligent Vehicles Symposium, Baden-Baden, Germany.10.1109/IVS.2011.5940418 ↗
5. Wei, Y., and Liu, H. (2022). Convolutional Long-Short Term Memory Network with Multi-Head Attention Mechanism for Traffic Flow Prediction. Sensors, 22.10.3390/s22207994 ↗
6. Zhang, L., and Lin, W. (2022, January 20–22). Calibration-free Traffic Signal Control Method Using Machine Learning Approaches. Proceedings of the 2022 International Conference on Electrical, Computer and Energy Technologies (ICECET), Prague, Czech Republic.10.1109/ICECET55527.2022.9873047 ↗
7. Gu, J., Jia, Z., Cai, T., Song, X., and Mahmood, A. (2023). Dynamic correlation adjacency-matrix-based graph neural networks for traffic flow prediction. Sensors, 23.10.3390/s23062897 ↗
8. Braz, F.J., Ferreira, J., Gonçalves, F., Weege, K., Almeida, J., Baldo, F., and Goncalves, P. (2022). Road traffic forecast based on meteorological information through deep learning methods. Sensors, 22.10.3390/s22124485 ↗
9. Highway traffic volume forecasting based on seasonal ARIMA model
10. Bai, S., Kolter, J.Z., and Koltun, V. (2018). An empirical evaluation of generic convolutional and recurrent networks for sequence modeling. arXiv.
11. Long short-term memory10.1162/neco.1997.9.8.1735 ↗
12. Cho, K., Van Merriënboer, B., Bahdanau, D., and Bengio, Y. (2014). On the properties of neural machine translation: Encoder-decoder approaches. arXiv.10.3115/v1/W14-4012 ↗
13. Vaswani, A., Shazeer, N., Parmar, N., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, L., and Polosukhin, I. (2017, January 4–9). Attention is all you need. Proceedings of the 2017 Conference on Advances in Neural Information Processing Systems, Long Beach, CA, USA.
14. The empirical mode decomposition and the Hilbert spectrum for nonlinear and non-stationary time series analysis10.1098/rspa.1998.0193 ↗
15. Variational mode decomposition10.1109/TSP.2013.2288675 ↗
16. Traffic flow forecasting by a least squares support vector machine with a fruit fly optimization algorithm10.1016/j.proeng.2016.01.234 ↗
17. A hybrid forecasting framework based on support vector regression with a modified genetic algorithm and a random forest for traffic flow prediction10.26599/TST.2018.9010045 ↗
18. Short-term traffic forecasting using self-adjusting k-nearest neighbours10.1049/iet-its.2016.0263 ↗
19. Zhang, N., Zhang, Y., and Lu, H. (2011, January 23–27). Short-term freeway traffic flow prediction combining seasonal autoregressive integrated moving average and support vector machines. Proceedings of the 90th Board Annual Conference on Transportation Research, Washington, DC, USA.
20. A sample-rebalanced outlier-rejected $ k $-nearest neighbor regression model for short-term traffic flow forecasting10.1109/ACCESS.2020.2970250 ↗
21. Deep temporal convolutional networks for short-term traffic flow forecasting10.1109/ACCESS.2019.2935504 ↗
22. Traffic flow prediction using LSTM with feature enhancement10.1016/j.neucom.2018.12.016 ↗
23. A data aggregation based approach to exploit dynamic spatio-temporal correlations for citywide crowd flows prediction in fog computing10.1007/s11042-020-10486-4 ↗
24. Spatiotemporal traffic flow prediction with KNN and LSTM10.1155/2019/4145353 ↗
25. LSTM network: A deep learning approach for short-term traffic forecast10.1049/iet-its.2016.0208 ↗
26. Liu, Y., Zheng, H., Feng, X., and Chen, Z. (2017, January 11–13). Short-term traffic flow prediction with Conv-LSTM. Proceedings of the 2017 9th International Conference on Wireless Communications and Signal Processing (WCSP), Nanjing, China.10.1109/WCSP.2017.8171119 ↗
27. Fu, R., Zhang, Z., and Li, L. (2016, January 11–13). Using LSTM and GRU neural network methods for traffic flow prediction. Proceedings of the 2016 31st Youth Academic Annual Conference of Chinese Association of Automation (YAC), Wuhan, China.10.1109/YAC.2016.7804912 ↗
28. Short-term traffic flow forecasting method with MB-LSTM hybrid network10.1109/TITS.2020.3009725 ↗
29. Informer: Beyond effificient transformer for long sequence time-series forecasting
30. Autoformer: Decomposition transformers with auto-correlation for long-term series forecasting
31. Zhou, T., Ma, Z., Wen, Q., Wang, X., Sun, L., and Jin, R. (2022). Fedformer: Frequency enhanced decomposed transformer for long-term series forecasting. arXiv.
32. A hybrid EMD-LSTM model for non-stationary wave prediction in offshore China10.1016/j.oceaneng.2022.110566 ↗
33. A hybrid deep learning framework for long-term traffic flow prediction10.1109/ACCESS.2021.3050836 ↗
34. A new crude oil price forecasting model based on variational mode decomposition10.1016/j.knosys.2020.106669 ↗
35. A hybrid stock price index forecasting model based on variational mode decomposition and LSTM network10.1007/s10489-020-01814-0 ↗
36. Wind speed forecasting based on variational mode decomposition and improved echo state network10.1016/j.renene.2020.09.109 ↗
37. Application of variational mode decomposition and chaotic grey wolf optimizer with support vector regression for forecasting electric loads10.1016/j.knosys.2021.107297 ↗
38. A hybrid short-term traffic flow multistep prediction method based on variational mode decomposition and long short-term memory model10.1155/2021/4097149 ↗
39. Hourly traffic flow forecasting using a new hybrid modelling method10.1007/s11771-022-5000-2 ↗
40. Tang, J., and Chien, Y.R. (2022). Research on Wind Power Short-Term Forecasting Method Based on Temporal Convolutional Neural Network and Variational Modal Decomposition. Sensors, 22.10.3390/s22197414 ↗
41. Cai, C., Li, Y., Su, Z., Zhu, T., and He, Y. (2022). Short-Term Electrical Load Forecasting Based on VMD and GRU-TCN Hybrid Network. Appl. Sci., 12.10.3390/app12136647 ↗