River water quality index prediction and uncertainty analysis: A comparative study of machine learning models

By S. Asadollah, A. Sharafati, D. Motta, Zaher Mundher Yaseen. 2020

320 citations recorded by the source corpus.

Abstract The Water Quality Index (WQI) is the most common indicator to characterize surface water quality. This study introduces a new ensemble machine learning model called Extra Tree Regression (ETR) for predicting monthly WQI values at the Lam Tsuen River in Hong Kong. The ETR model performance is compared with that of the classic standalone models, Support Vector Regression (SVR) and Decision Tree Regression (DTR). The monthly input water quality data including Biochemical Oxygen Demand (BOD), Chemical Oxygen Demand (COD), Dissolved Oxygen (DO), Electrical Conductivity (EC), Nitrate-Nitrogen ( NO 3 -N), Nitrite-Nitrogen ( NO 2 -N), Phosphate ( P O 4 3 - ), potential for Hydrogen (pH), Temperature (T) and Turbidity (TUR) are used for building the prediction models. Various input data combinations are investigated and assessed in terms of prediction performance, using numerical indices and graphical comparisons. The analysis shows that the ETR model generally produces more accurate WQI predictions for both training and testing phases. Although including all the ten input variables achieves the highest prediction performance ( R 2 t e s t = 0.98 , R M S E t e s t = 2.99 ), a combination of input parameters including only BOD, Turbidity and Phosphate concentration provides the second highest prediction accuracy ( R 2 t e s t = 0.97 , R M S E t e s t = 3.74 ). The uncertainty analysis relative to model structure and input parameters highlights a higher sensitivity of the prediction results to the former. In general, the ETR model represents an improvement on previous approaches for WQI prediction, in terms of prediction performance and reduction in the number of input parameters.

Subfields: Environmental Engineering, Water Science and Technology

View Paper: https://www.semanticscholar.org/paper/aeaf6aff2b3006dc85ffa5566365d556df85a853

Citations in the corpus

• Water quality prediction using machine learning models based on grid search method (2023, unknown): “Seyed Asadollah et al. [18] presented Extra Tree Regression (ETR), an ensemble machine learning technique, for forecasting monthly WQI rates along the Lam Tsuen River in Hong Kong.”