Abstract
The prediction of river water quality is key in water resource management. Data-driven machine learning models have been widely used for predicting river water quality. However, these models seldom consider the physical mechanisms of water quality variation, which degrades the accuracy and stability of the prediction results. Hence, we develop a physics-guided spatio–temporal neural network (PGSTNN) model to predict a critical parameter for water quality assessment, i.e. dissolved oxygen. Physical information regarding spatio–temporal interactions in a hydrological network is explicitly considered to construct the architecture of PGSTNN. Two physical rules of dissolved oxygen variation (i.e. Henry’s law and power-scaling law) are established for the loss function of PGSTNN to guarantee the physical consistency of the prediction results. Experiments on the 2020–2021 water quality dataset in Atlanta, USA show that PGSTNN outperforms seven baseline neural network models in terms of prediction accuracy and stability. PGSTNN typically brings at least 10% accuracy (e.g. root mean square error and mean absolute error) improvement over the comparison methods. The proposed PGSTNN may not only improve the emergency response ability of water resource management, but also provide useful ideas for integrating scientific knowledge with machine learning.
Talk to us
Join us for a 30 min session where you can share your feedback and ask us any queries you have
More From: International Journal of Geographical Information Science
Disclaimer: All third-party content on this website/platform is and will remain the property of their respective owners and is provided on "as is" basis without any warranties, express or implied. Use of third-party content does not indicate any affiliation, sponsorship with or endorsement by them. Any references to third-party content is to identify the corresponding services and shall be considered fair use under The CopyrightLaw.