Skip to main content
Water Science and TechnologyVolume 88, Issue 9, 1 November 2023, Pages 2297-2308

Application of machine learning in river water quality management: a review(Article)(Open Access)

  Save all to author list
  • aFaculty of Technical Sciences, Department of Environmental Engineering and Occupational Safety and Health, University of Novi Sad, Trg Dositeja Obradovica 6, Novi Sad, 21000, Serbia
  • bInstitute for Artificial Intelligence R&D of Serbia, Fruskogorska 1, Novi Sad, Serbia

Abstract

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. © 2023 IWA Publishing. All rights reserved.

Author keywords

artificial intelligenceenvironmental engineeringmachine learning algorithmswater quality index

Indexed keywords

Engineering controlled terms:Decision treesDeep learningDissolved oxygenForecastingHeavy metalsLearning algorithmsNeural networksQuality managementRiver pollutionRiversSupport vector machinesWater conservationWater managementWater quality
Engineering uncontrolled termsBiological oxygen demandChemical-oxygen demandsData-driven approachEngineering practicesMachine learning algorithmsMachine-learningNonlinear problemsRiver water qualityWater quality indexesWater quality management
Engineering main heading:Biochemical oxygen demand
EMTREE drug terms:calcium iondissolved oxygenheavy metalmagnesium ionriver water
GEOBASE Subject Index:artificial intelligenceartificial neural networkchemical oxygen demanddissolved oxygenheavy metalmachine learningsupport vector machinewater quality
EMTREE medical terms:Articleartificial neural networkback propagation neural networkbagged tree modelbayesian autoregressive modelbiochemical oxygen demandchemical oxygen demandconcentration (parameter)convolutional neural networkdecision treedecision tree regressiondeep learningdeep neural networkdiscriminant analysisenvironmental engineeringextra tree regressionextreme learning machinefeed forward neural networkfuzzy systemgeneralized regression neural networkgroup method of data handlingisolation forest algorithmk nearest neighborlong short term memory networkmachine learningmultinomial logistic regressionpHpredictionrandom forestrandom treerecurrent neural networkreduced error pruning treesupervised machine learningsupport vector machinetotal quality managementturbiditywater monitoringwater pollutantwater qualitywater temperaturealgorithmartificial intelligenceenvironmental monitoringmachine learningriver
MeSH:AlgorithmsArtificial IntelligenceEnvironmental MonitoringMachine LearningRiversSupport Vector MachineWater Quality

Chemicals and CAS Registry Numbers:

calcium ion, 14127-61-8; magnesium ion, 22537-22-0

Funding details

Funding sponsor Funding number Acronym
Ministry of Science, ICT and Future PlanningMSIP
European Commission
See opportunities by EC
EC
H2020 Marie Skłodowska-Curie Actions
See opportunities by MSCA
101086387,451-03-47/2023-01/200156MSCA
Science Fund of the Republic of Serbia6707
  • 1

    This research was supported by the Science Fund of the Republic of Serbia, grant number 6707, REmote WAter quality monitoRing anD IntelliGence \u2013 REWARDING and by the Ministry of Science, European Union\u2019s Horizon Europe Marie Sklodowska-Curie Actions (MSCA) under grant agreement project number 101086387 \u2013 REMARKABLE and Technological Development and Innovation through project no. 451-03-47/2023-01/200156 \u2018Innovative scientific and artistic research from the FTS (activity) domain\u2019.

  • ISSN: 02731223
  • CODEN: WSTED
  • Source Type: Journal
  • Original language: English
  • DOI: 10.2166/wst.2023.331
  • PubMed ID: 37966184
  • Document Type: Article
  • Publisher: IWA Publishing

  Dmitrasinovic, S.; Faculty of Technical Sciences, Department of Environmental Engineering and Occupational Safety and Health, University of Novi Sad, Trg Dositeja Obradovica 6, Novi Sad, Serbia;
© Copyright 2023 Elsevier B.V., All rights reserved.

Cited by 8 documents

Vinokić, L. , Dotlić, M. , Prodanović, V.
Effectiveness of three machine learning models for prediction of daily streamflow and uncertainty assessment
(2025) Water Research X
Timis, E.C. , Hangan, H. , Cristea, V.M.
High-Resolution Flow and Phosphorus Forecasting Using ANN Models, Catering for Extremes in the Case of the River Swale (UK)
(2025) Hydrology
Gill, S.S. , Golec, M. , Hu, J.
Edge AI: A Taxonomy, Systematic Review and Future Directions
(2025) Cluster Computing
View details of all 8 citations
{"topic":{"name":"Artificial Neural Network; Dissolved Oxygen; Water Pollution","id":16016,"uri":"Topic/16016","prominencePercentile":99.07041,"prominencePercentileString":"99.070","overallScholarlyOutput":0},"dig":"76fb2e6b69dca723a9eb2cd0517b2db6f5a8bd5d00fe5e272b53a1ce1dbdfcfa"}

SciVal Topic Prominence

Topic:
Prominence percentile: