动态构建图结构提升多站点污染预测精度
Auto-Configured Explainable Graph Neural Networks for Multi-Site Pollution Prediction

- 用监督学习混淆矩阵自动生成图结构,捕捉类别间关系
- 混合损失函数使模型在多步预测中对PM1、PM2.5、PM10预测误差更低
- 支持可解释性分析,适合需要透明决策的环境监测场景
准确预测颗粒物(PM)浓度对缓解空气污染至关重要。图神经网络(GNN)能有效建模时空依赖关系,但预设图结构限制了适应性,且部分数据集增加了学习难度。本文提出一种基于监督学习过程混淆矩阵的图构造方法,动态捕捉类别间关系。同时,采用结合能量距离与Huber损失的混合损失函数,缓解梯度消失问题,提升训练稳定性。在犹他大学盐湖城AirU空气质量监测网络数据上,评估了五种GNN模型:图卷积网络(GCNs)、简单图卷积网络(SGConv)、图同构网络(GINs)、图注意力网络(GATs)和GraphSage。单步与多步预测结果表明,GraphSage在不同时间跨度下对PM₁、PM₁₀和PM₂.₅的预测精度最高。此外,使用GNNExplainer与PGExplainer分析特征重要性与图结构,保障模型可解释性。实验显示,该方法显著优于传统机器学习与深度学习模型(如Prophet、LSTM、GRU)。
原文摘要 · Abstract (English)
Accurate particulate matter (PM) prediction is crucial for mitigating air pollution. Graph Neural Networks (GNNs) effectively model spatiotemporal dependencies, but predefined graphs limit adaptability, and some datasets complicate learning. This study introduces a graph construction method based on a confusion matrix from a supervised learning process to dynamically capture inter-class relationships. Additionally, a hybrid loss function that combines energy distance and Huber loss is applied to address the vanishing gradient problem and improve learning stability. The approach is evaluated using air pollution data from the University of Utah AirU Pollution Monitoring Network in Salt Lake City, UT, with five GNN models: Graph Convolutional Networks (GCNs), Simple Graph Convolutional Networks (SGConv), Graph Isomorphism Networks (GINs), Graph Attention Networks (GATs), and GraphSage. The experimental results of single- and multistep predictions confirm that GraphSage achieves the highest accuracy in predicting the concentrations of PM${1}$, PM${10}$, and PM$_{2.5}$ over different time horizons. Furthermore, {\color{black} GNNExplainer (Graph Neural Network Explainer) and PGExplainer (Probabilistic Graph Explainer)} are applied to interpret feature importance and graph structure, ensuring model transparency. Results show improved prediction accuracy, with GNN models outperforming traditional machine learning \textcolor{black}{and deep learning models (i.e., Prophet, Long short-term memory, Gated recurrent units} in air pollution forecasting.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。