Abstract:In the context of escalating global environmental concerns, the importance of preserving water resources and upholding ecological equilibrium has become increasingly apparent. As a result, the monitoring and prediction of water quality have emerged as vital tasks in achieving these objectives. However, ensuring the accuracy and dependability of water quality prediction has proven to be a challenging endeavor. To address this issue, this study proposes a comprehensive weight-based approach that combines entropy weighting with the Pearson correlation coefficient to select crucial features in water quality prediction. This approach effectively considers both feature correlation and information content, avoiding excessive reliance on a single criterion for feature selection. Through the utilization of this comprehensive approach, a comprehensive evaluation of the contribution and importance of the features was achieved, thereby minimizing subjective bias and uncertainty. By striking a balance among various factors, features with stronger correlation and greater information content can be selected, leading to improved accuracy and robustness in the feature-selection process. Furthermore, this study explored several machine learning models for water quality prediction, including Support Vector Machines (SVMs), Multilayer Perceptron (MLP), Random Forest (RF), XGBoost, and Long Short-Term Memory (LSTM). SVM exhibited commendable performance in predicting Dissolved Oxygen (DO), showcasing excellent generalization capabilities and high prediction accuracy. MLP demonstrated its strength in nonlinear modeling and performed well in predicting multiple water quality parameters. Conversely, the RF and XGBoost models exhibited relatively inferior performance in water quality prediction. In contrast, the LSTM model, a recurrent neural network specialized in processing time series data, demonstrated exceptional abilities in water quality prediction. It effectively captured the dynamic patterns present in time series data, offering stable and accurate predictions for various water quality parameters.

Water Quality Prediction Based on Improved Wavelet Transformation and Support Vector Machine

Support Vector Machine―an Alternative to Artificial Neuron Network for Water Quality Forecasting in an Agricultural Nonpoint Source Polluted River?

A water quality prediction model based on variational mode decomposition and the least squares support vector machine optimized by the sparrow search algorithm (VMD-SSA-LSSVM) of the Yangtze River, China

Water Quality Prediction Based on Machine Learning and Comprehensive Weighting Methods

Evaluating a Coupled Discrete Wavelet Transform and Support Vector Regression for Daily and Monthly Streamflow Forecasting

Water Quality Prediction of Small-Micro Water Body Based on the Intelligent-Algorithm-Optimized Support Vector Machine Regression Method and Unmanned Aerial Vehicles Multispectral Data

A hybrid water quality prediction model based on variational mode decomposition and bidirectional gated recursive unit

Modeling Daily and Monthly Water Quality Indicators in a Canal Using a Hybrid Wavelet-Based Support Vector Regression Structure

A novel hybrid model for water quality prediction based on VMD and IGOA optimized for LSTM

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

Application Of Least-Squares Support Vector Machines For Quantitative Evaluation Of Known Contaminant In Water Distribution System Using Online Water Quality Parameters

A novel approach for multivariate time series interval prediction of water quality at wastewater treatment plants

A multi-step water quality prediction model based on the Savitzky-Golay filter and Transformer optimized network

Study on Wavelet Multi-Scale Analysis and Prediction of Landslide Groundwater

Application of artificial intelligence based on synchrosqueezed wavelet transform and improved deep extreme learning machine in water quality prediction

Water Quality Prediction Based on an Innovated Physical and Data Driving Hybrid Model at Basin Scale

Prediction of River Sediment Transport Based on Wavelet Transform and Neural Network Model

Urban river ammonia nitrogen prediction model based on improved whale optimization support vector regression mixed synchronous compression wavelet transform

Beyond Tides and Time: Machine Learning Triumph in Water Quality

Research on SVR Water Quality Prediction Model Based on Improved Sparrow Search Algorithm

Application of Improved Least Squares Support Vector Machine in the Forecast of Daily Water Consumption