A multi-graph spatial-temporal attention network for air-quality prediction

Abstract

Air pollution poses a grave threat to human health and everyday life. Accurate air-quality prediction plays a crucial role in effectively preventing and controlling air pollution. A multi-graph spatial-temporal attention network is proposed to predict the air quality in a given area by analyzing the interconnections between stations and the individual characteristics of each station. Firstly, this paper constructs multi-scale spatial-temporal graphs from spatial and temporal perspectives to effectively capture the interconnections between stations, thereby comprehensively understanding the spatial-temporal relationships among them. Secondly, incorporating the nonlinear temporal correlation of station data, this paper proposes a temporal multi-graph attention-fusion module to integrate information from both neighboring stations and the station itself. Finally, predictions are made using a spatial-temporal graph network. The experiments in this paper were based on air-quality data from Beijing and Tianjin, and the following experimental results were obtained. (1) The proposed model outperforms the baselines significantly in both single-step and multi-step prediction tasks. (2) The ablation study confirms that the graph constructed in this paper contributes to improving the performance of the model. (3) Comparing the performance of attention components, the proposed attention mechanism exhibits better performance than other attention mechanisms.