A novel Encoder-Decoder model based on read-first LSTM for air pollutant prediction

Bo Zhang, Guojian Zou, Dongming Qin, Yunjie Lu, Yupeng Jin, Hui Wang*

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

70 Citations (Scopus)

Abstract

Accurate air pollutant prediction allows effective environment management to reduce the impact of pollution and prevent pollution incidents. Existing studies of air pollutant prediction are mostly interdisciplinary involving environmental science and computer science where the problem is formulated as time series prediction. A prevalent recent approach to time series prediction is the Encoder-Decoder model, which is based on recurrent neural networks (RNN) such as long short-term memory (LSTM), and great potential has been demonstrated. An LSTM network relies on various gate units, but in most existing studies the correlation between gate units is ignored. This correlation is important for establishing the relationship of the random variables in a time series as the stronger is this correlation, the stronger is the relationship between the random variables. In this paper we propose an improved LSTM, named Read-first LSTM or RLSTM for short, which is a more powerful temporal feature extractor than RNN, LSTM and Gated Recurrent Unit (GRU). RLSTM has some useful properties: (1) enables better store and remember capabilities in longer time series and (2) overcomes the problem of dependency between gate units. Since RLSTM is good at long term feature extraction, it is expected to perform well in time series prediction. Therefore, we use RLSTM as the Encoder and LSTM as the Decoder to build an Encoder-Decoder model (EDSModel) for pollutant prediction in this paper. Our experimental results show, for 1 to 24 h prediction, the proposed prediction model performed well with a root mean square error of 30.218. The effectiveness and superiority of RLSTM and the prediction model have been demonstrated.

Original languageEnglish
Article number144507
JournalScience of the Total Environment
Volume765
Early online date05 Jan 2021
DOIs
Publication statusPublished - 15 Apr 2021
Externally publishedYes

Bibliographical note

Funding Information:
This work is funded by National Natural Science Foundation of China ( 61572326 , 61802258 , 61702333 ), Natural Science Foundation of Shanghai ( 18ZR1428300 ), the Shanghai Committee of Science and Technology ( 17070502800 ).

Publisher Copyright:
© 2020 Elsevier B.V.

Keywords

  • Air pollutant prediction
  • Deep learning
  • Encoder-Decoder model
  • Long short term memory
  • Numerical analysis
  • Recurrent neural networks

ASJC Scopus subject areas

  • Environmental Engineering
  • Environmental Chemistry
  • Waste Management and Disposal
  • Pollution

Fingerprint

Dive into the research topics of 'A novel Encoder-Decoder model based on read-first LSTM for air pollutant prediction'. Together they form a unique fingerprint.

Cite this