TLT: Recurrent fine-tuning transfer learning for water quality long-term prediction.

The water quality long-term prediction is essential to water environment management decisions. In recent years, although water quality prediction methods based on deep learning have achieved excellent performance in short-term prediction, these methods are unsuitable for long-term prediction because the accumulation use of short-term prediction will easily introduce noise. Furthermore, The long-term prediction task requires a large amount of data to train the model to obtain accurate prediction results. For some monitoring stations with limited historical data, it is challenging to fully exploit the performance of deep learning models. To this end, we introduce a transfer learning framework into water quality prediction to improve the prediction performance in data-constrained scenarios. We propose a deep Transfer Learning based on Transformer (TLT) model to enable time dependency perception and facilitate long-term water quality prediction. In TLT, we innovatively introduce a recurrent fine-tuning transfer learning method, which can transfer the knowledge learned from source monitoring stations to the target station, while preventing the deep learning model from overfitting the source data during the pre-training phase. So, TLT can fully exert the performance of deep learning models with limited samples. We conduct experiments on data from 120 monitoring stations in major rivers and lakes in China to verify the effectiveness of TLT. The results show that TLT can effectively improve the long-term prediction accuracy of four water quality indicators (pH, DO, NH3-N, and CODMn) from monitoring stations with limited samples.

A study on water quality prediction by a hybrid CNN-LSTM model with attention mechanism.

The water environment plays an essential role in the mangrove wetland ecosystem. Predicting water quality will help us better protect water resources from pollution, allowing the mangrove ecosystem to perform its normal ecological role. New approaches to solve such nonlinear problems need further research since the complexity of water quality data and they are easily affected by the noise. In this paper, we propose a water quality prediction model named CNN-LSTM with Attention (CLA) to predict the water quality variables. We conduct a case study on the water quality dataset of Beilun Estuary to predict pH and NH3-N. Linear interpolation and wavelet techniques are used for missing data filling and data denoising, respectively. The hybrid model CNN-LSTM is highly capable of resolving nonlinear time series prediction problems, and the attention mechanism captures longer time dependence. The experimental results show that our model outperforms other ones, and can predict with different time lags in a stable manner.

A water quality prediction method based on the multi-time scale bidirectional long short-term memory network.

As an important factor affecting the mangrove wetland ecosystem, water quality has become the focus of attention in recent years. Therefore, many studies have focused on the prediction of water quality to help establish a regulatory framework for the assessment and management of water pollution and ecosystem health. To make a more accurate and comprehensive forecast analysis of water quality, we propose a method for water quality prediction based on the multi-time scale bidirectional LSTM network. In the method, we improve data integrity and data volume through data preprocessing. And the network processes input data forward and backward and considers the dependencies at multiple time scales. Besides, we use the Box-Behnken experimental design method to adjust hyper-parameters in the process of modeling. In this study, we apply this method to the water quality prediction research of Beilun Estuary, and the performance of our proposed model is evaluated and compared with other models. The experiment results show that this model has better performance in water quality prediction than that of using LSTM or bidirectional LSTM alone. Graphical Abstract Schematic of research work.