# Improved Prediction of Hourly PM2.5 Concentrations with a Long Short-Term Memory Optimized by Stacking Ensemble Learning and Ant Colony Optimization

## TL;DR

This paper introduces a new model for predicting PM2.5 concentrations using optimized machine learning techniques to improve accuracy and efficiency.

## Contribution

The novel contribution is a stacking-ACO-LSTM model that optimizes PM2.5 prediction with improved accuracy and reduced computational time.

## Key findings

- The stacking-ACO-LSTM model achieved a 99.88% decrease in mean square error compared to traditional models.
- The model's coefficient of determination increased by 2.39%, showing improved prediction accuracy.
- The model efficiently filters high-weight features, reducing predictive complexity.

## Abstract

To address the performance degradation in existing PM2.5 prediction models caused by excessive complexity, poor spatiotemporal efficiency, and suboptimal parameter optimization, we employ stacking ensemble learning for feature weighting analysis and integrate the ant colony optimization (ACO) algorithm for model parameter optimization. Combining meteorological and collaborative pollutant data, a model (namely the stacking-ACO-LSTM model) with a much shorter consuming time than that of only long short-term memory (LSTM) networks suitable for PM2.5 concentration prediction is established. It can effectively filter out feature variables with higher weights, thereby reducing the predictive power of the model. The prediction of hourly PM2.5 concentration of the model is trained and tested using real-time monitoring data in Nanchang City from 2017 to 2019. The results show that the established stacking-ACO-LSTM model has high accuracy in predicting PM2.5 concentration, and compared to the same model without considering time and space efficiency and defective parameter optimization, the mean square error (MSE) decreases by about 99.88%, and the coefficient of determination (R2) increases by about 2.39%. This study provides a new idea for predicting PM2.5 concentration in cities.