Download pdf
Traffic Prediction Based on the GRU Neural Network
- 1 The Affiliated International School of Shenzhen University
- 2 Wuhan No.6 Middle School International Department
- 3 Huili School Shanghai
* Author to whom correspondence should be addressed.
Abstract
Traffic prediction is an important and challenging task in the last decade. In this project, we trained a GRU neural network to predict traffic at four junctions using a normalization and differencing transform to achieve a stationary time series. Given the varying trends and seasonality observed at each junction, we employed unique strategies to make each stationary, and evaluated model performance using the root mean squared error. Additionally, we plotted predicted values against original test data. Data analysis reveals that the number of vehicles at Junction 1 is increasing at a faster rate than at Junctions 2 and 3, while sparse data precludes conclusions regarding Junction 4. Moreover, Junction 1 exhibits stronger weekly and hourly seasonality, while the other junctions display a more linear trend. The results of this study can effectively inform vehicle flow rate prediction. However, GRU models may suffer from slow convergence rates and low learning efficiency, leading to prolonged training times and underfitting. Despite these limitations, the use of GRU models remains the most accurate, achievable, and straightforward method for calculating traffic predictions.
Keywords
traffic prediction, machine learning, GRU
[1]. Thomson J M 1998 Reflections on the economics of traffic congestion Journal of transport economics and policy 93-112
[2]. Chiou J M 2012 Dynamical functional prediction and classification, with application to traffic flow prediction
[3]. Dey R Salem F M 2017 Gate-variants of gated recurrent unit (GRU) neural networks 2017 IEEE 60th international midwest symposium on circuits and systems (MWSCAS) IEEE 1597-1600
[4]. 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). IEEE 324-328
[5]. Bansal T Belanger D McCallum A 2016 Ask the gru: Multi-task learning for deep text recommendations proceedings of the 10th ACM Conference on Recommender Systems 107-114
[6]. Yu Y Si X Hu C et al. 2019 A review of recurrent neural networks: LSTM cells and network architectures Neural computation 31(7): 1235-1270
[7]. Kaggle 2021 Traffic prediction dataset https://www.kaggle.com/datasets/fedesoriano/traffic-prediction-dataset
[8]. Köksoy O 2006 Multiresponse robust design: Mean square error (MSE) criterion Applied Mathematics and Computation 175(2): 1716-1729.
[9]. Yu Q Chang C S Yan J L et al. 2019 Semantic segmentation of intracranial hemorrhages in head CT scans 2019 IEEE 10th International Conference on Software Engineering and Service Science (ICSESS) IEEE 112-115.
[10]. Zhang Z 2018 Improved adam optimizer for deep neural networks 2018 IEEE/ACM 26th international symposium on quality of service (IWQoS) Ieee 1-2
Cite this article
Ding,H.;Li,Z.;Su,N. (2023). Traffic Prediction Based on the GRU Neural Network. Applied and Computational Engineering,8,287-291.
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 2023 International Conference on Software Engineering and Machine Learning
© 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).