Long short-term memory network-based wastewater quality prediction model with sparrow search algorithm

The wastewater treatment process is characterized by uncertainty, non-linearity, time delay and complexity, and is susceptible to many dynamic factors. Since some key water quality parameters are not available in real time, a Long Short-Term Memory (LSTM) network water quality prediction model based on sparrow search algorithm (SSA-LSTM) and attention mechanism is proposed to solve the problem. In this model, we take historical data as input, constructs models to learn the feature of internal dynamic changes, introduces the attention mechanism, assigns different weights to the hidden state of the LSTM network by mapping weightings with the learning parameter matrix, and uses the SSA to select the optimal hyperparameters for prediction. As high-latitude feature vectors are subject to the curse of dimension, a PCA-LSTM model is further proposed to apply the Principal Component Analysis (PCA) method to the SSA-LSTM model to reduce the dimensionality of the original data. The SSA-LSTM model without the PCA method (NPCA-LSTM) and the PCA-LSTM model are applied to predict wastewater quality and the PCA-LSTM model shows higher predictive ability.