An Empirical Modal Decomposition-Improved Whale Optimization Algorithm-Long Short-Term Memory Hybrid Model for Monitoring and Predicting Water Quality Parameters

Abstract

Prediction of water quality parameters is a significant aspect of contemporary green development and ecological restoration. However, the conventional water quality prediction models have limited accuracy and poor generalization capability. This study aims to develop a dependable prediction model for ammonia nitrogen concentration in water quality parameters. Based on the characteristics of the long-term dependence of water quality parameters, the unique memory ability of the Long Short-Term Memory (LSTM) neural network was utilized to predict water quality parameters. To improve the accuracy of the LSTM prediction model, the ammonia nitrogen data were decomposed using Empirical Modal Decomposition (EMD), and then the parameters of the LSTM model were optimized using the Improved Whale Optimization Algorithm (IWOA), and a combined prediction model based on EMD-IWOA-LSTM was proposed. The study outcomes demonstrate that EMD-IWOA-LSTM displays improved prediction accuracy with reduced RootMean Square Error (RMSE) and Mean Absolute Percentage Error (MAPE) in comparison to the LSTM and IWOA-LSTM approaches. These research findings better enable the monitoring and prediction of water quality parameters, offering a novel approach to preventing water pollution rather than merely treating it afterwards.