Generative diffusion models for spatiotemporal influenza forecasting
- 1Department of Epidemiology, Gillings School of Global Public Health, University of North Carolina at Chapel Hill, Chapel Hill, North Carolina, United States of America
- 2Department of Epidemiology, Johns Hopkins Bloomberg School of Public Health, Baltimore, Maryland, United States of America
Abstract
Forecasting infectious disease incidence can provide important information to guide public health planning, yet it is difficult because epidemic dynamics are complex. Current mechanistic and statistical approaches often struggle to capture multimodal uncertainty or emergent trends. Influpaint adapts denoising diffusion probabilistic models to epidemic forecasting. By encoding influenza seasons as spatiotemporal images in which pixel intensity represents incidence, Influpaint learns a rich distribution of disease dynamics from a hybrid dataset of surveillance and simulated trajectories. Forecasting is formulated as a conditional generation (inpainting) task from partial observations. We show that Influpaint generates realistic, diverse epidemic trajectories and achieves forecast accuracy that is competitive with leading ensemble methods in retrospective evaluation. In real-time evaluation during the 2022–2025 U.S. CDC FluSight challenges, performance improved substantially across seasons, with highly accurate but somewhat overconfident projections in 2024–2025. The best performance was achieved with a training dataset containing 30% surveillance and 70% simulated trajectories. These results show that diffusion models can capture important spatiotemporal structure in influenza dynamics and provide a flexible framework for probabilistic infectious disease forecasting. Author summary: Accurately forecasting influenza activity is difficult, which complicates public health preparedness and response. We propose an influenza forecasting approach, named Influpaint, that treats an influenza season like an image, with weeks on one axis, locations on the other, and color intensity representing reported hospitalizations. This framing lets us use a type of generative artificial intelligence, the diffusion model, to learn the patterns of influenza activity and then predict (inpaint) future weeks from the part of the season already observed. We show that Influpaint generates plausible influenza seasons and produces forecasts that are competitive with established ensemble methods. The approach is also highly flexible and can be used to impute missing data without additional training. We also find that combining real surveillance data with simulated epidemic trajectories improves performance.
Related
- A Critical Audit of Spatiotemporal Forecasting Benchmark Datasets and Models
- Event-Aware Spatiotemporal Precipitation Forecasting with Geographic Context and Physics-Guided Regularization
- Forecasting seasonal influenza epidemics with physics-informed neural networks
- Beyond Point Forecasts: A Survey on Probabilistic Forecasting for Time Series and Spatiotemporal Data
- OmniST: Topology-Invariant Spatiotemporal Forecasting with Guaranteed-Coverage Uncertainty at Unseen Sensor Locations
BibTeX
@article{lemaitre2026generative,
title = {{Generative diffusion models for spatiotemporal influenza forecasting}},
author = {Joseph Lemaitre and Justin Lessler},
journal = {PLOS Computational Biology},
year = {2026},
month = sep,
doi = {10.1371/journal.pcbi.1014846},
url = {https://journals.plos.org/ploscompbiol/article?id=10.1371/journal.pcbi.1014846},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - Generative diffusion models for spatiotemporal influenza forecasting
AU - Joseph Lemaitre
AU - Justin Lessler
PY - 2026
DA - 2026/09/30
JO - PLOS Computational Biology
DO - 10.1371/journal.pcbi.1014846
UR - https://journals.plos.org/ploscompbiol/article?id=10.1371/journal.pcbi.1014846
AB - Forecasting infectious disease incidence can provide important information to guide public health planning, yet it is difficult because epidemic dynamics are complex. Current mechanistic and statistical approaches often struggle to capture multimodal uncertainty or emergent trends. Influpaint adapts denoising diffusion probabilistic models to epidemic forecasting. By encoding influenza seasons as spatiotemporal images in which pixel intensity represents incidence, Influpaint learns a rich distribution of disease dynamics from a hybrid dataset of surveillance and simulated trajectories. Forecasting is formulated as a conditional generation (inpainting) task from partial observations. We show that Influpaint generates realistic, diverse epidemic trajectories and achieves forecast accuracy that is competitive with leading ensemble methods in retrospective evaluation. In real-time evaluation during the 2022–2025 U.S. CDC FluSight challenges, performance improved substantially across seasons, with highly accurate but somewhat overconfident projections in 2024–2025. The best performance was achieved with a training dataset containing 30% surveillance and 70% simulated trajectories. These results show that diffusion models can capture important spatiotemporal structure in influenza dynamics and provide a flexible framework for probabilistic infectious disease forecasting. Author summary: Accurately forecasting influenza activity is difficult, which complicates public health preparedness and response. We propose an influenza forecasting approach, named Influpaint, that treats an influenza season like an image, with weeks on one axis, locations on the other, and color intensity representing reported hospitalizations. This framing lets us use a type of generative artificial intelligence, the diffusion model, to learn the patterns of influenza activity and then predict (inpaint) future weeks from the part of the season already observed. We show that Influpaint generates plausible influenza seasons and produces forecasts that are competitive with established ensemble methods. The approach is also highly flexible and can be used to impute missing data without additional training. We also find that combining real surveillance data with simulated epidemic trajectories improves performance.
SN - 1553-7358
ER -