Learned geospatial priors for malaria forecasting

Nazir U., Sina R., Arundhati W., Lais S., Khalid S.

Malaria remains one of the leading communicable causes of death globally, with approximately half the world’s population at risk, predominantly across African and South Asian countries. Accurate, region-specific outbreak prediction is hindered by strong spatio-temporal heterogeneity in environmental, climatological, and sociodemographic risk factors, as well as by the over-dispersed, count-valued nature of incidence data that violates assumptions of standard regression and many off-the-shelf machine learning approaches. We address these challenges through a unified framework combining count-aware statistical modelling, sequence learning, and geospatial foundation model representations, evaluated across India and Nigeria using monthly data from 2000 to 2024 integrating rainfall, temperature, vegetation index, nighttime lights, population counts, distance to water bodies, and historical malaria case counts. First, we model malaria incidence using lagstructured negative binomial regression, explicitly accounting for over-dispersion while capturing biologically grounded temporal effects; this yields interpretable estimates of how environmental and sociodemographic covariates drive transmission. Second, we develop an ensemble of LSTM and Transformer models that captures complementary temporal dependencies in incidence dynamics. Third, we augment both approaches with embeddings from the AlphaEarth foundation model, which encode high-dimensional satellite-derived environmental context. Incorporating these embeddings consistently improves predictive performance, with the ensemble model increasing R2 from 0.63 to 0.78 in Nigeria and from 0.87 to 0.88 in India, while remaining stable across random seeds. Together, these results suggest that foundation model embeddings convert previously unexplained variance into structured environmental signal, reducing effective over-dispersion and providing a robust and transferable framework for forecasting climate-sensitive infectious diseases.

Type

Conference paper

Publisher

NeurIPS

Publication Date

2026-06-18T00:00:00+00:00

Permalink More information Close