Time series AQI forecasting using Kalman-integrated Bi-GRU and Chi-square divergence optimization
摘要
Air pollution has become a pressing global concern, demanding accurate forecasting systems to safeguard public health. Existing AQI prediction models often falter due to missing data, high variability, and limited ability to handle distributional uncertainty. This study introduces a novel deep learning framework that integrates Kalman Attention with a Bi-Directional Gated Recurrent Unit (Bi-GRU) for robust AQI time-series forecasting. Unlike conventional attention mechanisms, Kalman Attention dynamically adjusts to data uncertainty, enhancing temporal feature weighting. Additionally, we incorporate a Chi-square Divergence-based regularization term into the loss function to explicitly minimize the distributional mismatch between predicted and actual pollutant levels–a contribution not explored in prior AQI models. Missing values are imputed using a pollutant-specific ARIMA model to preserve time-dependent trends. The proposed system is evaluated using real-world data from the U.S. Environmental Protection Agency (2022-2024) across six major pollutants (CO, NO