Short-Term Traffic Flow Prediction via Improved Mode Decomposition and Self-Attention Mechanism Based Deep Learning Approach

Abstract

Short-term traffic flow prediction (STFP) is one of the key technologies in Intelligence Transportation System (ITS). With the development of artificial intelligence technology, deep learning has been employed to STFP and certain achievements have been obtained. However, further improvement of STFP accuracy is constrained by the presence of noise, modal aliasing in the modal components and insufficient feature extraction of neural networks. Therefore, to more accurately capture the changing trend of traffic flow in different time periods, and increase the reliability and interpretability of the prediction models, improved variational mode decomposition (IVMD) and self-attention mechanism (SAM) based hybrid convolutional neural network (CNN) and long short term memory network (LSTM) are proposed for STFP in this study, benefited from the modal decomposition of IVMD and redistribution of the neural weights in CNN-LSTM networks by SAM, the proposed IVMD-CNN-LSTM-SAM effectively suppresses the influence of randomness and volatility of traffic flow, and accurately predict the short-term lane occupancy in the designated area. Furthermore, this study analyses the mechanism of SAM and IVMD in STFP, and proves the superiority of IVMD and SAM in signal decomposition with physical significance and weight allocation respectively. Therefore, the interpretability of the proposed neural network model is significantly improved. Eleven competitive neural networks forecasting methods are used as benchmarks and numerical studies based on actual traffic flow validate the accuracy and stability of the proposed IVMD-CNN-LSTM-SAM. Mean absolute percentage error and goodness of fit of the proposed model are 0.81% and 0.9968, respectively.