Abstract:The water quality index (WQI) is a widely used tool for comprehensive assessment of river environments. However, its calculation involves numerous water quality parameters, making sample collection and laboratory analysis time-consuming and costly. This study aimed to identify key water parameters and the most reliable prediction models that could provide maximum accuracy using minimal indicators. Water quality from 2020 to 2023 were collected including nine biophysical and chemical indicators in seventeen rivers in Yancheng and Nantong, two coastal cities in Jiangsu Province, China, adjacent to the Yellow Sea. Linear regression and seven machine learning models (Artificial Neural Network (ANN), Self-Organizing Maps (SOM), K-Nearest Neighbor (KNN), Support Vector Machines (SVM), Random Forest (RF), Extreme Gradient Boosting (XGB) and Stochastic Gradient Boosting (SGB)) were developed to predict WQI using different groups of input variables based on correlation analysis. The results indicated that water quality improved from 2020 to 2022 but deteriorated in 2023, with inland stations exhibiting better conditions than coastal ones, particularly in terms of turbidity and nutrients. The water environment was comparatively better in Nantong than in Yancheng, with mean WQI values of approximately 55.3-72.0 and 56.4-67.3, respectively. The classifications "Good" and "Medium" accounted for 80 % of the records, with no instances of "Excellent" and 2 % classified as "Bad". The performance of all prediction models, except for SOM, improved with the addition of input variables, achieving R2 values higher than 0.99 in models such as SVM, RF, XGB, and SGB. The most reliable models were RF and XGB with key parameters of total phosphorus (TP), ammonia nitrogen (AN), and dissolved oxygen (DO) (R2 = 0.98 and 0.91 for training and testing phase) for predicting WQI values, and RF using TP and AN (accuracy higher than 85 %) for WQI grades. The prediction accuracy for "Medium" and "Low" water quality grades was highest at 90 %, followed by the "Good" level at 70 %. The model results could contribute to efficient water quality evaluation by identifying key water parameters and facilitating effective water quality management in river basins.

High-frequency data significantly enhances the prediction ability of point and interval estimation

A hybrid model of ARIMA and MLP with a Grasshopper optimization algorithm for time series forecasting of water quality

A deep learning interpretable model for river dissolved oxygen multi-step and interval prediction based on multi-source data fusion

Comparison of Forecasting Models for Real-Time Monitoring of Water Quality Parameters Based on Hybrid Deep Learning Neural Networks

Time series-based machine learning for forecasting multivariate water quality in full-scale drinking water treatment with various reagent dosages

Estimation of water quality parameters based on time series hydrometeorological data in Miaowan Island

Analysis of dissolved oxygen influencing factors and concentration prediction using input variable selection technique: A hybrid machine learning approach

Dissolved oxygen concentration predictions for running waters with using hybrid machine learning techniques

Comparative Analysis of Surface Water Quality Prediction Performance and Identification of Key Water Parameters Using Different Machine Learning Models Based on Big Data.

Enhancing Water Quality Monitoring: an Integrated Machine Learning Framework with Physical Constraints for Imputation and Time Series Downscaling

A hybrid XGBoost-ISSA-LSTM model for accurate short-term and long-term dissolved oxygen prediction in ponds

Beyond Tides and Time: Machine Learning Triumph in Water Quality

Estimation of Water Quality Parameters with High-Frequency Sensors Data in a Large and Deep Reservoir

Assessment and prediction of Water Quality Index (WQI) by seasonal key water parameters in a coastal city: application of machine learning models

Water quality prediction based on sparse dataset using enhanced machine learning

Water Quality Prediction Based on Machine Learning and Comprehensive Weighting Methods

Enhanced predictive modeling of dissolved oxygen concentrations in riverine systems using novel hybrid temporal pattern attention deep neural networks

Water quality prediction based on IGRA-ISSA-LSTM model

Assessing and predicting water quality index with key water parameters by machine learning models in coastal cities, China

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

Interpretation of river water quality data is strongly controlled by measurement time and frequency