用扩散模型预测流感时空传播,生成多样且准确的疫情轨迹。
Generative diffusion models for spatiotemporal influenza forecasting

- 将流感季节转为时空图像,用去噪扩散模型学习疫情分布
- 在2023–2025年美国疾控中心挑战赛中表现优于多数方法
- 混合30%真实数据与70%模拟数据训练效果最佳
传染病发病率预测对公共卫生规划至关重要,但因疫情动态复杂而困难。现有机制与统计方法常难以捕捉多模态不确定性或新兴趋势。Influpaint 将去噪扩散概率模型应用于疫情预测,将流感季节编码为像素强度代表发病率的时空图像,从监测数据与模拟轨迹的混合数据集中学习疾病动态的丰富分布。预测被建模为从部分观测中进行条件生成(图像修复)任务。实验表明,Influpaint 能生成逼真且多样的疫情轨迹,在回溯评估中达到领先集合方法的预测精度。在2023–2025年美国疾控中心流感展望挑战赛的实时评估中,各季表现显著提升,2024–2025年预测高度准确但略显过度自信。最佳性能来自包含30%监测数据与70%模拟轨迹的训练集。结果表明,扩散模型可有效捕捉流感动态中的关键时空结构,并为传染病的概率化预测提供灵活框架。
原文摘要 · Abstract (English)
Forecasting infectious disease incidence can provide important information to guide public health planning, yet 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 2023--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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。