An Improved Attention-based LSTM for Multi-Step Dissolved Oxygen Prediction in Water Environment

Jing Bi, Yongze Lin, Quanxi Dong, Haitao Yuan, Meng Chu Zhou

Research output: Chapter in Book/Report/Conference proceedingConference contribution

18 Scopus citations

Abstract

The prediction of accurate water quality has great significance to the sustainable management of water resources and pollution prevention. Due to the complexity of water environment, it is difficult to do so. Traditional prediction methods are mainly linear methods. Their prediction accuracy is limited since they fail to reflect nonlinear characteristics in water quality data. To achieve much higher accuracy, this work proposes to combines a Savitzky-Golay filter with Attention-based Long Short-Term Memory to perform a multi-step prediction of water quality. The proposed model uses a Savitzky-Golay filter for smoothing sequences to reduce noise interference. The adoption of an attention mechanism can extract effective information from complex, long, and temporal dependence. Experimental results demonstrate that the proposed method outperforms other state-of-the-art peers.

Original languageEnglish (US)
Title of host publication2020 IEEE International Conference on Networking, Sensing and Control, ICNSC 2020
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9781728168531
DOIs
StatePublished - Oct 30 2020
Event2020 IEEE International Conference on Networking, Sensing and Control, ICNSC 2020 - Nanjing, China
Duration: Oct 30 2020Nov 2 2020

Publication series

Name2020 IEEE International Conference on Networking, Sensing and Control, ICNSC 2020

Conference

Conference2020 IEEE International Conference on Networking, Sensing and Control, ICNSC 2020
Country/TerritoryChina
CityNanjing
Period10/30/2011/2/20

All Science Journal Classification (ASJC) codes

  • Artificial Intelligence
  • Computer Networks and Communications
  • Statistics, Probability and Uncertainty
  • Control and Optimization
  • Sensory Systems

Keywords

  • Attention
  • Long Short-Term Memory (LSTM)
  • Savitzky-Golay filter
  • dissolved oxygen prediction
  • encoder-decoder architecture
  • water environment

Fingerprint

Dive into the research topics of 'An Improved Attention-based LSTM for Multi-Step Dissolved Oxygen Prediction in Water Environment'. Together they form a unique fingerprint.

Cite this