Water Quality Prediction Based on LSTM and Attention Mechanism: A Case Study of the Burnett River, Australia

Honglei Chen, Junbo Yang, Xiaohua Fu (), Qingxing Zheng, Xinyu Song, Zeding Fu, Jiacheng Wang, Yingqi Liang, Hailong Yin, Zhiming Liu (), Jie Jiang, He Wang and Xinxin Yang
Additional contact information
Honglei Chen: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Junbo Yang: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Xiaohua Fu: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Qingxing Zheng: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Xinyu Song: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Zeding Fu: School of Hydraulic and Environmental Engineering, Changsha University of Science & Technology, Changsha 410114, China
Jiacheng Wang: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Yingqi Liang: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Hailong Yin: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Zhiming Liu: Department of Biology, Eastern New Mexico University, Portales, NM 88130, USA
Jie Jiang: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
He Wang: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China
Xinxin Yang: Ecological Environment Management and Assessment Center, Central South University of Forestry and Technology, Changsha 410004, China

Sustainability, 2022, vol. 14, issue 20, 1-20

Abstract: Prediction of water quality is a critical aspect of water pollution control and prevention. The trend of water quality can be predicted using historical data collected from water quality monitoring and management of water environment. The present study aims to develop a long short-term memory (LSTM) network and its attention-based (AT-LSTM) model to achieve the prediction of water quality in the Burnett River of Australia. The models developed in this study introduced an attention mechanism after feature extraction of water quality data in the section of Burnett River considering the effect of the sequences on the prediction results at different moments to enhance the influence of key features on the prediction results. This study provides one-step-ahead forecasting and multistep forward forecasting of dissolved oxygen (DO) of the Burnett River utilizing LSTM and AT-LSTM models and the comparison of the results. The research outcomes demonstrated that the inclusion of the attention mechanism improves the prediction performance of the LSTM model. Therefore, the AT-LSTM-based water quality forecasting model, developed in this study, demonstrated its stronger capability than the LSTM model for informing the Water Quality Improvement Plan of Queensland, Australia, to accurately predict water quality in the Burnett River.

Keywords: water quality prediction; time series; attention mechanism; long short-term memory (LSTM) (search for similar items in EconPapers)
JEL-codes: O13 Q Q0 Q2 Q3 Q5 Q56 (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2071-1050/14/20/13231/pdf (application/pdf)
https://www.mdpi.com/2071-1050/14/20/13231/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jsusta:v:14:y:2022:i:20:p:13231-:d:942635

Access Statistics for this article

Sustainability is currently edited by Ms. Alexandra Wu

More articles in Sustainability from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().