mlsanityData to 7 Oct 2026

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

arXiv 2606.05692 ↗First posted 4 Jun 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

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. To address this gap, we develop a large-scale benchmark for counterfactual prediction in epidemic time series under dynamic interventions. Unlike existing benchmarks, it supports static and time-varying treatments, as well as both single-policy and multi-policy intervention settings, enabling evaluation of causal inference methods across a broad range of causal inference scenarios. Leveraging a calibrated agent-based model grounded in real-world demographic, mobility, epidemiological, and policy data, we generate realistic counterfactual trajectories across more than 150 U.S. counties. Using this benchmark, we evaluate widely used and state-of-the-art causal inference methods, revealing substantial performance differences and highlighting the challenges of realistic time-series causal reasoning.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 12 of its 50 references that are in the map. First posted on arXiv 4 Jun 2026.

Builds on

12 of 50 references.

References · 12

  1. 2017Attention Is All You Needcited by 59,932
  2. 2013Auto-Encoding Variational Bayescited by 3,980
  3. 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 475
  4. 2018Fundamentals of Recurrent Neural Network (RNN) and Long Short-Term Memory (LSTM) Networkcited by 415
  5. 2020Estimating Counterfactual Treatment Outcomes over Time Through Adversarially Balanced Representationscited by 154
  6. 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
  7. 2021Causal Inference for Time series Analysis: Problems, Methods and Evaluationcited by 42
  8. 2022Estimating average causal effects from patient trajectoriescited by 29
  9. 2022Normalizing Flows for Interventional Density Estimationcited by 21
  10. 2023Counterfactual Generative Models for Time-Varying Treatmentscited by 17
  11. 2025TimeGraph: Synthetic Benchmark Datasets for Robust Time-Series Causal Discoverycited by 9
  12. 2025Counterfactual Probabilistic Diffusion with Expert Modelscited by 4

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2024DiffusionCounterfactuals: Inferring High-dimensional Counterfactuals with Guidance of Causal Representations
  2. 2025Causal Time Series Generation via Diffusion Models
  3. 2026Toward World Models for Epidemiology
  4. 2026Beyond Time Series: Spatial Reasoning for Epidemic Forecasting via Multimodal Learning
  5. 2026Causal Diffusion Models for Counterfactual Outcome Distributions in Longitudinal Data
  6. 2026Causal State-Space Model for Causal Inference: Estimating Longitudinal Individual Treatment Effects
  7. 2026Are Good Generators Good Decision-Makers? Policy Learning for General Interventions via Retargeted Counterfactual Generation
  8. 2023Modeling Causal Mechanisms with Diffusion Models for Interventional and Counterfactual Queries