ArXiv TLDR

Observable Neural ODEs for Identifiable Causal Forecasting in Continuous Time

🐦 Tweet
2604.26070

Jennifer Wendland, Nicolas Freitag, Maik Kschischo

cs.LGmath.OCmath.STq-bio.QM

TLDR

ObsNODEs enable identifiable causal forecasting in continuous-time settings by ensuring latent state observability, even with hidden confounders.

Key contributions

  • Links control-theoretic observability to causal identifiability in continuous-time latent state-space models.
  • Derives a continuous-time adjustment formula for potential outcomes under dynamic treatment paths.
  • Proposes Observable Neural ODEs (ObsNODEs) for causal forecasting with reconstructible latent states.
  • Achieves strong performance on synthetic cancer, MIMIC-IV, and real-world sepsis datasets.

Why it matters

Causal forecasting in continuous time is crucial for sequential decision-making, but hidden confounders pose a major challenge. This paper provides a theoretical foundation and a practical model (ObsNODEs) to overcome this, enabling reliable predictions under interventions. This is vital for fields like healthcare.

Original Abstract

Causal inference in continuous-time sequential decision problems is challenged by hidden confounders. We show that, in latent state-space models with time-varying interventions, observability of the latent dynamics from observed data is necessary for identifying dynamic treatment effects, linking control-theoretic observability to causal identifiability, even when hidden confounders affect both treatments and outcomes. We derive a continuous-time adjustment formula expressing potential outcome distributions under treatment trajectories via the measurement model, latent dynamics, and the filtering distribution over latent states given observed histories. We propose Observable Neural ODEs (ObsNODEs), Neural ODE models in observable normal form for causal forecasting. ObsNODEs learn continuous-time dynamics with states reconstructible from observations, enabling outcome prediction under alternative treatment paths. Experiments on synthetic cancer data, semi-synthetic data based on MIMIC-IV, and real-world sepsis data show strong performance over recent sequence models.

📬 Weekly AI Paper Digest

Get the top 10 AI/ML arXiv papers from the week — summarized, scored, and delivered to your inbox every Monday.