Feature Extraction and Prediction of Water Quality Based on Candlestick Theory and Deep Learning Methods

Abstract

In environmental hydrodynamics, a research topic that has gained popularity is the transmission and diffusion of water pollutants. Various types of change processes in hydrological and water quality are directly related to meteorological changes. If these changing characteristics are classified effectively, this will be conducive to the application of deep learning theory in water pollution simulation. When periodically monitoring water quality, data were represented with a candlestick chart, and different classification features were displayed. The water quality data from the research area from 2012 to 2019 generated 24 classification results in line with the physics laws. Therefore, a deep learning water pollution prediction method was proposed to classify the changing process of pollution to improve the prediction accuracy of water quality, based on candlestick theory, visual geometry group, and gate recurrent unit (CT-VGG-GRU). In this method, after the periodic changes of water quality were represented by candlestick graphically, the features were extracted by the VGG network based on its advantages in graphic feature extraction. Then, this feature and other scenario parameters were fused as the input of the time series network model, and the pollutant concentration sequence at the predicted station constituted the output of the model. Finally, a hybrid model combining graphical and time series features was formed, and this model used continuous time series data from multiple stations on the Lijiang River watershed to train and validate the model. Experimental results indicated that, compared with other comparison models, such as the back propagation neural network (BPNN), support vector regression (SVR), GRU, and VGG-GRU, the proposed model had the highest prediction accuracy, especially for the prediction of extreme values. Additionally, the change trend of water pollution was closer to the real situation, which indicated that the process change information of water pollution could be fully extracted by the CT-VGG-GRU model based on candlestick theory. For the water quality indicators DO, CODMn, and NH3-N, the mean absolute errors (MAE) were 0.284, 0.113, and 0.014, the root mean square errors (RMSE) were 0.315, 0.122, and 0.016, and the symmetric mean absolute percentage errors (SMAPE) were 0.022, 0.108, and 0.127, respectively. The established CT-VGG-GRU model achieved superior computational performance. Using the proposed model, the classification information of the river pollution process could be obtained effectively and the time series information could also be retained, which made the application of the deep learning model to the transmission and diffusion process of river water pollution more explanatory. The proposed model can provide a new method for water quality prediction.