Predict water quality using an improved deep learning method based on spatiotemporal feature correlated: a case study of the Tanghe Reservoir in China

Min Han,Ziyan Su,Xiaodong Na
DOI: https://doi.org/10.1007/s00477-023-02405-4
IF: 3.821
2023-03-24
Stochastic Environmental Research and Risk Assessment
Abstract:The ecological health of water quality is an essential factor in the survival and development of human beings. By setting up multiple sensors in the water body, people observe the factual state of water quality, and collect a large number of sequence data containing time information. In order to effectively utilize water quality time series data and predict future water quality changes, this paper proposes an improved deep learning method based on spatiotemporal feature correlated, called the convolution recurrent basis expansion analysis architecture. The model improves the original model by using the ability of convolutional neural network structure to extract spatial features and the continuous memory ability of recurrent neural network structure. Then, the physical prior knowledge is integrated into the proposed network to limit the unreasonable results predicted in the feasible solution space to further improve the learning efficiency of the model. For specific research objects, the model can mine multi-dimensional features in water quality sequences at different depths, and learn the temporal features of the sequences hierarchically. In order to verify the performance of the proposed model, we apply the proposed model to the water quality dataset collected in Tanghe Reservoir for simulation. The results of the study demonstrate that our model is suitable for water quality prediction and has distinct advantages over traditional neural networks. Our model will make accurate predictions of future changes in water quality and provide technical support in the refined water quality automatic monitoring system.
environmental sciences,engineering, environmental,water resources, civil,statistics & probability
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to predict water quality changes by using an improved deep - learning method, especially in the case study of Tanghe Reservoir. Specifically, the paper aims to effectively utilize water quality time - series data and predict future water quality changes by proposing a Convolution Recurrent Basis Expansion Analysis Architecture (CRn - BEATS) based on spatio - temporal feature correlation. This method combines the ability of Convolutional Neural Network (CNN) to extract spatial features and the continuous memory ability of Recurrent Neural Network (RNN), and integrates physical prior knowledge to limit unreasonable result predictions, thereby improving the learning efficiency of the model. In addition, this model can mine multi - dimensional features in water quality sequences at different depths and learn the time features of the sequences hierarchically. By conducting simulations on the water quality data set collected from Tanghe Reservoir, the performance of the proposed model has been verified. The results show that this model has significant advantages in water quality prediction, can accurately predict future water quality changes, and provides technical support for a refined automatic water quality monitoring system.