File(s) under permanent embargo
A novel hybrid deep learning model with ARIMA Conv-LSTM networks and shuffle attention layer for short-term traffic flow prediction
journal contributionposted on 2023-08-14, 06:27 authored by AR Sattarzadeh, RJ Kutadinata, Pubudu PathiranaPubudu Pathirana, Van Thanh HuynhVan Thanh Huynh
Traffic flow prediction requires learning of nonlinear spatio-temporal dynamics which becomes challenging due to its inherent nonlinearity and stochasticity. Addressing this shortfall, we propose a new hybrid deep learning model based on an attention mechanism that uses multi-layered hybrid architectures to extract spatial–temporal, nonlinear characteristics. Firstly, by designing the autoregressive integral moving average (ARIMA) model, trends and linear regression are extracted; then, integration of convolutional neural network (CNN) and long short-term memory (LSTM) networks leads to better understanding of the model's correlations, serving for more accurate traffic prediction. Secondly, we develop a shuffle attention-based (SA) Conv-LSTM module to determine significance of flow sequences by allocating various weights. Thirdly, to effectively analyse short-term temporal dependencies, we utilise bidirectional LSTM (Bi-LSTM) components to capture periodic features. Experimental results illustrate that our Shuffle Attention ARIMA Conv-LSTM (SAACL) model provides better prediction than other comparable methods, particularly for short-term forecasting, using PeMS datasets.
JournalTransportmetrica A: Transport Science
PublisherInforma UK Limited
intelligent transportation systemstraffic flow predictiondeep learning ARIMA-LSTMConv-LSTMshuffle attention layers4005 Civil Engineering40 Engineering3509 Transportation, logistics and supply chains4005 Civil engineeringTransportation and Freight Services not elsewhere classifiedCivil Engineering not elsewhere classified