With the increasing demand for intelligent transportation systems, short-term traffic flow prediction has become an important research direction. The memory unit of a Long Short-Term Memory (LSTM) neural network can store data characteristics over a certain period of time, hence the suitability of this network for time series processing. This paper uses an improved Gate Recurrent Unit (GRU) neural network to study the time series of traffic parameter flows. The LSTM short-term traffic flow prediction based on the flow series is first investigated, and then the GRU model is introduced. The GRU can be regarded as a simplified LSTM. After extracting the spatial and temporal characteristics of the flow matrix, an improved GRU with a bidirectional positive and negative feedback called the Bi-GRU prediction model is used to complete the short-term traffic flow prediction and study its characteristics. The Rectified Adaptive (RAdam) model is adopted to improve the shortcomings of the common optimizer. The cosine learning rate attenuation is also used for the model to avoid converging to the local optimal solution and for the appropriate convergence speed to be controlled. Furthermore, the scientific and reliable model learning rate is set together with the adaptive learning rate in RAdam. In this manner, the accuracy of network prediction can be further improved. Finally, an experiment of the Bi-GRU model is conducted. The comprehensive Bi-GRU prediction results demonstrate the effectiveness of the proposed method.


    Access

    Check access

    Check availability in my library

    Order at Subito €


    Export, share and cite



    Title :

    A Short-Term Traffic Flow Prediction Model Based on an Improved Gate Recurrent Unit Neural Network


    Contributors:


    Publication date :

    2022-09-01


    Size :

    2055884 byte




    Type of media :

    Article (Journal)


    Type of material :

    Electronic Resource


    Language :

    English



    Short-term traffic flow prediction with LSTM recurrent neural network

    Kang, Danqing / Lv, Yisheng / Chen, Yuan-yuan | IEEE | 2017


    Short-term traffic flow prediction method based on graph convolution recurrent neural network

    GU JUNHUA / GUO RUIZHE / HE WENYING et al. | European Patent Office | 2024

    Free access


    Time Slot Recurrent Neural Networks for Short-Term Traffic Flow Prediction

    Qu, Licheng / Qie, Liyuan / Li, Xinze et al. | IEEE | 2022


    Prediction model for short‐term traffic flow based on a K‐means‐gated recurrent unit combination

    Sun, Zhaoyun / Hu, Yuanjiao / Li, Wei et al. | Wiley | 2022

    Free access