Efficient inference of large-scale air quality using a lightweight ensemble predictor

Abstract

Accurate and efficient air quality prediction is crucial for public health protection and environmental sustainability. While numerous grid-based and graph-based prediction models have been developed, they encounter challenges in large-scale scenarios: (1) Grid-based models, though computationally efficient, have limited prediction accuracy in large-scale sparse scenarios; (2) Graph-based models, despite higher prediction accuracy, suffer from significant computational inefficiencies when dealing with a large number of sensors, i.e. graph nodes. To address these issues, we propose a Lightweight Ensemble Predictor (LiEnPred) for efficient air quality prediction in large-scale sparse scenarios. First, we present a data structure transformation algorithm that converts sparse monitoring sensors from graph structures to compact grid structures, preserving the connections between graph nodes. Next, we present a lightweight parameter-shared spatio-temporal dilation convolution network that efficiently captures spatio-temporal dependencies in air quality data without significantly increasing computation time or parameter scale. In our experiments, we collected air quality data from over 2000 sensors across China over the past three years and evaluated LiEnPred’s prediction performance in large-scale scenarios using PM2.5 and NO2 concentration data. The experimental results demonstrate that the proposed LiEnPred model matches or exceeds the predictive accuracy of eight baselines with faster time efficiency and fewer model parameters.