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.

Predicting Water Turbidity in a Macro-Tidal Coastal Bay Using Machine Learning Approaches

Turbidity Prediction of Lake-Type Raw Water Using Random Forest Model Based on Meteorological Data: A Case Study of Tai Lake, China.

Comparative Analysis of Machine Learning Methods for Prediction of Chlorophyll-a in a River with Different Hydrology Characteristics: A Case Study in Fuchun River, China

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

Turbidity assessment in coastal regions combining machine learning, numerical modeling, and remote sensing

Water COD Prediction Based on Machine Learning

Beyond Tides and Time: Machine Learning Triumph in Water Quality

Application of Neural Network and MODIS 250m Imagery for Estimating Suspended Sediments Concentration in Hangzhou Bay, China

Tidal Current Prediction Based on a Hybrid Machine Learning Method

Prediction of long-term water quality using machine learning enhanced by Bayesian optimisation.

Prediction of the Area of High-Turbidity Water in the Yatsushiro Sea, Japan, Using Machine Learning with Satellite, Meteorological, and Oceanographic Data

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

Accurate water level predictions in a tidal reach: Integration of Physics-based and Machine Learning Approaches

Predicting Coastal Water Quality with Machine Learning, A Case Study of Beibu Gulf, China

Machine learning based marine water quality prediction for coastal hydro-environment management

An ensemble machine learning model for water quality estimation in coastal area based on remote sensing imagery

Modeling Chlorophyll-A in Taihu Lake with Machine Learning Models

Harmful Algal Blooms Prediction with Machine Learning Models in Tolo Harbour

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

Short‐term prediction of stream turbidity using surrogate data and a meta‐model approach: a case study

Inversion and Driving Force Analysis of Nutrient Concentrations in the Ecosystem of the Shenzhen-Hong Kong Bay Area