Short-Term Traffic Flow Forecasting Method Based on Secondary Decomposition and Conventional Neural Network–Transformer

Qichun Bing,Panpan Zhao,Canzheng Ren,Xueqian Wang,Yiming Zhao
DOI: https://doi.org/10.3390/su16114567
IF: 3.9
2024-05-29
Sustainability
Abstract:Because of the random volatility of traffic data, short-term traffic flow forecasting has always been a problem that needs to be further researched. We developed a short-term traffic flow forecasting approach by applying a secondary decomposition strategy and CNN–Transformer model. Firstly, traffic flow data are decomposed by using a Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN) algorithm, and a series of intrinsic mode functions (IMFs) are obtained. Secondly, the IMF1 obtained from the CEEMDAN is further decomposed into some sub-series by using Variational Mode Decomposition (VMD) algorithm. Thirdly, the CNN–Transformer model is established for each IMF separately. The CNN model is employed to extract local spatial features, and then the Transformer model utilizes these features for global modeling and long-term relationship modeling. Finally, we obtain the final results by superimposing the forecasting results of each IMF component. The measured traffic flow dataset of urban expressways was used for experimental verification. The experimental results reveal the following: (1) The forecasting performance achieves remarkable improvement when considering secondary decomposition. Compared with the VMD-CNN–Transformer, the CEEMDAN-VMD-CNN–Transformer method declined by 25.84%, 23.15% and 22.38% in three-step-ahead forecasting in terms of MAPE. (2) It has been proven that our proposed CNN–Transformer model could achieve more outstanding forecasting performance. Compared with the CEEMDAN-VMD-CNN, the CEEMDAN-VMD-CNN–Transformer method declined by 13.58%, 11.88% and 11.10% in three-step-ahead forecasting in terms of MAPE.
environmental sciences,environmental studies,green & sustainable science & technology
What problem does this paper attempt to address?
The paper attempts to address the issue of stochastic volatility in short-term traffic flow prediction. Specifically, due to the high randomness and volatility of traffic data, short-term traffic flow prediction becomes very challenging. To tackle this problem, the authors propose a short-term traffic flow prediction method based on a secondary decomposition strategy and a Convolutional Neural Network-Transformer (CNN–Transformer) model. This method is implemented through the following steps: 1. **Initial Decomposition**: First, the Complete Ensemble Empirical Mode Decomposition with Adaptive Noise (CEEMDAN) algorithm is used to decompose the traffic flow data, resulting in a series of Intrinsic Mode Functions (IMFs). 2. **Secondary Decomposition**: The highest frequency component IMF1 obtained from CEEMDAN is further decomposed using the Variational Mode Decomposition (VMD) algorithm to extract more detailed sub-sequences. 3. **Model Building**: A CNN–Transformer model is established for each IMF. The CNN model is used to extract local spatial features, while the Transformer model utilizes these features for global modeling and long-term relationship modeling. 4. **Result Aggregation**: Finally, the prediction results of each IMF component are aggregated to obtain the final prediction result. Through the above method, the authors aim to improve the accuracy of short-term traffic flow prediction, especially in handling the randomness and volatility of the data. Experimental results show that this method significantly improves the Mean Absolute Percentage Error (MAPE) in three-step predictions compared to other methods.