Counterfactual Probabilistic Diffusion with Expert Models

Wenhao Mu, Zhi Cao, Mehmed Uludag, Alexander Rodríguez
8/18/2025
cs.LGcs.AIstat.ME

No confident code implementation found for this paper. 2 possible matches can be shown below.

Abstract

Predicting counterfactual distributions in complex dynamical systems is essential for scientific modeling and decision-making in domains such as public health and medicine. However, existing methods often rely on point estimates or purely data-driven models, which tend to falter under data scarcity. We propose a time series diffusion-based framework that incorporates guidance from imperfect expert models by extracting high-level signals to serve as structured priors for generative modeling. Our method, ODE-Diff, bridges mechanistic and data-driven approaches, enabling more reliable and interpretable causal inference. We evaluate ODE-Diff across semi-synthetic COVID-19 simulations, synthetic pharmacological dynamics, and real-world case studies, demonstrating that it consistently outperforms strong baselines in both point prediction and distributional accuracy.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. 2 weaker matches are hidden by default — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{mu2025counterfactual,
  title  = {Counterfactual Probabilistic Diffusion with Expert Models},
  author = {Wenhao Mu and Zhi Cao and Mehmed Uludag and Alexander Rodríguez},
  year   = {2025},
  eprint = {2508.13355},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2508.13355v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2508.13355.svg)](https://codeofpaper.com/paper/2508.13355)

Discussion