Abstract
Accurate prediction of water quality contributes to the intelligent management of water resources. Water quality indices have time series characteristics and nonlinearity, but the existing models only focus on the forward time series when long short-term memory (LSTM) is introduced and do not consider the parallel computation on the model. Owing to this, a new neural network called LSTM-multihead attention (LMA) was constructed to predict water quality, using long short-term memory to process time series data and multihead attention for parallel computing and extracting feature information. Additionally, water quality indices have the issues of multiple data types and complex data correlations, as well as missing data and abnormal data problems in water quality data. In order to solve these problems, this study proposes a water quality prediction model called GRA-LMA-based linear interpolation, gray relational analysis and LMA. Two experiments are carried out to verify the predictive performance of the GRA-LMA with the water quality data of the Huaihe River Basin as a case study sample. The first experiment focuses on data processing, including the processing of missing data and abnormal data of water quality data, and the correlation analysis of water quality indices. Linear interpolation is adapted to process the missing data, while a combination of boxplot and histogram is adopted to analyze and eliminate the abnormal data, which is then repaired the abnormal data with linear interpolation. The gray relational analysis is adopted to calculate the correlation between different water quality indices, and water quality indices with high correlation are retained to determine the input variables of the water quality prediction model. The data processing results demonstrate that repairs can be made using linear interpolation without altering the pattern of data change and the model by using the gray relational analysis to reduce the quantity of data it needs as input. In the second experiment, the predictive capacity of GRA-LMA and existing models such as backpropagation neural network (BP), recurrent neural network (RNN), long short-term memory (LSTM), and gate recurrent unit (GRU) was evaluated and compared using different numerical and graphical performance evaluation metrics. Comparative experimental results show that the mean square error of pH, dissolved oxygen, chemical oxygen demand, ammonia nitrogen, electrical conductivity, turbidity, total phosphorus, and total nitrogen of GRA-LMA is reduced to 0.05890, 0.40196, 0.32454, 0.04368, 14.71003, 8.13252, 0.01558, and 0.14345. The results indicate that GRA-LMA has superior adaptability for predicting various water quality indices and can significantly lower the induced prediction error.
This is a preview of subscription content, log in via an institution to check access.
Similar content being viewed by others
Data availability
The datasets used and/or analyzed during the current study are available from the corresponding author on reasonable request.
References
Abiodun OI, Jantan A, Omolara AE, Dada KV, Mohamed NA, Arshad H (2018) State-of-the-art in artificial neural network applications: a survey. Heliyon 4(11):e00938. https://doi.org/10.1016/j.heliyon.2018.e00938
Akbariyeh S, Pena CAG, Wang T, Mohebbi A, Bartelt-Hunt S, Zhang J, Li Y (2019) Prediction of nitrate accumulation and leaching beneath groundwater irrigated corn fields in the Upper Platte basin under a future climate scenario. Sci Total Environ 685:514–526. https://doi.org/10.1016/j.scitotenv.2019.05.417
Akhtar S, Rozi S (2009) An autoregressive integrated moving average model for short-term prediction of hepatitis C virus seropositivity among male volunteer blood donors in Karachi, Pakistan. World Journal of Gastroenterology 15(13):1607–1612. https://doi.org/10.3748/wjg.15.1607
Alba JG, Bárcena JF, Ugarteburu C, García A (2019) Artificial neural networks as emulators of process-based models to analyse bathing water quality in estuaries. Water Res 150:283–295. https://doi.org/10.1016/j.watres.2018.11.063
Alcaide SD, Santos PM (2019) Mapping fecal pollution in rural groundwater supplies by means of artificial intelligence classifiers. J Hydrol 577:124006. https://doi.org/10.1016/j.jhydrol.2019.124006
Alessandro F, Michael S, Christoph RV (2020) Interpolation uncertainty of atmospheric temperature profiles. Atmospheric. Meas Tech 13(12). https://doi.org/10.5194/amt-13-6445-2020
Allawi MF, Salih SQ, Kassim M, Ramal MM, Mohammed AS, Yaseen ZM (2022) Application of computational model based probabilistic neural network for surface water quality prediction. Mathematics 10:3960. https://doi.org/10.3390/math10213960
Anwar S, Khalil B, Seddik M, Eltahan A, Saadi AE (2022) An evaluation of À Trous-based record extension techniques for water quality record extension. Water 14:2264. https://doi.org/10.3390/w14142264
Ashouri M, Haghighat F, Fung CB, Yoshino H (2018) Development of a ranking procedure for energy performance evaluation of buildings based on occupant behavior. Energ Buildings 183:659–671. https://doi.org/10.1016/j.enbuild.2018.11.050
Baek SS, Pyo J, Chun JA (2020) Prediction of water level and water quality using a CNN-LSTM combined deep learning approach. Water 12:3399. https://doi.org/10.3390/w12123399
Bayram S, Çıtakoğlu H (2023) Modeling monthly reference evapotranspiration process in Turkey: application of machine learning methods. Environ Monit Assess 195(1):67. https://doi.org/10.1007/s10661-022-10662-z
Bhagat SK, Tung TM, Yaseen ZM (2020) Development of artificial intelligence for modeling wastewater heavy metal removal: State of the art, application assessment and possible future research. J Clean Prod 250:119473. https://doi.org/10.1016/j.jclepro.2019.119473
Bi Z, Zhang Y, Shi P, Zhang X, Shan Z, Ren L (2021) The impact of land use and socio-economic factors on ammonia nitrogen pollution in Weihe River watershed, China. Environ Sci Pollut Res 28:17659–17674. https://doi.org/10.1007/s11356-020-11960-0
Bianchi FM, Livi L, Mikalsen KØ, Kampffmeyer M, Jenssen R (2019) Learning representations of multivariate time series with missing data. Pattern Recogn 96:106973. https://doi.org/10.1016/j.patcog.2019.106973
Bing W, Jiwei Z, Meng G, Jiancang X, Liu Y, Nan L, Biao W (2023) Do the protection and harnessing of river systems promote the society, economy, and ecological environment of cities? A case study of Xi’an, China. Sustain Cities Soc 97:104761. https://doi.org/10.1016/j.scs.2023.104761
Box GEP, Jenkins GM, Reinsel GC (1994) Time series analysis: forecasting and control. J Mark Res 14(2):238–242. https://doi.org/10.1080/00401706.1977.10489562
Cao J, Sun Q, Zhao D, Xu M, Shen Q, Wang D, Wang Y, Ding S (2020) A critical review of the appearance of black-odorous waterbodies in China and treatment methods. J Hazard Mater 385(C):121511. https://doi.org/10.1016/j.jhazmat.2019.121511
Chen S, Fang G, Huang X, Zhang Y (2018) Water quality prediction model of a water diversion project based on the improved artificial bee colony–backpropagation neural network. Water 10:806. https://doi.org/10.3390/w10060806
Chen C, Du Z, He L, Shi Y, Wang J, Dong W (2021) A novel gait pattern recognition method based on LSTM-CNN for lower limb exoskeleton. J Bionic Eng 18:1059–1072. https://doi.org/10.1007/s42235-021-00083-y
Chu H, Wei J, Wu W (2020) Streamflow prediction using LASSO-FCM-DBN approach based on hydro-meteorological condition classification. J Hydrol 580:124253. https://doi.org/10.1016/j.jhydrol.2019.124253
Chu QC, Xue XH (2023) A novel coupling preprocessing approach for handling missing data in water quality prediction. J Hydrol 617:128901. https://doi.org/10.1016/j.jhydrol.2022.128901
Cui F, Salih SQ, Choubin B, Bhagat SK, Samui P, Yaseen ZM (2020) Newly explored machine learning model for river flow time series forecasting at Mary River, Australia. Environ Monit Assess 192:761. https://doi.org/10.1007/s10661-020-08724-1
Coşkun Ö, Citakoglu H (2023) Prediction of the standardized precipitation index based on the long short-term memory and empirical mode decomposition-extreme learning machine models: the case of Sakarya, Türkiye. Phys Chem Earth, Parts A/B/C 131:103418. https://doi.org/10.1016/j.pce.2023.103418
David L, Svenja F, Alberto V, Günter B (2022) Significance testing of rank cross-correlations between autocorrelated time series with short-range dependence. J Appl Stat 50(14). https://doi.org/10.1080/02664763.2022.2137115
Dilmi S, Ladjal M (2021) A novel approach for water quality classification based on the integration of deep learning and feature extraction techniques. Chemom Intell Lab Syst 214:104329. https://doi.org/10.1016/j.chemolab.2021.104329
Duie TB, Khabat K, John T, Hoang N, Nerantzis K (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
Elham G, Massoud T, Sara N (2022) Improving the ARIMA model prediction for water quality parameters of urban water distribution networks (case study: CANARY dataset). International. J Environ Res 16(6). https://doi.org/10.1007/S41742-022-00482-X
Eze E, Halse S, Ajmal T (2021) Developing a novel water quality prediction model for a South African aquaculture farm. Water 13:1782. https://doi.org/10.3390/w13131782
Farhi N, Kohen E, Mamane H, Shavitt Y (2021) Prediction of wastewater treatment quality using LSTM neural network. Environ Technol Innov 23:101632. https://doi.org/10.1016/j.eti.2021.101632
Fu X, Zheng Q, Jiang G, Roy K, Huang L, Liu C, Li K, Chen HL, Song XY, Chen JY, Wang ZX (2023) Water quality prediction of copper-molybdenum mining-beneficiation wastewater based on the PSO-SVR model. Front Environ Sci Eng 17:98. https://doi.org/10.1007/s11783-023-1698-9
Guan Q, Chen Y, Wei Z, Heidari AA, Hu H, Yang XH, Zheng J, Zhou Q, Chen H, Chen F (2022) Medical image augmentation for lesion detection using a texture-constrained multichannel progressive GAN. Comput Biol Med 145:105444. https://doi.org/10.1016/j.compbiomed.2022.105444
Guo QN, Xiao HY, Chen C, Xiao YL, Dong HH, Bo Y, Ming ZH, Guang GY (2020) A novel effluent quality predicting model based on genetic-deep belief network algorithm for cleaner production in a full-scale paper-making wastewater treatment. J Clean Prod 265:121787. https://doi.org/10.1016/j.jclepro.2020.121787
Hairu W, Maotao Y, Rong W, Qingchuan F, Xiangyuan C, Jianhong X (2023) Assessment of schedulable potential for user-side resources participating in demand response based on improved G1 method using SRC. Front Energy Res 11. https://doi.org/10.3389/fenrg.2023.1248322
Han X, Liu X, Gao D, Ma B, Gao X, Cheng M (2022) Costs and benefits of the development methods of drinking water quality index: a systematic review. Ecol Indic 144:109501. https://doi.org/10.1016/j.ecolind.2022.109501
Hong HC, Zhang ZY, Guo AD, Shen LG, Sun HJ, Ling Y, Wu FY, Lin HJ (2020) Radial basis function artificial neural network (RBF ANN) as well as the hybrid method of RBF ANN and grey relational analysis able to well predict trihalomethanes levels in tap water. J Hydrol 591:125574. https://doi.org/10.1016/j.jhydrol.2020.125574
Hong SM, Abbas A, Kim S, Kwon DH, Yoon N, Yun D, Lee S, Pachepsky Y, Pyo J, Cho KH (2023) Autonomous calibration of EFDC for predicting chlorophyll-a using reinforcement learning and a real-time monitoring system. Environ Model Softw 168:105805. https://doi.org/10.1016/j.envsoft.2023.105805
Huang M, Tian D, Liu H, Zhang C, Yi X, Cai J, Ruan J, Zhang T, Kong S, Ying G (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 2018. https://doi.org/10.1155/2018/8241342
Huutuan T, Mohammed H, Thi THN, Hong GH, Minhky N, Khoi NN, Daiviet NV (2023) Machine learning approaches for predicting microplastic pollution in peatland areas. Mar Pollut Bull 194(A):115417. https://doi.org/10.1016/j.marpolbul.2023.115417
Ibrahim A, Ismail A, Juahir H, Iliyasu AB, Wailare BT, Mukhtar M, Aminu H (2022) Water quality modelling using principal component analysis and artificial neural network. Mar Pollut Bull 187:114493. https://doi.org/10.1016/j.marpolbul.2022.114493
Jaddi NS, Abdullah S (2017) A cooperative-competitive master-slave global-best harmony search for ANN optimization and water-quality prediction. Appl Soft Comput 51:209–224. https://doi.org/10.1016/j.asoc.2016.12.011
Jiang Y, Li C, Sun L, Guo D, Zhang Y, Wang W (2021a) 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
Jiang P, Huang Y, Liu X, Zhang J, Gin KYH (2021b) A feature reconstruction-based multi-task regression model for cyanobacterial distribution forecasting along the water column. J Clean Prod 292:126025. https://doi.org/10.1016/j.jclepro.2021.126025
Jiang C, Li D, Gao Y, Liu W, Zhang L (2017) Impact of climate variability and anthropogenic activity on streamflow in the Three Rivers Headwater Region, Tibetan Plateau, China. Theor Appl Climatol 129:667–681. https://doi.org/10.1007/s00704-016-1833-7
Jie C, Zhixin L, Zhengtong Y, Xuan L, Xiaolu L, Lirong Y, Wenfeng Z (2023) Predict the effect of meteorological factors on haze using BP neural network. Urban Clim 51:101630. https://doi.org/10.1016/j.uclim.2023.101630
Juna A, Umer M, Sadiq S, Karamti H, Eshmawi AA, Mohamed A, Ashraf I (2022) Water quality prediction using KNN imputer and multilayer perceptron. Water 14:2592. https://doi.org/10.3390/w14172592
Junfeng S, Haitao Z, Tengjiang Y, Yang S, Yancheng L (2023) Microscopic void characteristics of OGFC in Sponge City and its effect on noise reduction performance using X-ray CT. Constr Build Mater 408:133729. https://doi.org/10.1016/j.conbuildmat.2023.133729
Ke W, Panlong W, Xingxiu L, Shan H, Jimin L (2023) An adaptive outlier-robust Kalman filter based on sliding window and Pearson type VII distribution modeling. Signal Process 216:109306. https://doi.org/10.1016/j.sigpro.2023.109306
Kermorvant C, Liquet B, Litt G, Jones JB, Mengersen K, Peterson EE, Hyndman RJ, Leigh C (2021) Reconstructing missing and anomalous data collected from high-frequency in-situ sensors in fresh waters. Int J Environ Res Public Health 18:12803. https://doi.org/10.3390/ijerph182312803
Khatri N, Vyas AK, Abdul-Qawy ASH, Rene ER (2023) Artificial neural network based models for predicting the effluent quality of a combined upflow anaerobic sludge blanket and facultative pond: performance evaluation and comparison of different algorithms. Environ Res 217:114843. https://doi.org/10.1016/j.envres.2022.114843
Kisi O, Shiri J, Demir V (2017) Chapter 3 - Hydrological time series forecasting using three different heuristic regression techniques. In: Handbook of neural computation, pp 45–65. https://doi.org/10.1016/B978-0-12-811318-9.00003-X
Kumar P, Lai SH, Mohd NS, Kamal MR, Afan HA, Ahmed AN, Sherif M, Sefelnasr A, El-Shafie A (2020) Optimised neural network model for river-nitrogen prediction utilizing a new training approach. PLoS One 15(9). https://doi.org/10.1371/journal.pone.0239509
Lange B, Højstrup J (2001) Evaluation of the wind-resource estimation program WAsP for offshore applications. J Wind Eng Ind Aerodyn 89(3-4):271–291. https://doi.org/10.1016/S0167-6105(00)00082-9
Lehmann A, Rode M (2001) Long-term behaviour and cross-correlation water quality analysis of the river Elbe, Germany. Water Res 35(9):2153–2160
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:19879–19896. https://doi.org/10.1007/s11356-019-05116-y
Li D, Marshall L, Liang Z, Sharma A, Zhou Y (2021) Characterizing distributed hydrological model residual errors using a probabilistic long short-term memory network. J Hydrol 603(A):126888. https://doi.org/10.1016/j.jhydrol.2021.126888
Li B, Lu C, Zhao J, Tian J, Sun J, Hu C (2023) Operational parameter prediction of electrocoagulation system in a rural decentralized water treatment plant by interpretable machine learning model. J Environ Manag 333:117416. https://doi.org/10.1016/j.jenvman.2023.117416
Liang Z, Zou R, Chen X, Ren T, Su H, Liu Y (2020) Simulate the forecast capacity of a complicated water quality model using the long short-term memory approach. J Hydrol 581:124432. https://doi.org/10.1016/j.jhydrol.2019.124432
Liu P, Wang J, Sangaiah A, Xie Y, Yin X (2019) Analysis and prediction of water quality using LSTM deep neural networks in IoT environment. Sustainability 11(7). https://doi.org/10.3390/su11072058
Lu F, Zhang H, Liu W (2020) Development and application of a GIS-based artificial neural network system for water quality prediction: a case study at the Lake Champlain area. J Ocean Limnol 38:1835–1845. https://doi.org/10.1007/s00343-019-9174-x
Lu X, Magar R, Farimani AB (2022) Forecasting COVID-19 new cases using deep learning methods. Comput Biol Med 144:105342. https://doi.org/10.1016/j.compbiomed.2022.105342
Lu Y, Zhang Y, Cao X, Wang C, Wang Y, Zhang M, Robert CF, Alan J, Yuan J, Mark JB, Chen D, Tian H, Li H, Ulrich VW, Zhang Z (2019) Forty years of reform and opening up: China’s progress toward a sustainable path. Sci Adv 5:eaau9413. https://doi.org/10.1126/sciadv.aau9413
Haq RAKP, Harigovindan VP (2022) Water quality prediction for smart aquaculture using hybrid deep learning models. IEEE Access 10:60078–60098. https://doi.org/10.1109/ACCESS.2022.3180482
Manteaux S, Sauvage S, Samie R, Monteil C, Garnier J, Thieu V, Cakir R, Sánchez-Pérez JM (2023) Modeling in-stream biogeochemical processes at catchment scale: coupling SWAT and RIVE models. Environ Model Softw 170:105856. https://doi.org/10.1016/j.envsoft.2023.105856
Moon T, Hong S, Choi YH, Jung DH, Chang SH, Son JE (2019) Interpolation of greenhouse environment data using multilayer perceptron. Comput Electron Agric 166:105023. https://doi.org/10.1016/j.compag.2019.105023
Mostafiz R, Uddin MS, Alam NA, Reza M, Rahman MM (2022) Covid-19 detection in chest X-ray through random forest classifier using a hybridization of deep CNN and DWT optimized features. J King Univ – Comput Inform Sci 34(6PB):1319–1578. https://doi.org/10.1016/j.jksuci.2020.12.010
Mustafa AS, Sulaiman SO, Shahooth SH (2017) Application of QUAL2K for water quality modeling and management in the lower reach of the Diyala river. Iraqi J Civ Eng 11:66–80. https://doi.org/10.6000/1929-6002.2017.06.02.3
Nhantumbo CL, Carvalho F, Uvo C, Larsson R, Larson M (2018) Applicability of a processes-based model and artificial neural networks to estimate the concentration of major ions in rivers. J Geochem Explor 193:32–40. https://doi.org/10.1016/j.gexplo.2018.07.003
Ni QJ, Cao XH, Tan CQ, Peng WQ, Kang XY (2023) An improved graph convolutional network with feature and temporal attention for multivariate water quality prediction. Environ Sci Pollut Res Int 30(5):11516–11529. https://doi.org/10.1007/s11356-022-22719-0
Poor CJ, Ullman JL (2010) Using regression tree analysis to improve predictions of low-flow nitrate and chloride in Willamette River Basin watersheds. Environ Manag 46(5):771–780. https://doi.org/10.1007/s00267-010-9550-y
Pyo J, Park LJ, Pachepsky Y, Baek SS, Kim K, Cho KH (2020) Using convolutional neural network for predicting cyanobacteria concentrations in river water. Water Res 186:116349. https://doi.org/10.1016/j.watres.2020.116349
Ray SS, Verma RK, Singh A, Myung S, Park YI, Kim IC, Lee HK, Kwon YN (2022) Exploration of time series model for predictive evaluation of long-term performance of membrane distillation desalination. Process Saf Environ Prot 160:1–12. https://doi.org/10.1016/j.psep.2022.01.058
Rizk Y, Awad M (2019) On extreme learning machines in sequential and time series prediction: a non-iterative and approximate training algorithm for recurrent neural networks. Neurocomputing 325:1–19. https://doi.org/10.1016/j.neucom.2018.09.012
Sarfaraz HZ, Ömer FG, Pradip K, Hande K (2022) Container vessel selection for maritime shipping companies by using an extended version of the grey relation analysis (GRA) with the help of type-2 neutrosophic fuzzy sets (T2NFN). Comput Ind Eng 171:108376. https://doi.org/10.1016/j.cie.2022.108376
Shaheer A, Afida A, Hossain ML, Aini H, Maher GA, Muhammad AAMZ, Mohamad HMS (2023) Optimized data-driven approach for remaining useful life prediction of lithium-ion batteries based on sliding window and systematic sampling. Journal of Energy Storage 73:109198. https://doi.org/10.1016/j.est.2023.109198
Shahid F, Mehmood A, Khan R, Smadi AA, Yaqub M, Alsmadi MK, Zheng Z (2023) 1D Convolutional LSTM-based wind power prediction integrated with PkNN data imputation technique. J King Univ – Comput Inform Sci 35(10):101816. https://doi.org/10.1016/j.jksuci.2023.101816
Shim J, Hong S, Lee J, Lee S, Kim YM, Chon K, Park S, Cho KH (2023) Deep learning with data preprocessing methods for water quality prediction in ultrafiltration. J Clean Prod 428:139217. https://doi.org/10.1016/j.jclepro.2023.139217
Shuai Y, Xie J, Lu K, Chen Z (2023) Multi-attention network for sewage treatment plant detection. Sustainability 15:5880. https://doi.org/10.3390/su15075880
Song C, Yao L, Hua C, Ni Q (2021a) A novel hybrid model for water quality prediction based on synchrosqueezed wavelet transform technique and improved long short-term memory. J Hydrol 603(A):126879. https://doi.org/10.1016/j.jhydrol.2021.126879
Song C, Yao L, Hua C, Ni Q (2021b) A water quality prediction model based on variational mode decomposition and the least squares support vector machine optimized by the sparrow search algorithm (VMD-SSA-LSSVM) of the Yangtze River, China. Environ Monit Assess 193:363. https://doi.org/10.1007/s10661-021-09127-6
Su X, He X, Zhang G, Chen YH, Li KY (2022) Research on SVR water quality prediction model based on improved sparrow search algorithm. Comput Intell Neurosci. https://doi.org/10.1155/2022/7327072
Sudani ZAA, Salih SQ, Sharafati A, Yaseen ZM (2019) Development of multivariate adaptive regression spline integrated with differential evolution model for streamflow simulation. J Hydrol 573:1–12. https://doi.org/10.1016/j.jhydrol.2019.03.004
Sun X, Wang X, Cai D, Li Z, Gao Y, Wang X (2020) Multivariate seawater quality prediction based on PCA-RVM supported by edge computing towards smart ocean. IEEE Access 8:54506–54513. https://doi.org/10.1109/ACCESS.2020.2981528
Sun X, Zhang Y, Shi K, Zhang Y, Li N, Wang W, Huang X, Qin B (2022) Monitoring water quality using proximal remote sensing technology. Sci Total Environ 803:149805. https://doi.org/10.1016/j.scitotenv.2021.149805
Suraj KB, Tran MT, Zaher MY (2020) Heavy metal contamination prediction using ensemble model: case study of Bay sedimentation, Australia. J Hazard Mater 403:123492. https://doi.org/10.1016/j.jhazmat.2020.123492
Su J, Ji D, Lin M, Chen Y, Sun Y, Huo S, Zhu J, Xi B (2017) Developing surface water quality standards in China. Resour Conserv Recycl 117:294–303. https://doi.org/10.1016/j.resconrec.2016.08.003
Tan R, Wang Z, Wu T, Wu J (2023) A data-driven model for water quality prediction in Tai Lake, China, using secondary modal decomposition with multidimensional external features. J Hydrol : Region Stud 47:101435. https://doi.org/10.1016/j.ejrh.2023.101435
Tao T, Xin K (2014) A sustainable plan for China’s drinking water: tackling pollution and using different grades of water for different tasks is more efficient than making all water potable. Nature 511:527–529
Tchobanoglous G, Schroeder EE. (1985) Water quality: characteristics, modeling, modification.
Tian S, Zhang J, Shu X, Chen L, Niu L, Wang Y (2022a) A novel evaluation strategy to artificial neural network model based on bionics. J Bionic Eng 19:224–239. https://doi.org/10.1007/s42235-021-00136-2
Tian X, Wang Z, Taalab E, Zhang B, Li X, Wang J, Ong MC, Zhu Z (2022b) Water quality predictions based on grey relation analysis enhanced LSTM algorithms. Water 14:3851. https://doi.org/10.3390/w14233851
Tiyasha TTM, Yaseen ZM (2020) A survey on river water quality modelling using artificial intelligence models: 2000–2020. J Hydrol 585:124670. https://doi.org/10.1016/j.jhydrol.2020.124670
Vaswani A, Shazeer N, Parmar N, Uszkoreit J, Jones L, Gomez AN, Kaiser L, Polosukhin I (2017) Attention is all you need. In: Conference and Workshop on Neural Information Processing Systems Mon, 12. https://doi.org/10.48550/arXiv.1706.03762
Wang X, Wang K, Ding J, Chen X, Li Y, Zhang W (2021) Predicting water quality during urbanization based on a causality-based input variable selection method modified back-propagation neural network. Environ Sci Pollut Res Int 28:960–973. https://doi.org/10.1007/s11356-020-10514-8
Wang Y, Zhou J, Chen K, Wang Y, Liu L (2017) Water quality prediction method based on LSTM neural network. In: 2017 12th International Conference on Intelligent Systems and Knowledge Engineering (ISKE), Nanjing, China, pp 1–5. https://doi.org/10.1109/ISKE.2017.8258814
Wang D, Tao T, Zhang S, Zhao J, Wang Z, Zhang Z, Wu C, Xie S, Lv Y, Qian G (2023) Tributaries protect the water quality of central lake in an artificial spiderweb-structured water system, from the perspective of runoff pollution control. J Clean Prod 429:139566. https://doi.org/10.1016/j.jclepro.2023.139566
Wang C, Schürz C, Zoboli O, Zessner M, Schulz K, Watzinger A, Bodner G, Mehdi-Schulz B (2022a) N2O emissions from two Austrian agricultural catchments simulated with an N2O submodule developed for the SWAT model. Atmosphere 13:50. https://doi.org/10.3390/atmos13010050
Wang H, Liu Z, Peng D, Cheng Z (2022b) Attention-guided joint learning CNN with noise robustness for bearing fault diagnosis and vibration signal denoising. ISA Trans 128(Pt B):470–484. https://doi.org/10.1016/j.isatra.2021.11.028
Wu Z, Lu C, Sun Q, Lu W, He X, Qin T, Yan L, Wu C (2023) Predicting groundwater level based on machine learning: a case study of the Hebei Plain. Water 15:823. https://doi.org/10.3390/w15040823
Xie Z, Li Z, Mo C, Wang J (2022) A PCA–EEMD–CNN–attention–GRU–encoder–decoder accurate prediction model for key parameters of seawater quality in Zhanjiang Bay. Materials 15:5200. https://doi.org/10.3390/ma15155200
Xie J, Yao B (2022) Physics-constrained deep active learning for spatiotemporal modeling of cardiac electrodynamics. Comput Biol Med 146:105586. https://doi.org/10.1016/j.compbiomed.2022.105586
Yaseen ZM, Sulaiman SO, Deo RC, Chau KW (2018) An enhanced extreme learning machine model for river flow forecasting: state-of-the-art, practical applications in water resource engineering area and future research direction. J Hydrol 569:387–408. https://doi.org/10.1016/j.jhydrol.2018.11.069
Zhang YF, Fitch P, Thorburn PJ (2020) Predicting the trend of dissolved oxygen based on the kPCA-RNN model. Water 12:585. https://doi.org/10.3390/w12020585
Zhang H, Cao X, Huo S, Ma C, Li W, Liu Y, Tong Y, Wu F (2023a) Changes in China’s river water quality since 1980: management implications from sustainable development. Clean Water 6:45. https://doi.org/10.1038/s41545-023-00260-y
Zhang S, Gao H, Wei G, Chen X (2021) Grey relational analysis method based on cumulative prospect theory for intuitionistic fuzzy multi-attribute group decision making. J Intell Fuzzy Syst 41(2):3783–3795. https://doi.org/10.3233/JIFS-211461
Zhang X, Chen X, Zheng G, Cao G (2023b) Improved prediction of chlorophyll-a concentrations in reservoirs by GRU neural network based on particle swarm algorithm optimized variational modal decomposition. Environ Res 221:115259. https://doi.org/10.1016/j.envres.2023.115259
Zhang Q, Wang R, Qi Y, Wen F (2022) A watershed water quality prediction model based on attention mechanism and Bi-LSTM. Environ Sci Pollut Res 29:75664–75680. https://doi.org/10.1007/s11356-022-21115-y
Zhou Q, Qu S, Wang Q, She Y, Yu Y, Bi J (2023) Sliding window-based machine learning for environmental inspection resource allocation. Environ Sci Technol 57(44):16743–16754. https://doi.org/10.1021/acs.est.3c05088
Zhou Y, Zhang Q, Singh PV, Xiao MZ (2015) General correlation analysis: a new algorithm and application. Stoch Env Res Risk A 29(3):665–677. https://doi.org/10.1007/s00477-014-0970-8
Zhu Q, Cai Y (2023) Integrating ecological risk, ecosystem health, and ecosystem services for assessing regional ecological security and its driving factors: insights from a large river basin in China. Ecol Indic 155:110954. https://doi.org/10.1016/j.ecolind.2023.110954
Acknowledgements
The authors thank the reviewers and the editor for their constructive comments.
Funding
This research was funded by the National Natural Science Foundation of China (51874010) and the Natural Science Research Project of the Anhui Provincial Department of Education University (KJ2018A0087).
Author information
Authors and Affiliations
Contributions
JC and HL designed the study; MF, YC, and KZ conducted the research; YC and KZ analyzed the data; JC and HL wrote the manuscript. All authors reviewed the manuscript.
Corresponding author
Ethics declarations
Ethics approval and consent to participate
Not applicable.
Consent for publication
All authors agree to publish.
Competing interests
The authors declare no competing interests.
Additional information
Responsible Editor: Marcus Schulz
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
Chen, J., Li, H., Felix, M. et al. Water quality prediction of artificial intelligence model: a case of Huaihe River Basin, China. Environ Sci Pollut Res 31, 14610–14640 (2024). https://doi.org/10.1007/s11356-024-32061-2
Received:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s11356-024-32061-2