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.

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

Predicting Water Quality Based on EEMD and LSTM Networks

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

Water quality prediction based on sparse dataset using enhanced machine learning

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

Developing a Novel Water Quality Prediction Model for a South African Aquaculture Farm

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

Water Quality Prediction Based on Machine Learning and Comprehensive Weighting Methods

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

Water Quality Prediction Based on Hybrid Deep Learning Algorithm

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

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

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

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

Dynamic Real-Time Forecasting Technique for Reclaimed Water Volumes in Urban River Environmental Management

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

Simulate the Forecast Capacity of a Complicated Water Quality Model Using the Long Short-Term Memory Approach

Water quality prediction using ARIMA-SSA-LSTM combination model

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

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

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