Reproduce “DeepBlip: Estimating Conditional Average Treatment Effects Over Time” on OpenResearch

Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.

Universal app for Apple silicon and Intel. Requires macOS 11 or later.

About this paper

Haorui Ma, Dennis Frauen, Stefan Feuerriegel

DeepBlip is a neural framework that estimates Conditional Average Treatment Effects (CATE) over time by integrating Structural Nested Mean Models (SNMMs) with deep learning. It introduces a novel double optimization trick, enabling end-to-end, gradient-based training to enhance robustness against time-varying confounding and improve stability over long prediction horizons.

Key insights

  • The double optimization trick effectively converts an intractable sequential optimization problem into a single, gradient-based training objective, overcoming a major bottleneck for neural SNMMs.
  • The L2-moment loss function used in DeepBlip is Neyman-orthogonal, which ensures robustness of the blip coefficient estimates even if nuisance models are moderately misspecified.
  • By decomposing total treatment effects into localized blip effects, DeepBlip offers enhanced interpretability for clinicians and enables highly efficient offline evaluation of optimal personalized treatment sequences.

Summary by alphaXiv.

Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.