Hyperparameter-Optimization-Inspired Long Short-Term Memory Network for Air Quality Grade Prediction

Dushi Wen, Sirui Zheng, Jiazhen Chen, Zhouyi Zheng, Chen Ding, Lei Zhang

Research output: Contribution to journalArticlepeer-review

10 Scopus citations

Abstract

In the world, with the continuous development of modern society and the acceleration of urbanization, the problem of air pollution is becoming increasingly salient. Methods for predicting the air quality grade and determining the necessary governance are at present most urgent problems waiting to be solved by human beings. In recent years, more and more machine-learning-based methods have been used to solve the air quality prediction problem. However, the uncertainty of environmental changes and the difficulty of precisely predicting quantitative values seriously influence prediction results. In this paper, the proposed air pollutant quality grade prediction method based on a hyperparameter-optimization-inspired long short-term memory (LSTM) network provides two advantages. Firstly, the definition of air quality grade is introduced in the air quality prediction task, which turns a fitting problem into a classification problem and makes the complex problem simple; secondly, the hunter–prey optimization algorithm is used to optimize the hyperparameters of the LSTM structure to obtain the optimal network structure adaptively determined through the use of input data, which can include more generalization abilities. The experimental results from three real Xi’an air quality datasets display the effectiveness of the proposed method.

Original languageEnglish
Article number243
JournalInformation (Switzerland)
Volume14
Issue number4
DOIs
StatePublished - Apr 2023

Keywords

  • artificial intelligence
  • hunter–prey optimization
  • hyperparameter
  • LSTM
  • network structure

Fingerprint

Dive into the research topics of 'Hyperparameter-Optimization-Inspired Long Short-Term Memory Network for Air Quality Grade Prediction'. Together they form a unique fingerprint.

Cite this