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.

Machine-Learning-Based Approach to Assessing Water Quality in a Specific Basin: the Case of Wujingang Basin

Analysis of spatio-temporal variations of river water quality and construction of a novel cost-effective assessment model: a case study in Hong Kong

Water quality assessment and source identification of the Shuangji River (China) using multivariate statistical methods

An integrated regional water quality assessment method considering interrelationships among monitoring indicators

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

Assessment of Water Quality and Source Apportionment in a Typical Urban River in China Using Multivariate Statistical Methods

Water Quality Evaluation and Pollution Source Apportionment of Surface Water in a Major City in Southeast China Using Multi-Statistical Analyses and Machine Learning Models

Surface Water Quality Assessment and Contamination Source Identification Using Multivariate Statistical Techniques: A Case Study of the Nanxi River in the Taihu Watershed, China

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

Assessment of Water Quality Using Multivariate Statistical Methods: A Case Study of an Urban Landscape Water, Beijing

Assessment of temporal and spatial variations in surface water quality using multivariate statistical techniques: A case study of Nenjiang River basin, China

Optimization of water quality index models using machine learning approaches

A New Approach for Water Quality Assessment Based on Multivariate Statistical Analysis and Radial Basis Function Neural Networks

Deep optimization of water quality index and positive matrix factorization models for water quality evaluation and pollution source apportionment using a random forest model

Surface Water Quality Assessment Using Multivariate Statistical Techniques: Case Study of Songhua River Basin, China

Water Quality Evaluation of the Yangtze River in China Using Machine Learning Techniques and Data Monitoring on Different Time Scales

Assessment of surface water quality via multivariate statistical techniques: A case study of the Songhua River Harbin region, China

Application of Multivariate Statistical Techniques in Assessment of Surface Water Quality in Second Songhua River Basin, China

Multivariate statistical techniques for the evaluation of water quality: A case study in Shitoukoumen Reservoir catchment

Application of Grey Clustering Method Based on Improved Analytic Hierarchy Process in Water Quality Evaluation

Spatiotemporal Analysis of Water Quality Using Multivariate Statistical Techniques and the Water Quality Identification Index for the Qinhuai River Basin, East China