Traffic Volume Prediction using Memory-Based Recurrent Neural Networks: A comparative analysis of LSTM and GRU

Lokesh Chandra Das

2023-03-22

Abstract:Predicting traffic volume in real-time can improve both traffic flow and road safety. A precise traffic volume forecast helps alert drivers to the flow of traffic along their preferred routes, preventing potential deadlock situations. Existing parametric models cannot reliably forecast traffic volume in dynamic and complex traffic conditions. Therefore, in order to evaluate and forecast the traffic volume for every given time step in a real-time manner, we develop non-linear memory-based deep neural network models. Our extensive experiments run on the Metro Interstate Traffic Volume dataset demonstrate the effectiveness of the proposed models in predicting traffic volume in highly dynamic and heterogeneous traffic environments.

Machine Learning

What problem does this paper attempt to address?

The main objective of this paper is to address the issue of real-time traffic flow prediction. Specifically, the authors attempt to improve the accuracy of traffic flow prediction by developing Recurrent Neural Network (RNN) models based on memory mechanisms. The paper mentions that existing parametric models cannot reliably predict traffic flow under dynamic and complex traffic conditions, thus necessitating nonlinear memory mechanisms to enhance prediction performance. The authors employed two types of Recurrent Neural Network models: Long Short-Term Memory (LSTM) networks and Gated Recurrent Units (GRU), and conducted extensive experiments on the "Metro Interstate Traffic Volume" dataset. These models can predict future traffic flow based on historical observation data, thereby helping to improve traffic mobility and road safety. The experimental results indicate that both models can effectively predict traffic flow in highly dynamic and heterogeneous traffic environments, and they exhibit different advantages under various settings. For instance, the GRU model performs better when a shorter historical sequence is required, whereas the LSTM model is superior in handling complex datasets that require longer historical sequences. Overall, these models can significantly improve prediction accuracy, contributing to the alleviation of traffic congestion.

Traffic Volume Prediction using Memory-Based Recurrent Neural Networks: A comparative analysis of LSTM and GRU

A Survey of Traffic Flow Prediction Methods Based on Long Short-Term Memory Networks

Urban Traffic Flow Forecasting Based on Memory Time-Series Network.

Using LSTM and GRU neural network methods for traffic flow prediction

Short-Term Traffic Flow Prediction for Hybrid Time Series Decomposition Analysis and LSTM Neural Network

Traffic Flow Prediction Based on Deep Neural Networks

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

Traffic flow prediction using bi-directional gated recurrent unit method

Development and Evaluation of Recurrent Neural Network based Models for Hourly Traffic Volume and AADT Prediction

Deep Learning with Bidirectional Long Short-Term Memory for Traffic Flow Prediction

An accurate traffic flow prediction using long-short term memory and gated recurrent unit networks

Predictive Modelling of Traffic Flow with Deep Learning Techniques

STLGRU: Spatio-Temporal Lightweight Graph GRU for Traffic Flow Prediction

Long short-term memory (LSTM) recurrent neural network (RNN) based traffic forecasting for intelligent transportation

LSTM Deep Learning Long-Term Traffic Volume Prediction Model Based on Markov State Description

LSTM Network: a Deep Learning Approach for Short-Term Traffic Forecast

Predicting Short-Term Traffic Flow by Long Short-Term Memory Recurrent Neural Network.

Short-Term Traffic Flow Prediction for Urban Road Sections Based on Time Series Analysis and LSTM_BILSTM Method

Short-term Traffic Flow Prediction Based on Recurrent Neural Network

A Novel Confined Attention Mechanism Driven Bi-GRU Model for Traffic Flow Prediction

Research on Short-term Traffic Flow Prediction Based on KNN-GRU