arXiv Machine Learning

Deep Generative Spatiotemporal Engression for Probabilistic Forecasting of Epidemics

arXiv:2603. 07108v2 Announce Type: replace-cross Abstract: Accurate and reliable forecasting of epidemic incidences is critical for public health preparedness, yet it remains a challenging task due to complex nonlinear temporal dependencies and heterogeneous spatial interactions.

arXiv Machine Learning
Sep 15

Generative diffusion models for spatiotemporal influenza forecasting

The paper introduces Influpaint, a denoising diffusion probabilistic model adapted for forecasting influenza incidence. By representing influenza seasons as spatiotemporal images and training on a hybrid dataset of surveillance and simulated trajectories, the model learns a rich distribution of disease dynamics and performs forecasting as a conditional generation task. In retrospective and real‑time evaluations, Influpaint produces realistic, diverse epidemic trajectories and achieves forecast accuracy competitive with leading ensemble methods, especially when trained with 30% surveillance and 70% simulated data.

By Joseph Lemaitre, Justin Lessler
arXiv Machine Learning
Aug 28

SimCast-S2S: An Efficient Generative Model for Subseasonal Precipitation Forecasting via Transfer Learning from Climate Simulations

SimCast‑S2S is a generative latent‑diffusion model designed for probabilistic subseasonal‑to‑seasonal precipitation forecasting. It tackles three key challenges: it uses a diffusion pipeline to capture uncertainty, operates in a compact latent space to enable efficient large‑ensemble generation, and leverages transfer learning with low‑rank adaptation to train on limited reanalysis data after pretraining on climate simulations. The model outperforms deep‑learning baselines and competes with, or surpasses, operational systems such as the ECMWF‑S2S baseline without requiring extensive post‑processing.

By Hiep V. Dang, Antonios Mamalakis
arXiv Machine Learning
Sep 4

SimCast-S2S: A Computationally Efficient Diffusion Model for Subseasonal Precipitation Forecasting

SimCast‑S2S is a generative latent‑diffusion framework designed for probabilistic subseasonal‑to‑seasonal precipitation forecasting. It tackles three key challenges: it uses a diffusion‑based generative pipeline for uncertainty quantification, operates in a compact latent space learned by VAEs for efficient large‑ensemble generation, and employs transfer learning with LoRA to overcome limited training data. On reanalysis data, it outperforms deep‑learning baselines and competes with or surpasses state‑of‑the‑art operational systems such as ECMWF‑S2S.

By Hiep V. Dang, Antonios Mamalakis
Hugging Face Trending Papers
Jun 4

Benchmarking Counterfactual Prediction in Epidemic Time Series with Time-Varying Interventions

Deep learning has enabled significant advances in time-series causal inference, yet progress remains constrained by the lack of realistic benchmarks with observable counterfactual outcomes. Existing datasets either rely on real-world observations without ground-truth counterfactuals or on simplified simulations that fail to capture complex causal dynamics.

arXiv AI
Sep 17

TERN: A Delta-rule Memory with a Seasonal Reference and Online Adaptation for Epidemic Forecasting

The paper introduces TERN, a forecasting model that uses a delta‑rule fast‑weight memory with channel‑wise decay and learned erasure gates, combined with a seasonal reference and online adaptation. It addresses challenges in influenza forecasting such as limited seasonal data, shifting wave patterns, and misleading information after peaks. On three Cola‑GNN influenza benchmarks, TERN outperformed existing epidemic graph models and general forecasters, matching or exceeding seasonal references and demonstrating the value of its memory component.

By Shunya Nagashima, Yuta Funayama