Abstract
The water is the main pivotal sources of irrigation in agricultural activities and affects human daily activities such as drinking. The water quality has a significant impact on various aspects and thus this review aims to addresses existing problems related to water quality prediction methods that have been found in the literature. We explore numerous quality parameters incorporated in the modelling process to measure the quality of water. Furthermore, we review the commonly adopted artificial intelligence-based models which have been utilized to forecast the water quality. 83 studies published from 2009 to 2023 were selected and reviewed based on their success in modelling and forecasting the water quality in multiple regions. We compared these articles in terms of parameters, modelling algorithms, time scale scenarios, and performance measurement indicators. This paper is beneficial to researchers that have interests to conduct future studies related to water quality forecasting. Additionally, we discuss a variety of modelling methods such as deep learning (DL) that have proven to boost the efficiency compared to traditional machine learning (ML) models. As a result, the hybrid-DL models were found to outperform other models such as standalone ML, standalone DL, and hybrid-ML. This study shows a significant limitation of the data-hungry DL models which require a big data size for modelling. Hence, at the end of this review study, we discuss the potential of some methods such as generative adversarial networks (GANs) and attention-based transformer to open the door for water quality prediction improvement. GAN has shown promising performance in other domains for synthetic data generation. The potential usage of GAN for water quality domain can overcome the limitations of lack of data and enhance the performance of the predictive models reviewed in this study. Similarly, transformer was found to be state of the art model for time series prediction and thus it can be good candidate to predict water quality.
Similar content being viewed by others
References
Peterson KT, Sagan V, Sidike P, Hasenmueller EA, Sloan JJ, Knouft JH (2019) Machine learning-based ensemble prediction of water-quality variables using feature-level and decision-level fusion with proximal remote sensing. Photogramm Eng Remote Sensing 85(4):269–280. https://doi.org/10.14358/PERS.85.4.269
Barzegar R, Adamowski J, Moghaddam AA (2016) Application of wavelet-artificial intelligence hybrid models for water quality prediction: a case study in Aji-Chay River, Iran. Stoch Env Res Risk Assess 30(7):1797–1819. https://doi.org/10.1007/S00477-016-1213-Y/METRICS
Alizadeh MJ, Kavianpour MR (2015) Development of wavelet-ANN models to predict water quality parameters in Hilo Bay, Pacific Ocean. Mar Pollut Bull 98(1–2):171–178. https://doi.org/10.1016/J.MARPOLBUL.2015.06.052
Yan J et al (2021) Water quality prediction in the Luan river based on 1-DRCNN and BiGRU hybrid neural network model. Water 13:1273. https://doi.org/10.3390/W13091273
Olyaie E, Banejad H (2011) Application of an artificial neural network model to rivers water quality indexes prediction-a case study. J Am Sci 7(1):1545–1003
Sani Gaya M et al (2020) Estimation of water quality index using artificial intelligence approaches and multi-linear regression. IAES Int J Artif Intell 9(1):126–134. https://doi.org/10.11591/ijai.v9.i1.pp126-134
Pham QB, Mohammadpour R, Linh NT, Mohajane M, Pourjasem A, Sammen SS, Anh DT, Nam VT (2021) Application of soft computing to predict water quality in wetland. Environ Sci Pollut Res 28:185–200
Y. Khan and C. S. See, “Predicting and analyzing water quality using machine learning: a comprehensive model,” 2016 IEEE Long Island systems, applications and technology conference, LISAT 2016, Jun. 2016, doi: https://doi.org/10.1109/LISAT.2016.7494106.
Najah Ahmed A et al (2019) Machine learning methods for better water quality prediction. J Hydrol (Amst) 578:124084. https://doi.org/10.1016/J.JHYDROL.2019.124084
Gao C, Wang Z, Ji X, Wang W, Wang Q, Qing D (2023) Coupled improvements on hydrodynamics and water quality by flowing water in towns with lakes. Environ Sci Pollut Res 30(16):46813–46825. https://doi.org/10.1007/s11356-023-25348-3
Liu P, Wang J, Sangaiah AK, Xie Y, Yin X (2019) Analysis and prediction of water quality using LSTM deep neural networks in IoT environment. Sustainability 11(7):2058. https://doi.org/10.3390/SU11072058
Bui DT, Khosravi K, Tiefenbacher J, Nguyen H, Kazakis N (2020) Improving prediction of water quality indices using novel hybrid machine-learning algorithms. Sci Total Environ 721:137612. https://doi.org/10.1016/J.SCITOTENV.2020.137612
Tiwari S, Babbar R, Kaur G (2018) Performance evaluation of two ANFIS models for predicting water quality index of river Satluj (India). Adv Civil Eng. https://doi.org/10.1155/2018/8971079
Chen H et al (2022) Water quality prediction based on LSTM and attention mechanism: a case study of the Burnett River Australia. Sustainability 14(20):13231. https://doi.org/10.3390/SU142013231
Sha J, Li X, Zhang M, Wang ZL (2021) Comparison of forecasting models for real-time monitoring of water quality parameters based on hybrid deep learning neural networks. Water 13(11):1547. https://doi.org/10.3390/W13111547
Li L, Jiang P, Xu H, Lin G, Guo D, Wu H (2019) Water quality prediction based on recurrent neural network and improved evidence theory: a case study of Qiantang River, China. Environ Sci Pollut Res 26(19):19879–19896. https://doi.org/10.1007/S11356-019-05116-Y/METRICS
Emamgholizadeh S, Kashi H, Marofpoor I, Zalaghi E (2014) Prediction of water quality parameters of Karoon River (Iran) by artificial intelligence-based models. Int J Environ Sci Technol 11(3):645–656. https://doi.org/10.1007/S13762-013-0378-X/METRICS
Asadollah SBHS, Sharafati A, Motta D, Yaseen ZM (2021) River water quality index prediction and uncertainty analysis: A comparative study of machine learning models. J Environ Chem Eng 9(1):104599. https://doi.org/10.1016/J.JECE.2020.104599
Haghiabi AH, Nasrolahi AH, Parsaie A (2018) Water quality prediction using machine learning methods. Water Qual Res J 53(1):3–13. https://doi.org/10.2166/WQRJ.2018.025
Lu H, Ma X (2020) Hybrid decision tree-based machine learning models for short-term water quality prediction. Chemosphere 249:126169. https://doi.org/10.1016/J.CHEMOSPHERE.2020.126169
Ahmed U, Mumtaz R, Anwar H, Shah AA, Irfan R, García-Nieto J (2019) Efficient water quality prediction using supervised machine learning. Water 11(11):2210. https://doi.org/10.3390/W11112210
Zhou Y (2020) Real-time probabilistic forecasting of river water quality under data missing situation: deep learning plus post-processing techniques. J Hydrol (Amst) 589:125164. https://doi.org/10.1016/J.JHYDROL.2020.125164
Hayder G, Kurniawan I, Mustafa HM (2020) Implementation of machine learning methods for monitoring and predicting water quality parameters. Biointerface Res Appl Chem. https://doi.org/10.33263/BRIAC112.92859295
Baek SS, Pyo J, Chun JA (2020) Prediction of water level and water quality using a CNN-LSTM combined deep learning approach. Water 12(12):3399. https://doi.org/10.3390/W12123399
Jin T, Cai S, Jiang D, Liu J (2019) A data-driven model for real-time water quality prediction and early warning by an integration method. Environ Sci Pollut Res 26(29):30374–30385. https://doi.org/10.1007/S11356-019-06049-2/METRICS
Isiyaka HA, Mustapha A, Juahir H, Phil-Eze P (2019) Water quality modelling using artificial neural network and multivariate statistical techniques. Model Earth Syst Environ 5(2):583–593. https://doi.org/10.1007/S40808-018-0551-9/METRICS
Liu S, Tai H, Ding Q, Li D, Xu L, Wei Y (2013) A hybrid approach of support vector regression with genetic algorithm optimization for aquaculture water quality prediction. Math Comput Model 58(3–4):458–465. https://doi.org/10.1016/J.MCM.2011.11.021
Ouma YO, Okuku CO, Njau EN (2020) Use of artificial neural networks and multiple linear regression model for the prediction of dissolved oxygen in rivers: case study of hydrographic basin of river Nyando, Kenya. Complexity. https://doi.org/10.1155/2020/9570789
Khoi DN, Quan NT, Linh DQ, Nhi PTT, Thuy NTD (2022) Using machine learning models for predicting the Water Quality Index in the La Buong River, Vietnam. Water 14(10):1552. https://doi.org/10.3390/W14101552
Alqahtani A, Shah MI, Aldrees A, Javed MF (2022) Comparative assessment of individual and ensemble machine learning models for efficient analysis of river water quality. Sustainability 14(3):1183. https://doi.org/10.3390/SU14031183
Ziyad Sami BF et al (2022) Machine learning algorithm as a sustainable tool for dissolved oxygen prediction: a case study of Feitsui Reservoir, Taiwan. Sci Rep 12(1):1–12. https://doi.org/10.1038/s41598-022-06969-z
Izhar Shah M, Alaloul WS, Alqahtani A, Aldrees A, Ali Musarat M, Javed MF (2021) Predictive modeling approach for surface water quality: development and comparison of machine learning models. Sustainability 13(14):7515. https://doi.org/10.3390/SU13147515
Kisi O, Parmar KS (2016) Application of least square support vector machine and multivariate adaptive regression spline models in long term prediction of river water pollution. J Hydrol (Amst) 534:104–112. https://doi.org/10.1016/J.JHYDROL.2015.12.014
Melesse AM et al (2020) River water salinity prediction using hybrid machine learning models. Water 12(10):2951. https://doi.org/10.3390/W12102951
Hameed M, Sharqi SS, Yaseen ZM, Afan HA, Hussain A, Elshafie A (2017) Application of artificial intelligence (AI) techniques in water quality index prediction: a case study in tropical region, Malaysia. Neural Comput Appl 28(1):893–905. https://doi.org/10.1007/S00521-016-2404-7/METRICS
Ahmed AAM, Shah SMA (2017) Application of adaptive neuro-fuzzy inference system (ANFIS) to estimate the biochemical oxygen demand (BOD) of Surma River. J King Saud Univ Eng Sci 29(3):237–243. https://doi.org/10.1016/J.JKSUES.2015.02.001
Maier PM, Keller S (2018) Machine learning regression on hyperspectral data to estimate multiple water parameters. Workshop Hyperspectral Image Signal Process, Evol Remote Sensing. https://doi.org/10.1109/WHISPERS.2018.8747010
Heddam S, Kisi O (2018) Modelling daily dissolved oxygen concentration using least square support vector machine, multivariate adaptive regression splines and M5 model tree. J Hydrol (Amst) 559:499–509. https://doi.org/10.1016/J.JHYDROL.2018.02.061
Ömer Faruk D (2010) A hybrid neural network and ARIMA model for water quality time series prediction. Eng Appl Artif Intell 23(4):586–594. https://doi.org/10.1016/J.ENGAPPAI.2009.09.015
Sattari MT, Joudi AR, Kusiak A (2016) Estimation of water quality parameters with data-driven model. J Am Water Works Assoc 108(4):E232–E239. https://doi.org/10.5942/JAWWA.2016.108.0012
Abba SI et al (2020) Implementation of data intelligence models coupled with ensemble machine learning for prediction of water quality index. Environ Sci Pollut Res 27(33):41524–41539. https://doi.org/10.1007/S11356-020-09689-X/METRICS
Yan T, Zhou A, Shen SL (2023) Prediction of long-term water quality using machine learning enhanced by Bayesian optimisation. Environ Pollut 318:120870. https://doi.org/10.1016/J.ENVPOL.2022.120870
Malek NHA, Yaacob WFW, Nasir SAM, Shaadan N (2022) Prediction of water quality classification of the kelantan river basin, Malaysia, using machine learning techniques. Water 14(7):1067. https://doi.org/10.3390/W14071067
Huang M et al (2018) A hybrid fuzzy wavelet neural network model with self-adapted fuzzy c-means clustering and genetic algorithm for water quality prediction in rivers. Complexity. https://doi.org/10.1155/2018/8241342
Rizal NNM, Hayder G, Mnzool M, Elnaim BME, Mohammed AOY, Khayyat MM (2022) Comparison between regression models, support vector machine (SVM), and artificial neural network (ANN) in river water quality prediction. Processes 10(8):1652. https://doi.org/10.3390/PR10081652
W. Xuan, J. Lv, and D. Xie, “A hybrid approach of support vector machine with particle swarm optimization for water quality prediction,” ICCSE 2010—5th International conference on computer science and education, final program and book of abstracts, pp. 1158–1163, 2010, doi: https://doi.org/10.1109/ICCSE.2010.5593697.
Than NH, Ly CD, van Tat P, Thanh NN (2016) Application of a neural network technique for prediction of the Water Quality index in the Dong Nai River, Vietnam. J Environ Sci Eng B 5:363–370. https://doi.org/10.17265/2162-5263/2016.07.007
Singh KP, Basant A, Malik A, Jain G (2009) Artificial neural network modeling of the river water quality—a case study. Ecol Modell 220(6):888–895. https://doi.org/10.1016/J.ECOLMODEL.2009.01.004
Q. Ye, X. Yang, C. Chen, and J. Wang, “River water quality parameters prediction method based on LSTM-RNN model,” Proceedings of the 31st Chinese control and decision conference, CCDC 2019, pp. 3024–3028, Jun. 2019, doi: https://doi.org/10.1109/CCDC.2019.8832885.
Azad A, Karami H, Farzin S, Mousavi SF, Kisi O (2019) Modeling river water quality parameters using modified adaptive neuro fuzzy inference system. Water Sci Eng 12(1):45–54. https://doi.org/10.1016/J.WSE.2018.11.001
Chou JS, Ho CC, Hoang HS (2018) Determining quality of water in reservoir using machine learning. Ecol Inform 44:57–75. https://doi.org/10.1016/J.ECOINF.2018.01.005
Elkiran G, Nourani V, Abba SI (2019) Multi-step ahead modelling of river water quality parameters using ensemble artificial intelligence-based approach. J Hydrol (Amst). https://doi.org/10.1016/J.JHYDROL.2019.123962
Chen K et al (2020) Comparative analysis of surface water quality prediction performance and identification of key water parameters using different machine learning models based on big data. Water Res. https://doi.org/10.1016/j.watres.2019.115454
Ly QV et al (2021) Application of machine learning for eutrophication analysis and algal bloom prediction in an urban river: a 10-year study of the Han River, South Korea. Sci Total Environ 797:149040. https://doi.org/10.1016/J.SCITOTENV.2021.149040
Ahmed M, Mumtaz R, Mohammad S, Zaidi H (2021) Analysis of water quality indices and machine learning techniques for rating water pollution: a case study of Rawal Dam, Pakistan. Water Supply. https://doi.org/10.2166/ws.2021.082
Zanoni MG, Majone B, Bellin A (2022) A catchment-scale model of river water quality by machine learning. Sci Total Environ 838:156377. https://doi.org/10.1016/J.SCITOTENV.2022.156377
Uddin MG, Nash S, Rahman A, Olbert AI (2023) Performance analysis of the water quality index model for predicting water state using machine learning techniques. Process Saf Environ Prot 169:808–828. https://doi.org/10.1016/J.PSEP.2022.11.073
Al-Sulttani AO, Al-Mukhtar M, Roomi AB, Farooque AA, Khedher KM, Yaseen ZM (2021) Proposition of New ensemble data-intelligence models for surface water quality prediction. IEEE Access 9:108527–108541. https://doi.org/10.1109/ACCESS.2021.3100490
Gazzaz NM, Yusoff MK, Aris AZ, Juahir H, Ramli MF (2012) Artificial neural network modeling of the water quality index for Kinta River (Malaysia) using water quality variables as predictors. Mar Pollut Bull 64(11):2409–2420. https://doi.org/10.1016/J.MARPOLBUL.2012.08.005
Kouadri S, Elbeltagi A, Islam ARMT, Kateb S (2021) Performance of machine learning methods in predicting water quality index based on irregular data set: application on Illizi region (Algerian southeast). Appl Water Sci 11(12):1–20. https://doi.org/10.1007/S13201-021-01528-9/TABLES/9
Anmala J, Venkateshwarlu T (2019) Statistical assessment and neural network modeling of stream water quality observations of Green River watershed, KY, USA. Water Supply 19(6):1831–1840. https://doi.org/10.2166/WS.2019.058
Ma C, Zhao J, Ai B, Sun S, Yang Z (2022) Machine learning based long-term water quality in the turbid pearl river Estuary, China. J Geophys Res Oceans. https://doi.org/10.1029/2021JC018017
Adusei YY, Quaye-Ballard J, Adjaottor AA, Mensah AA (2021) Spatial prediction and mapping of water quality of Owabi reservoir from satellite imageries and machine learning models. Egypt J Remote Sensing Space Sci 24(3):825–833. https://doi.org/10.1016/J.EJRS.2021.06.006
Othman F et al (2020) Efficient river water quality index prediction considering minimal number of inputs variables. Eng Appl Comput Fluid Mech 14(1):751–763. https://doi.org/10.1080/19942060.2020.1760942
Bhoi SK, Mallick C, Mohanty CR (2022) Estimating the water quality class of a major irrigation canal in Odisha, India: a supervised machine learning approach. Nat Environ Pollut Technol. https://doi.org/10.46488/NEPT.2022.v21i02.002
Aldhyani THH, Al-Yaari M, Alkahtani H, Maashi M (2020) Water quality prediction using artificial intelligence algorithms. Appl Bionics Biomech. https://doi.org/10.1155/2020/6659314
Lee HW, Kim M, Son HW, Min B, Choi JH (2022) Machine-learning-based water quality management of river with serial impoundments in the Republic of Korea. J Hydrol Reg Stud 41:101069. https://doi.org/10.1016/J.EJRH.2022.101069
Li J et al (2019) Hybrid soft computing approach for determining water quality indicator: Euphrates River. Neural Comput Appl 31(3):827–837. https://doi.org/10.1007/S00521-017-3112-7/METRICS
Fijani E, Barzegar R, Deo R, Tziritis E, Konstantinos S (2019) Design and implementation of a hybrid model based on two-layer decomposition method coupled with extreme learning machines to support real-time environmental monitoring of water quality parameters. Sci Total Environ 648:839–853. https://doi.org/10.1016/J.SCITOTENV.2018.08.221
Kumar L, Afzal MS, Ahmad A (2022) Prediction of water turbidity in a marine environment using machine learning: a case study of Hong Kong. Reg Stud Mar Sci 52:102260. https://doi.org/10.1016/J.RSMA.2022.102260
Ho JY et al (2019) Towards a time and cost effective approach to water quality index class prediction. J Hydrol (Amst) 575:148–165. https://doi.org/10.1016/J.JHYDROL.2019.05.016
Koranga M, Pant P, Kumar T, Pant D, Bhatt AK, Pant RP (2022) Efficient water quality prediction models based on machine learning algorithms for Nainital Lake, Uttarakhand. Mater Today Proc 57:1706–1712. https://doi.org/10.1016/J.MATPR.2021.12.334
Uddin MG, Nash S, Mahammad Diganta MT, Rahman A, Olbert AI (2022) Robust machine learning algorithms for predicting coastal water quality index. J Environ Manag 321:115923. https://doi.org/10.1016/J.JENVMAN.2022.115923
Gómez D, Salvador P, Sanz J, Casanova JL (2021) A new approach to monitor water quality in the Menor sea (Spain) using satellite data and machine learning methods. Environ Pollut 286:117489. https://doi.org/10.1016/J.ENVPOL.2021.117489
Zhu X, Guo H, Huang JJ, Tian S, Xu W, Mai Y (2022) An ensemble machine learning model for water quality estimation in coastal area based on remote sensing imagery. J Environ Manag 323:116187. https://doi.org/10.1016/J.JENVMAN.2022.116187
Saberioon M, Brom J, Nedbal V (2020) Chlorophyll-a and total suspended solids retrieval and mapping using Sentinel-2A and machine learning for inland waters. Ecol Indic 113:106236. https://doi.org/10.1016/J.ECOLIND.2020.106236
Xu T, Coco G, Neale M (2020) A predictive model of recreational water quality based on adaptive synthetic sampling algorithms and machine learning. Water Res 177:115788. https://doi.org/10.1016/J.WATRES.2020.115788
Deng T, Chau KW, Duan HF (2021) Machine learning based marine water quality prediction for coastal hydro-environment management. J Environ Manag 284:112051. https://doi.org/10.1016/J.JENVMAN.2021.112051
Al-Adhaileh MH, Alsaade FW (2021) Modelling and prediction of water quality by using artificial intelligence. Sustainability 13:4259. https://doi.org/10.3390/SU13084259
Khullar S, Singh N (2022) Water quality assessment of a river using deep learning Bi-LSTM methodology: forecasting and validation. Environ Sci Pollut Res 29(9):12875–12889. https://doi.org/10.1007/S11356-021-13875-W/METRICS
Latif SD et al (2022) Development of prediction model for phosphate in reservoir water system based machine learning algorithms. Ain Shams Eng J 13(1):101523. https://doi.org/10.1016/J.ASEJ.2021.06.009
A. P. Kogekar, R. Nayak, and U. C. Pati, “A CNN-BiLSTM-SVR based deep hybrid model for water quality forecasting of the river Ganga,” Proceedings of the 2021 IEEE 18th India council international conference, INDICON 2021, 2021, doi: https://doi.org/10.1109/INDICON52576.2021.9691532.
Wang S, Peng H, Liang S (2022) Prediction of estuarine water quality using interpretable machine learning approach. J Hydrol (Amst) 605:127320. https://doi.org/10.1016/J.JHYDROL.2021.127320
F. H. Garabaghi, S. Benzer, and R. Benzer, “Performance evaluation of machine learning models with ensemble learning approach in classication of water quality indices based on different subset of features,” (2022), doi: https://doi.org/10.21203/rs.3.rs-876980/v2.
Jiang Y, Li C, Sun L, Guo D, Zhang Y, Wang W (2021) A deep learning algorithm for multi-source data fusion to predict water quality of urban sewer networks. J Clean Prod 318:128533. https://doi.org/10.1016/J.JCLEPRO.2021.128533
Attention is all you need. A Vaswani, N Shazeer, N Parmar, J Uszkoreit, L Jones, AN Gomez, ... Advances in neural information processing systems 30, 2017.
Amanambu AC, Mossa J, Chen Y-H (2022) Hydrological drought forecasting using a deep transformer model. Water 14:3611. https://doi.org/10.3390/w14223611
Méndez M, Montero C, Núñez M (2022) Using deep transformer based models to predict ozone levels. In: Nguyen NT, Tran TK, Tukayev U, Hong TP, Trawiński B, Szczerbicki E (eds) Intelligent information and database systems ACIIDS 2022. Springer, Cham
Xu J, Fan H, Luo M, Li P, Jeong T, Xu L (2023) Transformer based water level prediction in Poyang Lake, China. Water 15:576. https://doi.org/10.3390/w15030576
Roushangar K, Shahnazi S, Azamathulla HM (2023) Sediment transport modeling through machine learning methods: review of current challenges and strategies. In: Pandey M, Azamathulla H, Pu JH (eds) River dynamics and flood hazards disaster. Resilience and green growth. Springer, Singapore
Azamathulla HM, Ghani AA, Chang CK, Hasan ZA, Zakaria NA (2010) Machine learning approach to predict sediment load–a case study. Clean-Soil Air Water 38:969–976
Wu A, Azamathulla HM, Wu FC (2011) Support vector machine approach for longitudinal dispersion coefficients in natural streams. Appl Soft Comput 11(2):2902–2905
Acknowledgements
This research was supported by the Ministry of Education (MOE) through Fundamental Research Grant Scheme (FRGS/1/2021/TK0/UIAM/03/1).
Author information
Authors and Affiliations
Corresponding author
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Irwan, D., Ali, M., Ahmed, A.N. et al. Predicting Water Quality with Artificial Intelligence: A Review of Methods and Applications. Arch Computat Methods Eng 30, 4633–4652 (2023). https://doi.org/10.1007/s11831-023-09947-4
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11831-023-09947-4