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.

On the Assessment Method Based on Correlation Weighted Comprehensive Nutritional Index in the Three Gorges Reservoir

Application of Euclidean Distance Model with Varying Weights in Comprehensive Assessment of Surface Water Quality

RELATIONSHIP BETWEEN WATER QUALITY OF SURFACE WATER AND BIOCHEMICAL CHARACTERS OF CYANOBACTERIAL BIOMASS IN THE TAIHU LAKE AND THE DIANCHI LAKE

Linking Land Use Metrics Measured In Aquatic-Terrestrial Interfaces To Water Quality Of Reservoir-Based Water Sources In Eastern China

Evaluation of water quality near the dam before and after the storage of Three Gorges Reservoir

Nutrient estimation by HJ-1 satellite imagery of Xiangxi Bay, Three Gorges Reservoir, China

Seasonal variation of trophic states in backwater areas of tributaries in Three Gorges Reservoir

Spatio-temporal Variation of Nitrogen, Phosphorus in Different Period in Three Gorges Reservoir after Its Impoundment

Long-term (2003−2021) evolution trend of water quality in the Three Gorges Reservoir: An evaluation based on an enhanced water quality index

Temporal Variations and Trends Prediction of Water Quality During 2010–2019 in the Middle Yangtze River, China

Assessing the eutrophication risk of the Danjiangkou Reservoir based on the EFDC model

Determining Ecoregional Numeric Nutrient Criteria by Stressor-Response Models in Yungui Ecoregion Lakes, China

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

The comparison between multidimensional normal cloud model method and several other methods for water eutrophication evaluation

Distribution Features of Nutrient and Phytoplankton in Incipient Three Gorges Reservoir

Eutrophication Risk Assessment of Tributaries in the Three Gorges Reservoir: A Case Study of the Pengxi River

A Bayesian method for comprehensive water quality evaluation of the Danjiangkou Reservoir water source area, for the middle route of the South-to-North Water Diversion Project in China

[Spatial Distribution Characteristics of Chlorophyll-a and Nutrient Salts in Tributaries of Different River Sections in the Three Gorges Reservoir Area During the Flood Season]

The Ecosystem Health Assessment of the Three Gorges Reservoir Area (chongqing Section)

An innovative method based on Gaussian cloud distribution and sample information richness for eutrophication assessment of Yangtze's lakes and reservoirs under uncertainty

Assessments of surface water quality through the use of multivariate statistical techniques: A case study for the watershed of the Yuqiao Reservoir, China