Water quality prediction of copper-molybdenum mining-beneficiation wastewater based on the PSO-SVR model

● Data acquisition and pre-processing for wastewater treatment were summarized. ● A PSO-SVR model for predicting COD_eff in wastewater was proposed. ● The COD_eff prediction performances of the three models in the paper were compared. ● The COD_eff prediction effects of different models in other studies were discussed. The mining-beneficiation wastewater treatment is highly complex and nonlinear. Various factors like influent quality, flow rate, pH and chemical dose, tend to restrict the effluent effectiveness of mining-beneficiation wastewater treatment. Chemical oxygen demand (COD) is a crucial indicator to measure the quality of mining-beneficiation wastewater. Predicting COD concentration accurately of mining-beneficiation wastewater after treatment is essential for achieving stable and compliant discharge. This reduces environmental risk and significantly improves the discharge quality of wastewater. This paper presents a novel AI algorithm PSO-SVR, to predict water quality. Hyperparameter optimization of our proposed model PSO-SVR, uses particle swarm optimization to improve support vector regression for COD prediction. The generalization capacity tested on out-of-distribution (OOD) data for our PSO-SVR model is strong, with the following performance metrics of root means square error (RMSE) is 1.51, mean absolute error (MAE) is 1.26, and the coefficient of determination (R²) is 0.85. We compare the performance of PSO-SVR model with back propagation neural network (BPNN) and radial basis function neural network (RBFNN) and shows it edges over in terms of the performance metrics of RMSE, MAE and R², and is the best model for COD prediction of mining-beneficiation wastewater. This is because of the less overfitting tendency of PSO-SVR compared with neural network architectures. Our proposed PSO-SVR model is optimum for the prediction of COD in copper-molybdenum mining-beneficiation wastewater treatment. In addition, PSO-SVR can be used to predict COD on a wide variety of wastewater through the process of transfer learning.