Beyond the Hype: Comparing Lightweight and Deep Learning Models for Air Quality Forecasting
Summary
This study compares lightweight additive models (Facebook Prophet, NeuralProphet) against deep learning, machine learning, and traditional statistical models for urban air quality forecasting (PM$_{2.5}$, PM$_{10}$) in Beijing. It found that Facebook Prophet consistently outperformed all other models, achieving high accuracy ($R^2 > 0.94$), demonstrating that interpretable additive models can be highly competitive for public health-critical predictions.
Medical Relevance
Accurate and timely forecasting of air pollutants like PM$_{2.5}$ and PM$_{10}$ is critical for public health, as it enables informed decision-making regarding preventative measures, health advisories for vulnerable populations, and the implementation of effective mitigation policies to reduce exposure and associated health risks.
AI Health Application
AI models (Lightweight additive models like Facebook Prophet, NeuralProphet, as well as Deep Learning, LSTM, and LightGBM) are applied to forecast air quality parameters (PM2.5, PM10). This forecasting supports public health by enabling timely warnings, informing health advisories, and guiding policy decisions to mitigate health risks associated with air pollution.
Key Points
- Addressed the need for accurate, interpretable, and operationally viable urban air pollution (PM$_{2.5}$, PM$_{10}$) forecasting for public health and policy guidance.
- Compared two lightweight additive models (Facebook Prophet - FBP, NeuralProphet - NP) against deep learning (LSTM), machine learning (LightGBM), and traditional statistical (SARIMAX) baselines.
- Utilized multi-year pollutant and meteorological data from Beijing, employing systematic feature selection (correlation, mutual information, mRMR) and leakage-safe scaling.
- Chronological data splits and a 7-day holdout period were used for model training and evaluation, employing MAE, RMSE, and $R^2$ as performance metrics.
- Facebook Prophet (FBP) consistently achieved superior performance, outperforming NP, SARIMAX, LSTM, and LightGBM across all evaluated metrics.
- FBP demonstrated high accuracy for both pollutants, with test $R^2$ values exceeding 0.94 for both PM$_{2.5}$ and PM$_{10}$ forecasts.
- The study concludes that interpretable additive models like FBP offer a practical balance of accuracy, transparency, and ease of deployment, challenging the dominance of more complex deep learning approaches.
Methodology
The study utilized multi-year pollutant (PM$_{2.5}$, PM$_{10}$) and meteorological data from Beijing, China. It employed systematic feature selection methods (correlation, mutual information, mRMR) and leakage-safe scaling for data preprocessing. Data was split chronologically for training and testing, with a 7-day holdout period used for final performance evaluation. The models investigated included Facebook Prophet (FBP) and NeuralProphet (NP), with NP additionally leveraging lagged dependencies. For comparison, a Long Short-Term Memory (LSTM) network, LightGBM, and SARIMAX were implemented as baselines. Model performance was assessed using Mean Absolute Error (MAE), Root Mean Square Error (RMSE), and R-squared ($R^2$).
Key Findings
Facebook Prophet (FBP) consistently delivered the best performance among all models tested, outperforming NeuralProphet (NP), SARIMAX, LSTM, and LightGBM. FBP achieved high accuracy for both pollutants, with test $R^2$ values above 0.94 for both PM$_{2.5}$ and PM$_{10}$ forecasting. These results demonstrate that lightweight, interpretable additive models can be highly competitive with, and even surpass, complex deep learning and machine learning approaches for critical air quality prediction tasks.
Clinical Impact
The development of accurate, transparent, and easily deployable air quality forecasting models, such as Facebook Prophet, has significant clinical and public health impact. It enables health authorities to issue more timely and precise public health advisories, allowing vulnerable populations (e.g., individuals with asthma, COPD, cardiovascular conditions, children, elderly) to take proactive preventative measures, thereby reducing their exposure to harmful pollutants and mitigating acute health impacts. Furthermore, these reliable forecasts provide actionable data for urban planners and policymakers to design and implement effective long-term strategies for air quality improvement and public health protection.
Limitations
The abstract does not explicitly state limitations of this specific study's methodology. It highlights the general complexity and limited interpretability of Deep Learning (DL) and hybrid pipelines as a challenge that this research aims to address.
Future Directions
Not explicitly mentioned in the abstract.
Medical Domains
Keywords
Abstract
Accurate forecasting of urban air pollution is essential for protecting public health and guiding mitigation policies. While Deep Learning (DL) and hybrid pipelines dominate recent research, their complexity and limited interpretability hinder operational use. This study investigates whether lightweight additive models -- Facebook Prophet (FBP) and NeuralProphet (NP) -- can deliver competitive forecasts for particulate matter (PM$_{2.5}$, PM$_{10}$) in Beijing, China. Using multi-year pollutant and meteorological data, we applied systematic feature selection (correlation, mutual information, mRMR), leakage-safe scaling, and chronological data splits. Both models were trained with pollutant and precursor regressors, with NP additionally leveraging lagged dependencies. For context, two machine learning baselines (LSTM, LightGBM) and one traditional statistical model (SARIMAX) were also implemented. Performance was evaluated on a 7-day holdout using MAE, RMSE, and $R^2$. Results show that FBP consistently outperformed NP, SARIMAX, and the learning-based baselines, achieving test $R^2$ above 0.94 for both pollutants. These findings demonstrate that interpretable additive models remain competitive with both traditional and complex approaches, offering a practical balance of accuracy, transparency, and ease of deployment.