Open Access iconOpen Access



Air Pollution Prediction Via Graph Attention Network and Gated Recurrent Unit

Shun Wang1, Lin Qiao2, Wei Fang3, Guodong Jing4, Victor S. Sheng5, Yong Zhang1,*

1 Beijing Key Laboratory of Multimedia and Intelligent Software Technology, Beijing Artificial Intelligence Institute, the Faculty of Information Technology, Beijing University of Technology, Beijing, 100124, China
2 Beijing Meteorological Observatory, Beijing, 100089, China
3 Nanjing University of Information Science & Technology, Nanjing, 210044, China
4 China Meteorological Administration Training Centre, Beijing, 100081, China
5 Texas Tech University, Lubbock, TX79409, United States

* Corresponding Author: Yong Zhang. Email: email

Computers, Materials & Continua 2022, 73(1), 673-687.


PM2.5 concentration prediction is of great significance to environmental protection and human health. Achieving accurate prediction of PM2.5 concentration has become an important research task. However, PM2.5 pollutants can spread in the earth’s atmosphere, causing mutual influence between different cities. To effectively capture the air pollution relationship between cities, this paper proposes a novel spatiotemporal model combining graph attention neural network (GAT) and gated recurrent unit (GRU), named GAT-GRU for PM2.5 concentration prediction. Specifically, GAT is used to learn the spatial dependence of PM2.5 concentration data in different cities, and GRU is to extract the temporal dependence of the long-term data series. The proposed model integrates the learned spatio-temporal dependencies to capture long-term complex spatio-temporal features. Considering that air pollution is related to the meteorological conditions of the city, the knowledge acquired from meteorological data is used in the model to enhance PM2.5 prediction performance. The input of the GAT-GRU model consists of PM2.5 concentration data and meteorological data. In order to verify the effectiveness of the proposed GAT-GRU prediction model, this paper designs experiments on real-world datasets compared with other baselines. Experimental results prove that our model achieves excellent performance in PM2.5 concentration prediction.


Cite This Article

APA Style
Wang, S., Qiao, L., Fang, W., Jing, G., Sheng, V.S. et al. (2022). Air pollution prediction via graph attention network and gated recurrent unit. Computers, Materials & Continua, 73(1), 673-687.
Vancouver Style
Wang S, Qiao L, Fang W, Jing G, Sheng VS, Zhang Y. Air pollution prediction via graph attention network and gated recurrent unit. Comput Mater Contin. 2022;73(1):673-687
IEEE Style
S. Wang, L. Qiao, W. Fang, G. Jing, V.S. Sheng, and Y. Zhang "Air Pollution Prediction Via Graph Attention Network and Gated Recurrent Unit," Comput. Mater. Contin., vol. 73, no. 1, pp. 673-687. 2022.

cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 1600


  • 891


  • 0


Share Link