skip to main content
10.1145/3590003.3590037acmotherconferencesArticle/Chapter ViewAbstractPublication PagescacmlConference Proceedingsconference-collections
research-article

A water quality parameter prediction method based on transformer architecture and multi-sensor data fusion

Authors Info & Claims
Published:29 May 2023Publication History

ABSTRACT

Water quality monitoring provides a basis for water quality control and water resources management. Prediction of water quality parameters can plan water use strategies, prevent further water pollution and improve water resource utilization efficiency. We propose a water quality parameter prediction method based on transformer architecture model and multi-sensor data fusion. The proposed multiple water quality parameter prediction model accepts multiple types of water quality parameter data input at the same time. The data embedding module integrates multiple types of water quality parameter information and assigns a unique position code to the data at each time step. The self-attention mechanism of the model mining the potential correlation between different time step data. The model can learn the internal relationship of the fusion data of multiple water quality parameters, and effectively predict the future trend of water quality parameters. The effectiveness of the proposed algorithm is verified by the measured data, and the advantages of the proposed method are verified by comparative experiments.

References

  1. R. Gong, L. Xu, D. Wang, H. Li, and J. Xu, “Water Quality Modeling for a Typical Urban Lake Based on the EFDC Model,” Environ. Model. Assess., vol. 21, no. 5, pp. 643–655, Oct. 2016, doi: 10.1007/s10666-016-9519-1.Google ScholarGoogle ScholarCross RefCross Ref
  2. J. Liang, Q. Yang, T. Sun, J. D. Martin, H. Sun, and L. Li, “MIKE 11 model-based water quality model as a tool for the evaluation of water quality management plans,” J. Water Supply Res. Technol.-Aqua, vol. 64, no. 6, pp. 708–718, Sep. 2015, doi: 10.2166/aqua.2015.048.Google ScholarGoogle ScholarCross RefCross Ref
  3. S. A. Dellana and D. West, “Predictive modeling for wastewater applications: Linear and nonlinear approaches,” Environ. Model. Softw., vol. 24, no. 1, pp. 96–106, Jan. 2009, doi: 10.1016/j.envsoft.2008.06.002.Google ScholarGoogle ScholarDigital LibraryDigital Library
  4. P. T. M. Hanh, N. V. Anh, D. T. Ba, S. Sthiannopkao, and K.-W. Kim, “Analysis of variation and relation of climate, hydrology and water quality in the lower Mekong River,” Water Sci. Technol., vol. 62, no. 7, pp. 1587–1594, Oct. 2010, doi: 10.2166/wst.2010.449.Google ScholarGoogle ScholarCross RefCross Ref
  5. D. Ömer Faruk, “A hybrid neural network and ARIMA model for water quality time series prediction,” Eng. Appl. Artif. Intell., vol. 23, no. 4, pp. 586–594, Jun. 2010, doi: 10.1016/j.engappai.2009.09.015.Google ScholarGoogle ScholarDigital LibraryDigital Library
  6. H. Chen , “A deep learning CNN architecture applied in smart near-infrared analysis of water pollution for agricultural irrigation resources,” Agric. Water Manag., vol. 240, p. 106303, Oct. 2020, doi: 10.1016/j.agwat.2020.106303.Google ScholarGoogle ScholarCross RefCross Ref
  7. D. Valadkhan, R. Moghaddasi, and A. Mohammadinejad, “Groundwater quality prediction based on LSTM RNN: An Iranian experience,” Int. J. Environ. Sci. Technol., vol. 19, no. 11, pp. 11397–11408, Nov. 2022, doi: 10.1007/s13762-022-04356-9.Google ScholarGoogle ScholarCross RefCross Ref
  8. W. Tan , “Application of CNN and Long Short-Term Memory Network in Water Quality Predicting,” Intell. Autom. Soft Comput., vol. 34, no. 3, pp. 1943–1958, 2022, doi: 10.32604/iasc.2022.029660.Google ScholarGoogle ScholarCross RefCross Ref
  9. A. Vaswani , “Attention Is All You Need.” arXiv, Dec. 05, 2017. Available: http://arxiv.org/abs/1706.03762Google ScholarGoogle Scholar
  10. W. Li, H. Fu, Z. Han, X. Zhang, and H. Jin, “Intelligent tool wear prediction based on Informer encoder and stacked bidirectional gated recurrent unit,” Robot. Comput.-Integr. Manuf., vol. 77, p. 102368, Oct. 2022, doi: 10.1016/j.rcim.2022.102368.Google ScholarGoogle ScholarCross RefCross Ref
  11. H. Zhou , “Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecasting.” arXiv, Mar. 28, 2021. Available: http://arxiv.org/abs/2012.07436Google ScholarGoogle Scholar
  12. H.-K. Wang, K. Song, and Y. Cheng, “A Hybrid Forecasting Model Based on CNN and Informer for Short-Term Wind Power,” Front. Energy Res., vol. 9, p. 788320, Jan. 2022, doi: 10.3389/fenrg.2021.788320.Google ScholarGoogle ScholarCross RefCross Ref
  13. S. Yao, Y. Zhang, P. Wang, Z. Xu, Y. Wang, and Y. Zhang, “Long-Term Water Quality Prediction Using Integrated Water Quality Indices and Advanced Deep Learning Models: A Case Study of Chaohu Lake, China, 2019–2022,” Appl. Sci., vol. 12, no. 22, p. 11329, Nov. 2022, doi: 10.3390/app122211329.Google ScholarGoogle ScholarCross RefCross Ref

Index Terms

  1. A water quality parameter prediction method based on transformer architecture and multi-sensor data fusion
        Index terms have been assigned to the content through auto-classification.

        Recommendations

        Comments

        Login options

        Check if you have access through your login credentials or your institution to get full access on this article.

        Sign in
        • Published in

          cover image ACM Other conferences
          CACML '23: Proceedings of the 2023 2nd Asia Conference on Algorithms, Computing and Machine Learning
          March 2023
          598 pages
          ISBN:9781450399449
          DOI:10.1145/3590003

          Copyright © 2023 ACM

          Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

          Publisher

          Association for Computing Machinery

          New York, NY, United States

          Publication History

          • Published: 29 May 2023

          Permissions

          Request permissions about this article.

          Request Permissions

          Check for updates

          Qualifiers

          • research-article
          • Research
          • Refereed limited

          Acceptance Rates

          CACML '23 Paper Acceptance Rate93of241submissions,39%Overall Acceptance Rate93of241submissions,39%
        • Article Metrics

          • Downloads (Last 12 months)46
          • Downloads (Last 6 weeks)10

          Other Metrics

        PDF Format

        View or Download as a PDF file.

        PDF

        eReader

        View online with eReader.

        eReader

        HTML Format

        View this article in HTML Format .

        View HTML Format