Prediction of Total Phosphorus Concentration in Canals by GAT-Informer Model Based on Spatiotemporal Correlations
Year: 2024
Type: article
Abstract: The accurate prediction of total phosphorus (TP) is crucial for the early detection of water quality eutrophication. However, predicting TP concentrations among canal sites is challenging due to their complex spatiotemporal dependencies. To address this issue, this study proposes a GAT-Informer prediction method based on spatiotemporal correlations to predict TP concentrations in the Beijing–Hangzhou Grand Canal Basin in Changzhou City. The method begins by creating feature sequences for each site based on the time lag relationship of total phosphorus concentration between sites. It then constructs spatiotemporal graph data by combining the real river distance between sites and the correlation of feature sequences. Next, spatial features are extracted by fusing node features using the graph attention (GAT) module. The study employs the Informer network, which uses a sparse attention mechanism to extract temporal features efficiently for simulating and predicting total phosphorus data of the sites. The model was evaluated using R2, MAE, and RMSE, with the experimental results yielding values of 0.9619, 0.1489%, and 0.1999%, respectively. The GAT-Informer model exhibits enhanced robustness and superior predictive accuracy in comparison to traditional water quality prediction models.
Source: Water
DOI: 10.3390/w17010012
Authors: Juan Huan, Xincheng Li, Jialong Yuan, Hao Zhang, Xiangen Xu, Qucheng Hu, Chen Zhang, Yixiong Fan, Wenxin Cai, Haoran Ju, Shiling Gu
Institution: Changzhou University
Language: English
Subfield: Environmental Engineering
Field: Environmental Science
Domain: Physical Sciences
Open Access status: gold
Estimated APC paid: $2,955
Awards: G137640429, G1403970619, G1778328533, G1931049499, G2762933807, G2776265683, G3048778376, G3251932416, G3687360510, G3938299897, G3974116165, G40428890, G4150547413, G4399039219, G4447829945, G4470532398, G4999984342, G5246369269, G5651980981, G6202901884, G6263537644, G6512362477, G6909277965, G6914802981, G6962327096, G7110926564, G7253650025, G762139585, G7937406410, G8066447514, G8321683214, G8574040224
OpenAlex ID: W4405736817