EAAI Journal 2026 Journal Article
Improving low air quality prediction accuracy: An optimal variational mode decomposition and state space-temporal fusion transformers framework
- Yong Zhang
- Weiting Zhao
- Fenghong Wang
- Jingyu Zhang
- Feng Xu
- Shuhao Jiang
- Ming-Lang Tseng
The increasingly serious issues of urban air pollution monitoring and quality prediction could help reveal patterns of regional air quality changes and provide solutions for enhancing air quality monitoring. This study aims to address the problems of incomplete feature extraction from various influencing factors and low air quality prediction accuracy. The incomplete feature extraction refers to the inability to fully capture useful information from the data, and it usually due to the not well-suited methods for the dataset or the suboptimal parameter selection in decomposition methods. To overcome these limitations, this study proposes a hybrid prediction model based on the optimal Variational Mode Decomposition (VMD) method combined with the State Space-Temporal Fusion Transformers(SS-TFT) framework. First, the optimal VMD parameters were selected using the beluga sample entropy algorithm to enhance decomposition efficiency and improve feature extraction accuracy from the original air monitoring dataset. Second, the SS-TFT framework, which integrates the state space model with the deep learning-based Temporal Fusion Transformers (TFT), is designed to capture both linear and nonlinear features in decomposed subsequences. The proposed hybrid prediction model demonstrated outstanding performance in error evaluation metrics, achieving the lowest values for Mean Squared Error, Root Mean Squared Error, and Mean Absolute Error at 85. 800, 9. 263, and 6. 994, respectively. The model exhibited the best fit, effectively reducing the impact of complex nonlinear and non-stationary data on the experimental results and improving the accuracy of air quality prediction.