Predicting Air Quality Using a Multi-Scale Spatiotemporal Graph Attention Network

Xinmeng Zhou,Jingyi Wang,Junyi Wang,Qingfeng Guan
DOI: https://doi.org/10.1016/j.ins.2024.121072
IF: 8.1
2024-01-01
Information Sciences
Abstract:As urbanization accelerates, air quality has become a pressing concern. Accurate air quality prediction is essential for informed governmental decision-making and for protecting public health. Variations in air quality are influenced by complex multi-scale spatiotemporal processes. Existing research primarily relies on capturing single spatiotemporal features of air quality to predict changes. Meanwhile, when constructing spatiotemporal dynamic graphs, the inherent characteristics of the input data and the comprehensive effects of both global and local influences are not fully considered. To address these problems, we propose a graph-attention-based approach, named Multi-scale Spatiotemporal Graph Attention Network (MSTGAN). MSTGAN addresses the intricate spatiotemporal patterns of air quality across various scales through three key components: (1) a multistation transformer to model the temporal patterns of air quality at individual monitoring stations; (2) a bilinear spatiotemporal attention mechanism to capture the spatiotemporal dynamic global dependencies among all stations in a region; and (3) a set of spatiotemporal dependence graph-coupled Chebyshev graph convolution gate recurrent units to extract and aggregate the local spatiotemporal features of interrelated stations. Experiments conducted on three real-world datasets demonstrated that MSTGAN achieved significant improvements of 4.2%, 3.9%, and 7.8% in the mean absolute error, root mean square error, and R2 evaluation metrics, respectively, compared to seven state-of-the-art time-series forecasting methods. This code is publicly available at https://github.com/HPSCIL/MSTGAN-airquality-prediction.
What problem does this paper attempt to address?