Application of machine learning in river water quality management: a review

Machine learning (ML), a branch of artificial intelligence (AI), has been increasingly used in environmental engineering due to the ability to analyze complex nonlinear problems (such as ones connected with water quality management) through a data-driven approach. This study provides an overview of different ML algorithms applied for monitoring and predicting river water quality. Different parameters could be monitored or predicted, such as dissolved oxygen (DO), biological and chemical oxygen demand (BOD and COD), turbidity levels, the concentration of different ions (such as Mg²⁺ and Ca²⁺), heavy metal or other pollutant's concentration, pH, temperature, and many more. Although many algorithms have been investigated for the prediction of river water quality, there are several which are most commonly used in engineering practice. These models mostly include so-called supervised learning algorithms, such as artificial neural network (ANN), support vector machine (SVM), random forest (RF), decision tree (DT), and deep learning (DL). To further enhance prediction power, novel hybrid algorithms, could be used. However, the quality of prediction is not only dependent on the applied algorithm but also on the availability of previously mentioned water quality parameters, their selection, and the combination of input data used to train the ML model.