Air pollution forecasting based on attention-based LSTM neural network and ensemble learning

Duen-Ren Liu*, Hero SJ Lee, Yang Huang, Chien Ju Chiu


研究成果: Article同行評審

61 引文 斯高帕斯(Scopus)


With air pollution having become a global concern, scientists are committed to working on its amelioration. In the field of air pollution prediction, there have been good results in experimental research so far, but few studies have integrated weather forecast information and the properties of air pollution drift. In this work, we propose a novel wind-sensitive attention mechanism with a long short-term memory (LSTM) neural network model to predict the air pollution - PM2.5 concentrations by considering the influence of wind direction and speed on the changes of spatial–temporal PM2.5 concentrations in neighbouring areas. Preliminary predictions for PM2.5 are then made by an LSTM neural network regarding neighbouring pollution; these predictions are “paid attention to” and we finally apply an ensemble learning method based on eXtreme Gradient Boosting (XGBoost) to combine the preliminary predictions with weather forecasting to make second phase predictions of PM2.5. The experiment is conducted using PM2.5 data and weather forecast data. Our results illustrate that the proposed method is superior to other methods in predicting PM2.5 concentrations, including multi-layer perceptron, support vector regression, LSTM neural network, and extreme gradient boosting algorithm.

頁(從 - 到)1-16
期刊Expert Systems
出版狀態Published - 16 6月 2020


深入研究「Air pollution forecasting based on attention-based LSTM neural network and ensemble learning」主題。共同形成了獨特的指紋。