Download pdf
Research on traffic flow prediction method based on LSTM model and PSO-LSTM model
- 1 School of Sports Engineering, Beijing Sport University, Beijing 100080, China
* Author to whom correspondence should be addressed.
Abstract
With the acceleration of urbanization, the number of cars owned by residents has also significantly increased. The contradiction between the number of cars and the road carrying capacity has become increasingly severe, resulting in very serious congestion. This paper selects road data from 0:00 to 10:00 every morning in Beijing from April 2nd to April 12th, 2016, and uses the average speed of vehicles as a variable to measure road congestion. Based on these data, this article uses LSTM models to predict the speed of vehicles on two roads representing main and non main roads. Research has found that LSTM model has good predictive performance for the speed of vehicles on two roads and is very accurate in predicting the data trend. However, the accuracy of LSTM model in predicting non periodic and highly discrete data is not ideal. This article also uses the PSO-LSTM model to predict one of the roads, and the results show that the model is more accurate than the LSTM model in predicting non periodic data. The prediction of short-term traffic flow will greatly help the transportation department coordinate and manage resources, alleviate traffic congestion, and facilitate residents' travel.
Keywords
LSTM, traffic flow prediction, average speed, PSO-LSTM.
[1]. Yang N 2014 Analysis of the hazards and causes of urban traffic congestion. China Market, (42), 100-101.
[2]. Cao C 2020 Research on algorithm of vehicle flow monitoring and prediction based on deep learning. Jimei University.
[3]. Sun H 2020 Research on traffic signal control method based on deep reinforcement learning. Nanjing University.
[4]. Xiang Q 2021 Research on deep reinforcment learning for urban adaptive traffic signal control. Southeast University.
[5]. Shen L, Lu Y and Guo J 2021 Adaptability of Kalman Filter for short-time traffic flow forecasting on national and provincial highways. Journal of Transport Information and Safety. 39(05), 117-27.
[6]. Nihan N L and Holmesland K O 1980 Use of the Box and Jenkins time series technique in traffic forecasting. Transportation, 9(2), 125-143.
[7]. Li T, Ni A, Zhang C, Xiao G and Gao L 2021 Short-term traffic congestion prediction with Conv iLSTM considering spatiotemporal features. IET Intelligent Transport Systems, 14(1).
[8]. Rasheed F, Yau K, and Low Y C 2020 Deep reinforcement learning for traffic signal control under disturbances: A case study on Sunway city Malaysia. Future Generation Computer Systems.
[9]. Smith B L and Demetsky M J 1997 Traffic flow forecasting: comparison of modeling approaches. Journal of transportation engineering, 123(4), 261-266.
[10]. Vanajakshi L and Rilett L R 2004 A comparison of the performance of artificial neural networks and support vector machines for the prediction of traffic speed. IEEE Intelligent Vehicles Symposium. IEEE, 194-199.
[11]. Liu Q, Zhai J, Zhang Z, Zhong S, Zhou Q, Zhang P Xu J 2018 A brief overview of deep reinforcement learning. Chinese Journal of Computers, 1-27.
[12]. Zhang W, Yu Y, Qi Y, Shu F and Wang Y 2019 Short-term traffic flow prediction based on spatiotemporal analysis and CNN deep learning. Transportmetrica, 15(2), 1688-711.
[13]. Zhao M 2021 Prediction for short-term passenger flow of subway based on improved PSO Optimized LSTM neural network. Shijiazhuang Tiedao University.
[14]. Zhu H 2024 Intercity railway passenger flow prediction method based on improved LSTM model. The technology and management of transportation system, 5(13), 4-7.
[15]. Zhang G and Jin H 2021 Passenger flow prediction of urban rail transit stations based on improved PSO-LSTM model. Computer Applications and Software, 3(812), 110-114+134.
Cite this article
Xu,F. (2024). Research on traffic flow prediction method based on LSTM model and PSO-LSTM model. Applied and Computational Engineering,101,154-163.
Data availability
The datasets used and/or analyzed during the current study will be available from the authors upon reasonable request.
Disclaimer/Publisher's Note
The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of EWA Publishing and/or the editor(s). EWA Publishing and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.
About volume
Volume title: Proceedings of the 2nd International Conference on Machine Learning and Automation
© 2024 by the author(s). Licensee EWA Publishing, Oxford, UK. This article is an open access article distributed under the terms and
conditions of the Creative Commons Attribution (CC BY) license. Authors who
publish this series agree to the following terms:
1. Authors retain copyright and grant the series right of first publication with the work simultaneously licensed under a Creative Commons
Attribution License that allows others to share the work with an acknowledgment of the work's authorship and initial publication in this
series.
2. Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the series's published
version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgment of its initial
publication in this series.
3. Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) prior to and
during the submission process, as it can lead to productive exchanges, as well as earlier and greater citation of published work (See
Open access policy for details).