Abstract
It is essential to make an accurate prediction of the concentration of dissolved oxygen (DO), hydrogen ion concentration (pH), and potassium permanganate (KMnO4) in order to ensure the quality of the drinking water. The lack of monitoring data and the large fluctuation increase the difficulty of predicting DO, pH, and KMnO4 in the tide-sensing estuary. In this research, improved grey association analysis (IGRA) was provided to determine the correlation between DO, pH, KMnO4, and other water quality indicators, thereby resolving the dimension disaster problem of long short-term memory (LSTM). Furthermore, LSTM based on the improved sparrow search algorithm (ISSA) was established, and five LSTM parameters—learning rate, batch size, training times, hidden layer nodes, and fully connected hidden layer nodes—are automatically optimized, which could accurately predict the concentration of DO, pH, and KMnO4. Using the data from the Qiantang River Gate Observation Station from November 8, 2020, to June 27, 2021, 70% of which were training sets and 30% of which were test sets, predicted data for day 4. The results show that the coefficient of determination (R2) of the IGRA-ISSA-LSTM model for DO, pH, and KMnO4 were 0.92, 0.93, and 0.726, respectively, which are greater than that of IGRA-BP model, IGRA-LSTM model, and IGRA-SSA-LSTM model. Therefore, this research provides technical support for water quality management in tidal estuaries.
Similar content being viewed by others
Data Availability
The data that support the China Environmental Monitoring Station.
References
Arpan, K. K. (2016). Bio inspired computing a review of algorithms and scope of applications. Expert Systems with Applications, 59, 20–32. https://doi.org/10.1016/j.eswa.2016.04.018
Bevilacqua, A., & Mazzanti, A. (2021). Doubly-tuned transformer networks: A tutorial. IEEE Transactions on Circuits and Systems II: Express Briefs, 68(2), 550–555. https://doi.org/10.1109/TCSII.2020.3046021
Boudardara, F., & Gorkemli, B. (2020). Solving artificial ant problem using two artificial bee colony programming versions. Applied Intelligence, 50, 3695–3717. https://doi.org/10.1007/s10489-020-01741-0
Cao, S. Q., Zhou, L. X., & Zhang, Z. (2021). A prediction model of dissolved oxygen in aquaculture was developed based on improved short and long-term memory neural network. Transactions of the Chinese Society of Agricultural Engineering, 37(1–4), 235–242. https://doi.org/10.11975/j.issn.1002-6819.2021.14.027
Chia, S. L., Chia, M. Y., Chai, H. K., & Huang, Y. F. (2022). Integration of advanced optimization algorithms into least-square support vector machine (LSSVM) for water quality index prediction. Water Supply, 22(2), 1951–1963. https://doi.org/10.2166/ws.2021.303
Deepa, P., & Thilagavathi, K. (2015). Data reduction techniques of hyperspectral images: A comparative study.International Conference on Signal Processing (pp. 1–6). Communication and Networking (ICSCN). https://doi.org/10.1109/ICSCN.2015.7219866
Du, L. J., Tang, X. L., Zhou, J., Chen, Y. L., & Chen, J. (2022). Classification of Alzheimer’s disease based on attention mechanism and multi-task learning. Computer Science, 49(S1), 60–65. https://doi.org/10.11896/jsjkx.201200072
Hu Z, Zhang Y, Zhao Y, Xie M, Zhong J, Tu Z & Liu J. (2019). A water quality prediction method based on the deep LSTM network considering correlation in smart mariculture. Sensors (Basel, Switzerland), 19(6). https://doi.org/10.3390/s19061420
Jiang, X. R. (2021). Research on construction period risk management of housing projects based on principal component analysis. Qingdao University. https://doi.org/10.27262/d.cnki.gqdau.2021.001691
Karaboga, D., Gorkemli, B., Ozturk, C., & Karaboga, N. (2014). A comprehensive survey: Artificial bee colony (ABC) algorithm and applications. Artificial Intelligence Review, 42(1), 21–57. https://doi.org/10.1007/s10462-012-9328-0
Li, L., Jiang, P., Xu, H., Lin, G., Guo, D., & Wu, H. (2019). Water quality prediction based on recurrent neural network and impro-ved evidence theory: A case study of Qiantang River, China. Environmental Science and Pollution Research, 26(19), 19879–19896. https://doi.org/10.1007/s11356-019-05116-y
Liu, F. T., Ting, K. M., & Zhou, Z. H. (2008). Isolation forest. Eighth IEEE International Conference on Data Mining. (pp. 413–422)https://doi.org/10.1109/ICDM.2008.17
Loos, S., Shin, C. M., Sumihar, J., Kim, K., Cho, J., & Weerts, A. H. (2020). Ensemble data assimilation methods for improving river water quality forecasting accuracy. Water Res, 1–71, 115343.
Ong, B. T., Sugiura, K., & Zettsu, K. (2014). Dynamic pre-training of deep recurrent neural networks for predicting environmental monitoring data. IEEE International Conference on Big Data (Big Data) (pp. 760–765). IEEE. https://doi.org/10.1109/BigData.2014.7004302
Pascanu, R., Mikolov, T., & Bengio, Y. (2013). On the difficulty of training recurrent neural networks[C]//International conference on machine learning. Pmlr, 1310–1318. https://doi.org/10.48550/arXiv.1211.5063
Pravallika, M. S., Vasavi, S., & Vighneshwar, S. P. (2022). Prediction of temperature anomaly in Indian Ocean based on autoregressive long short-term memory neural network. Neural Computing and Applications, 34, 7537–7545. https://doi.org/10.1007/s00521-021-06878-8
Rani P & Mahapatra G S. (2018). Neural network for software reliability analysis of dynamically weighted NHPP growth models with imperfect debugging. Software Testing Verification & Reliability, 28(5). https://doi.org/10.1002/stvr.1663
Sakaa, B., Elbeltagi, A., Boudibi, S., Chaffaï, H., Islam, A. R. M. T., Kulimushi, L. C., Choudhari, P., Hani, A., Brouziyne, Y., & Wong, Y. J. (2022). Water quality index modeling using random forest and improved SMO algorithm for support vector machine in Saf-Saf river basin. Environmental Science and Pollution Research, 29(32), 48491–48508. https://doi.org/10.1007/s11356-022-18644-x
Sun M, Wei S K, Wang Y J, Zhao J D & Yuan M X. (2020). LSTM water quality prediction model based on wavelet decomposition. Computer System Application, (12),55–63. https://doi.org/10.15888/j.cnki.csa.007695
Tang, J., Liu, G., & Pan, Q. T. (2021). A review on representative swarm intelligence algorithms for solving optimization problems: Applications and trends. IEEE/CAA Journal of Automatica Sinica, 8(10), 1627–1643. https://doi.org/10.1109/JAS.2021.1004129
Wai, K. P., Chia, M. Y., Koo, C. H., Huang, Y. F., & Chong, W. C. (2022). Applications of deep learning in water quality management: A state-of-the-art review. Journal of Hydrology, 613(1), 128332. https://doi.org/10.1016/j.jhydrol.2022.128332
Wang, J. F., Yu, J. Q., Zhao, A. J., & Li, R. L. (2019). Warehouse picking path optimization based on swarm intelligence algorithm. Industrial Control Computer, 32(03), 61–63. https://doi.org/10.3969/j.issn.1001-182X.2019.03.027
Wong, Y. J., Shimizu, Y., & He, K. (2020). Comparison among different ASEAN water quality indices for the assessment of the spatial variation of surface water quality in the Selangor river basin. Malaysia. Environ Monit Assess, 192(10), 644. https://doi.org/10.1007/s10661-020-08543-4
Wong, Y. J., Shimizu, Y., Kamiya, A., Maneechot, L., Bharambe, K. P., Fong, C. S., Nik, S., & Nik, M. (2021). Application of artificial intelligence methods for monsoonal river classification in Selangor river basin, Malaysia. Environmental Monitoring and Assessment, 193(7), 438. https://doi.org/10.1007/s10661-021-09202-y
Wu, J., Li, Z., Zhu, L., Li, G., & Peng, F. (2018). Optimized BP neural network for dissolved oxygen prediction. IFAC-Papers OnLine, 51(17), 596–601. https://doi.org/10.1016/j.ifacol.2018.08.132
Xiong W Z, Chen X Q & Li H J. (2017). Study on water quality evaluation method based on PCA-SVM. Jingchu Institute of Technology, (02):62–66. https://doi.org/10.14151/j.cnki.jclgxyxb.2017.02.011
Xue, J. K. (2020). Research and application of a new swarm intelligence optimization technology. Donghua university. https://doi.org/10.27012/d.cnki.gdhuu.2020.000178
Yang, M., Xu, P. P., & Qian, H. (2018). Water quality model of BP double hidden layer neural network based on artificial bee colony algorithm [J]. Environmental monitoring management and technology, 30(01), 21–26. https://doi.org/10.19501/j.cnki.1006-2009.20171229.014
YangJiaqiang, H. D. H. D. (2009). An adaptive chaos immune optimization algorithm with mutative scale and its application. Control Theory & Applications, 6(10), 1069–1074. https://doi.org/10.7641/j.issn.1000-8152.2009.10.CCTA080581
Zhao, Y. Y., & Ji, J. (2021). Missing data recovery of national and provincial trunk Lines based on Lagrange interpolation method. Wireless Interconnection Technology, 18(10), 97–101. https://doi.org/10.3969/j.issn.1672-6944.2021.10.048
Zhou, P. J. (2020). Construction and application of water quality prediction model based on PLS-GWO-SVR. Yanshan university. https://doi.org/10.27440/d.cnki.gysdu.2020.000837
Zhou, J., Wang, Y. Y., Xiao, F., & Sun, L. J. (2018). Water quality prediction method based on IGRA and LSTM. Water, 10, 1148. https://doi.org/10.3390/w10091148
Zhou X Z, Chen Y J & Pu W P. (2020). Application of machine learning in water quality prediction. Proceedings of the Annual Conference of Science and Technology. Chinese Society for Environmental Sciences, 4:. https://doi.org/10.26914/c.cnkihy.2020.039984
Zou Q H, Xiong Q Y, Li Q D, Yi H L, Yu Y & W C. (2020). A water quality prediction method based on the multi-time scale bidirectional long short-term memory network[J]. Environmental Science and Pollution Research, 27(2). https://doi.org/10.1007/s11356-020-08087-7
Acknowledgements
This study is funded by the Natural Science Foundation of Zhejiang Province (LQ20E090006).
Author information
Authors and Affiliations
Corresponding author
Ethics declarations
Competing Interests
The authors declare no competing interests.
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
Jiange, J., Liqin, Z., Senjun, H. et al. Water quality prediction based on IGRA-ISSA-LSTM model. Water Air Soil Pollut 234, 172 (2023). https://doi.org/10.1007/s11270-023-06117-x
Received:
Accepted:
Published:
DOI: https://doi.org/10.1007/s11270-023-06117-x