Abstract:Water quality prediction plays a crucial role in pollution treatment. However, inaccurate long-term prediction resulting from complex information patterns and insufficient feature extraction may lead to unnecessary environmental costs. In this study, a novel Interval Decomposition Correlation Particle Swarm Optimization-Extreme Learning Machine (IDCPSO-ELM) model is proposed to improve long-term water quality prediction ability. We employ Multivariate Variational Mode Decomposition (MVMD), Variational Mode Decomposition (VMD), Sliding Correlation and Permutation Entropy to reconstruct features to reduce complexity. Seasonal and Trend decomposition using Loess-Empirical Wavelet Transform decomposition (STL-EWT) is applied to target variables to improve feature extraction ability. Particle Swarm Optimization-Extreme Learning Machine (PSO-ELM) is used to predict high-frequency components based on the reconstructive features, and Back Propagation Neural Network-Extreme Learning Machine (BPNN-ELM) is employed to predict the trend and lowest-frequency components. Grey Wolf Optimization Algorithm (GWO) is used to combine these results. Six stations with high potential pollution threats in the Haihe River basin of Beijing from 2021 to 2022 are taken into model competition. The results show that: (1) In short-term prediction, average MAPE, RMSE and NSE of IDCPSO-ELM model reach 0.0934, 0.0410 and 0.8061, respectively, which are better than Genetic Algorithm-Elman Network (GA-ELMAN), Cuckoo Search Algorithm-Back Propagation Neural Network (CS-BPNN), PSO-ELM and Sparrow Search Algorithm-Kernel Extreme Learning Machine (SSA-KELM) short-term prediction models. (2) In long-term prediction, average MAPE, RMSE and NSE of IDCPSO-ELM model reach 0.1114, 0.0420 and 0.8268, respectively, which are better than the Radial Basis Function Neural Network (RBFNN), Grey Wolf Optimization-Support Vector Machine Regression (GWO-SVR), Whale Optimization Algorithm-Deep Extreme Learning Machine (WOA-DELM), Wavelet Transform decomposition-GWO-SVR-BPNN (WT-GWO-SVR-BPNN) and Complete Ensemble Empirical Mode Decomposition with Adaptive Noise-PSO-ELM-Long Short-Term Memory (CEEMDAN-PSO-ELM-LSTM) long-term prediction models. IDCPSO-ELM model is considered competitive and promising for long-term water quality prediction, particularly in areas with high potential pollution threats.

Pattern Classification and Prediction of Water Quality by Neural Network with Particle Swarm Optimization

Improved BP neural network prediction model based on particle swarm algorithm

Prediction of Salinity in Qiantang Estuary Based on Wavelet Neural Network Optimized by Particle Swarm Optimization

Modeling and Diagnosis of Water Quality Parameters in Wastewater Treatment Process Based on Improved Particle Swarm Optimization and Self-Organizing Neural Network

A novel hybrid BPNN model based on adaptive evolutionary Artificial Bee Colony Algorithm for water quality index prediction

A water quality prediction method based on k-nearest-neighbor probability rough sets and PSO-LSTM

A Hybrid Model for Water Quality Prediction Based on an Artificial Neural Network, Wavelet Transform, and Long Short-Term Memory

Application of Artificial Neural Networks to Classify Water Quality of the Yellow River.

A novel interval decomposition correlation particle swarm optimization-extreme learning machine model for short-term and long-term water quality prediction

A Review of the Artificial Neural Network Models for Water Quality Prediction

Artificial Neural Networks in the Prediction and Assessment for Water Quality: A Review

Optimized BP Neural Network for Dissolved Oxygen Prediction

A Prediction Model Based on Deep Belief Network and Least Squares SVR Applied to Cross-Section Water Quality

Prediction of particulate matter at street level using artificial neural networks coupling with chaotic particle swarm optimization algorithm

Optimization of Pump Allocation in Real Water Networks for Water Quality Improvement Using Particle Swarm Optimization

Water Quality Prediction Based on Hybrid Deep Learning Algorithm

A water quality assessment method based on an improved grey relational analysis and particle swarm optimization multi-classification support vector machine

Water quality polluted by total suspended solids classified within an Artificial Neural Network approach

A novel model for water quality prediction caused by non-point sources pollution based on deep learning and feature extraction methods

Improved prediction of chlorophyll-a concentrations in reservoirs by GRU neural network based on particle swarm algorithm optimized variational modal decomposition