Dynamic Graph Neural Network with Adaptive Edge Attributes for Air Quality Prediction: A Case Study in China.

Jing Xu,Shuo Wang,Na Ying,Xiao,Jiang Zhang,Zhiling Jin,Yun Cheng,Gangfeng Zhang
DOI: https://doi.org/10.1016/j.heliyon.2023.e17746
IF: 3.776
2023-01-01
Heliyon
Abstract:Air quality prediction is a typical Spatiotemporal modeling problem, which always uses different components to handle spatial and temporal dependencies in complex systems separately. Previous models based on time series analysis and recurrent neural network (RNN) methods have only modeled time series while ignoring spatial information. Previous graph convolution neural networks (GCNs) based methods usually require providing spatial correlation graph structure of observation sites in advance. The correlations among these sites and their strengths are usually calculated using prior information. However, due to the limitations of human cognition, limited prior information cannot reflect the real station-related structure or bring more effective information for accurate prediction. To this end, we propose a novel Dynamic Graph Neural Network with Adaptive Edge Attributes (DGN-AEA) on the message passing network, which generates the adaptive bidirected dynamic graph by learning the edge attributes as model parameters. Unlike prior information to establish edges, our method can obtain adaptive edge information through end-to-end training without any prior information. Thus reducing the complexity of the problem. Besides, the hidden structural information between the stations can be obtained as model by-products, which can help make some subsequent decision-making analyses. Experimental results show that our model received state-of-the-art performance than other baselines.
What problem does this paper attempt to address?