An Enhanced Water Quality Index for Water Quality Monitoring Using Remote Sensing and Machine Learning

Water quality deterioration is a serious problem with the increase in the urbanization rate. However, water quality monitoring uses grab sampling of physico-chemical parameters and a water quality index method to assess water quality. Both processes are lengthy and expensive. These traditional indices are biased towards the physico-chemical parameters because samples are only collected from certain sampling points. These limitations make the current water quality index method unsuitable for any water body in the world. Thus, we develop an enhanced water quality index method based on a semi-supervised machine learning technique to determine water quality. This method follows five steps: (i) parameter selection, (ii) sub-index calculation, (iii) weight assignment, (iv) aggregation of sub-indices and (v) classification. Physico-chemical, air, meteorological and hydrological, topographical parameters are acquired for the stream network of the Rawal watershed. Min-max normalization is used to obtain sub-indices, and weights are assigned with tree-based techniques, i.e., LightGBM, Random Forest, CatBoost, AdaBoost and XGBoost. As a result, the proposed technique removes the uncertainties in the traditional indexing with a 100% classification rate, removing the necessity of including all parameters for classification. Electric conductivity, secchi disk depth, dissolved oxygen, lithology and geology are amongst the high weighting parameters of using LightGBM and CatBoost with 99.1% and 99.3% accuracy, respectively. In fact, seasonal variations are observed for the classified stream network with a shift from 55:45% (January) to 10:90% (December) ratio for the medium to bad class. This verifies the validity of the proposed method that will contribute to water management planning globally.

Automated summary

Water quality deterioration is a serious problem in urbanization, and the current water quality index method has limitations. Therefore, an enhanced water quality index method based on machine learning is developed to address this issue. The method includes five steps: parameter selection, sub-index calculation, weight assignment, sub-index aggregation, and classification. Experimental results demonstrate that the proposed method effectively eliminates the uncertainties in the traditional indexing and achieves high accuracy in classifying different parameters.