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.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.