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.

Predicting Water Quality Based on EEMD and LSTM Networks

Daily water supply prediction method based on integrated learning and deep learning

Accurate prediction of water quality in urban drainage network with integrated EMD-LSTM model

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

Study on Water Quality Prediction of Urban Reservoir by Coupled CEEMDAN Decomposition and LSTM Neural Network Model

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

Development of a water quality prediction model using ensemble empirical mode decomposition and long short-term memory

Water quality prediction based on sparse dataset using enhanced machine learning

Water Quality Prediction in Urban Waterways Based on Wavelet Packet Denoising and LSTM

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

Real-time Water Quality Detection Based on Fluctuation Feature Analysis with the LSTM Model

An ensemble model for accurate prediction of key water quality parameters in river based on deep learning methods

Water Quality Prediction Based on the KF-LSTM Encoder-Decoder Network: A Case Study with Missing Data Collection

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

Comparative analysis of water quality prediction performance based on LSTM in the Haihe River Basin, China

Analysis of Water Quality Prediction in the Yangtze River Delta under the River Chief System

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

Water Quality Prediction Based on Machine Learning and Comprehensive Weighting Methods

Hybrid deep learning based prediction for water quality of plain watershed

A watershed water quality prediction model based on attention mechanism and Bi-LSTM

Predict water quality using an improved deep learning method based on spatiotemporal feature correlated: a case study of the Tanghe Reservoir in China