# Estimating average causal effects from patient trajectories

arXiv: https://arxiv.org/abs/2203.01228 · HTML: https://mlsanity.com/paper/2203.01228/

## Key facts

- Cited by 29 papers in the map (data through 14 Oct 2026).
- Builds on 10 of its 50 references that are in the map.
- First posted on arXiv 2 Mar 2022.

## Builds on (most cited first, 10 of 10)

- [Adam: A Method for Stochastic Optimization](https://mlsanity.com/paper/1412.6980/) (2014)
- [Estimation and Inference of Heterogeneous Treatment Effects using Random Forests](https://mlsanity.com/paper/1510.04342/) (2015)
- [Adapting Neural Networks for the Estimation of Treatment Effects](https://mlsanity.com/paper/1906.02120/) (2019)
- [A Theoretically Grounded Application of Dropout in Recurrent Neural Networks](https://mlsanity.com/paper/1512.05287/) (2015)
- [Estimating Counterfactual Treatment Outcomes over Time Through Adversarially Balanced Representations](https://mlsanity.com/paper/2002.04083/) (2020)
- [Orthogonal Statistical Learning](https://mlsanity.com/paper/1901.09036/) (2019)
- [MIMIC-Extract: A Data Extraction, Preprocessing, and Representation Pipeline for MIMIC-III](https://mlsanity.com/paper/1907.08322/) (2019)
- [Time Series Deconfounder: Estimating Treatment Effects over Time in the Presence of Hidden Confounders](https://mlsanity.com/paper/1902.00450/) (2019)
- [Estimating Individual Treatment Effects with Time-Varying Confounders](https://mlsanity.com/paper/2008.13620/) (2020)
- [Disentangled Counterfactual Recurrent Networks for Treatment Effect Inference over Time](https://mlsanity.com/paper/2112.03811/) (2021)

## Built on by (most cited first, 29 of 29)

- [Causal Transformer for Estimating Counterfactual Outcomes](https://mlsanity.com/paper/2204.07258/) (2022)
- [Causal machine learning for predicting treatment outcomes](https://mlsanity.com/paper/2410.08770/) (2024)
- [Bayesian Neural Controlled Differential Equations for Treatment Effect Estimation](https://mlsanity.com/paper/2310.17463/) (2023)
- [Sharp Bounds for Generalized Causal Sensitivity Analysis](https://mlsanity.com/paper/2305.16988/) (2023)
- [Model-agnostic meta-learners for estimating heterogeneous treatment effects over time](https://mlsanity.com/paper/2407.05287/) (2024)
- [Estimating individual treatment effects under unobserved confounding using binary instruments](https://mlsanity.com/paper/2208.08544/) (2022)
- [Counterfactual Generative Models for Time-Varying Treatments](https://mlsanity.com/paper/2305.15742/) (2023)
- [A Neural Framework for Generalized Causal Sensitivity Analysis](https://mlsanity.com/paper/2311.16026/) (2023)
- [Reliable Off-Policy Learning for Dosage Combinations](https://mlsanity.com/paper/2305.19742/) (2023)
- [HealthFlow: A Self-Evolving AI Agent with Meta Planning for Autonomous Healthcare Research](https://mlsanity.com/paper/2508.02621/) (2025)
- [Instrumental Variable Estimation for Causal Inference in Longitudinal Data with Time-Dependent Latent Confounders](https://mlsanity.com/paper/2312.07175/) (2023)
- [Longitudinal Targeted Minimum Loss-based Estimation with Temporal-Difference Heterogeneous Transformer](https://mlsanity.com/paper/2404.04399/) (2024)
- [IGC-Net for conditional average potential outcome estimation over time](https://mlsanity.com/paper/2405.21012/) (2024)
- [G-Transformer: Counterfactual Outcome Prediction under Dynamic and Time-varying Treatment Regimes](https://mlsanity.com/paper/2406.05504/) (2024)
- [Causal Contrastive Learning for Counterfactual Regression Over Time](https://mlsanity.com/paper/2406.00535/) (2024)
- [Stabilized Neural Prediction of Potential Outcomes in Continuous Time](https://mlsanity.com/paper/2410.03514/) (2024)
- [Constructing Confidence Intervals for Average Treatment Effects from Multiple Datasets](https://mlsanity.com/paper/2412.11511/) (2024)
- [Data-Driven Allocation of Preventive Care With Application to Diabetes Mellitus Type II](https://mlsanity.com/paper/2308.06959/) (2023)
- [Consistency of Neural Causal Partial Identification](https://mlsanity.com/paper/2405.15673/) (2024)
- [Overlap-weighted orthogonal meta-learner for treatment effect estimation over time](https://mlsanity.com/paper/2510.19643/) (2025)
- [Identification of Average Causal Effects in Confounded Additive Noise Models](https://mlsanity.com/paper/2407.10014/) (2024)
- [Frugal, Flexible, Faithful: Causal Data Simulation via Frengression](https://mlsanity.com/paper/2508.01018/) (2025)
- [DeepBlip: Estimating Conditional Average Treatment Effects Over Time](https://mlsanity.com/paper/2511.14545/) (2025)
- [Beyond the Known: Decision Making with Counterfactual Reasoning Decision Transformer](https://mlsanity.com/paper/2505.09114/) (2025)
- [Leaning Time-Varying Instruments for Identifying Causal Effects in Time-Series Data](https://mlsanity.com/paper/2411.17774/) (2024)
- [Latent Variable Modeling for Robust Causal Effect Estimation](https://mlsanity.com/paper/2508.20259/) (2025)
- [Deep Doubly Debiased Longitudinal Effect Estimation with ICE G-Computation](https://mlsanity.com/paper/2602.12379/) (2026)
- [Smooth Multi-Policy Causal Effect Estimation in Longitudinal Settings](https://mlsanity.com/paper/2605.14284/) (2026)
- [Benchmarking Counterfactual Prediction in Epidemic Time Series with Time-Varying Interventions](https://mlsanity.com/paper/2606.05692/) (2026)

## Abstract (arXiv)

In medical practice, treatments are selected based on the expected causal effects on patient outcomes. Here, the gold standard for estimating causal effects are randomized controlled trials; however, such trials are costly and sometimes even unethical. Instead, medical practice is increasingly interested in estimating causal effects among patient (sub)groups from electronic health records, that is, observational data. In this paper, we aim at estimating the average causal effect (ACE) from observational data (patient trajectories) that are collected over time. For this, we propose DeepACE: an end-to-end deep learning model. DeepACE leverages the iterative G-computation formula to adjust for the bias induced by time-varying confounders. Moreover, we develop a novel sequential targeting procedure which ensures that DeepACE has favorable theoretical properties, i.e., is doubly robust and asymptotically efficient. To the best of our knowledge, this is the first work that proposes an end-to-end deep learning model tailored for estimating time-varying ACEs. We compare DeepACE in an extensive number of experiments, confirming that it achieves state-of-the-art performance. We further provide a case study for patients suffering from low back pain to demonstrate that DeepACE generates important and meaningful findings for clinical practice. Our work enables practitioners to develop effective treatment recommendations based on population effects.
