Incorporating county-level NO2 data into a metapopulation model: a case study of the first COVID-19 wave in Oklahoma

Nitrogen dioxide (NO2) is a widespread air pollutant associated with the risk and prevalence of various respiratory infectious diseases. Epidemiological studies have reported associations between NO2 exposure and COVID-19 incidence, and biological evidence indicates that NO2 exposure increases vulnerability to infection. Motivated by these findings and by the scarcity of mechanistic epidemic models that incorporate air pollutant data, here we introduce an NO2- and mobility-informed metapopulation model and study the first COVID-19 wave in Oklahoma. County-level NO2 concentrations enter the infection rates through a first-order parameterization, alongside high-resolution mobility patterns capturing heterogeneous inter-county connectivity. Using numerical simulations, we demonstrate that higher environmental sensitivity produces earlier and more synchronized outbreaks across counties, while lower sensitivity yields delayed, asynchronous trajectories. When fitted to county-level COVID-19 incidence data, our model outperformed the classic SIR model (fitted independently in each county) in over 75% of counties in Oklahoma across all standard error metrics. The improvements were even more pronounced in forecasting tasks across epidemic stages, with county-level win rates against the SIR model of 59.74% (surge), 87.01% (peak), 80.52% (post-peak), and 88.31% (decline) by nRMSE. To isolate the contribution of NO2, we compared our model against a mobility-only model with NO2 removed and homogeneous infection rates across counties. The NO2 term improved county-level forecasts during the surge (94.81% of counties by nRMSE), post-peak (88.31%), and peak (62.34%) stages, though not during the early and decline stages, suggesting that NO2 information can generate better forecasts than mobility alone during periods of high incidence. Our results demonstrate that environmental exposure can enter the transmission structure of a mechanistic epidemic model directly and that doing so yields measurable forecasting gains.