SyncTwin: Treatment Effect Estimation with Longitudinal Outcomes
Zhaozhi Qian, Yao Zhang, Ioana Bica, Angela M. Wood, Mihaela van der Schaar
Abstract
Most of the medical observational studies estimate the causal treatment effects using electronic health records (EHR), where a patient's covariates and outcomes are both observed longitudinally. However, previous methods focus only on adjusting for the covariates while neglecting the temporal structure in the outcomes. To bridge the gap, this paper develops a new method, SyncTwin, that learns a patient-specific time-constant representation from the pre-treatment observations. SyncTwin issues counterfactual prediction of a target patient by constructing a synthetic twin that closely matches the target in representation. The reliability of the estimated treatment effect can be assessed by comparing the observed and synthetic pre-treatment outcomes. The medical experts can interpret the estimate by examining the most important contributing individuals to the synthetic twin. In the real-data experiment, SyncTwin successfully reproduced the findings of a randomized controlled clinical trial using observational data, which demonstrates its usability in the complex real-world EHR.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers9
- Causal Transformer for Estimating Counterfactual OutcomesValentyn Melnychuk, Dennis Frauen, Stefan FeuerriegelICML 2022 · 146 citations
- Estimating Average Causal Effects from Patient TrajectoriesDennis Frauen, Tobias Hatt, Valentyn Melnychuk, Stefan FeuerriegelAAAI 2023 · 34 citations
- Transfer Learning on Heterogeneous Feature Spaces for Treatment Effects EstimationIoana Bica, Mihaela van der SchaarNeurIPS 2022 · 33 citations
- Accounting For Informative Sampling When Learning to Forecast Treatment Outcomes Over TimeToon Vanderschueren, Alicia Curth, Wouter Verbeke, Mihaela van der SchaarICML 2023 · 18 citations
- Causal Contrastive Learning for Counterfactual Regression Over TimeMouad El Bouchattaoui, Myriam Tami, Benoit Lepetit, Paul-Henry CournèdeNeurIPS 2024 · 10 citations
Builds on4
- Estimating counterfactual treatment outcomes over time through adversarially balanced representationsIoana Bica, Ahmed M. Alaa, James Jordon, Mihaela van der SchaarICLR 2020 · 224 citations
- Learning Disentangled Representations for CounterFactual RegressionNegar Hassanpour, Russell GreinerICLR 2020 · 176 citations
- DeepMatch: Balancing Deep Covariate Representations for Causal Inference Using Adversarial TrainingNathan KallusICML 2020 · 84 citations
- Counterfactual Prediction for Bundle TreatmentHao Zou, Peng Cui, Bo Li, Zheyan Shen et al.NeurIPS 2020 · 53 citations
Related papers
- A Linear Algebraic Framework for Counterfactual GenerationJong-Hoon Ahn, Akshay VashistICLR 2024
- TWIN: Personalized Clinical Trial Digital Twin GenerationTrisha Das, Zifeng Wang, Jimeng SunKDD 2023 · 22 citations
- A Two-Stage Pretraining-Finetuning Framework for Treatment Effect Estimation with Unmeasured ConfoundingChuan Zhou, Yaxuan Li, Chunyuan Zheng, Haiteng Zhang et al.KDD 2025 · 6 citations
- SurvHTE-Bench: A Benchmark for Heterogeneous Treatment Effect Estimation in Survival AnalysisShahriar Noroozizadeh, Xiaobin Shen, Jeremy C. Weiss, George H. ChenICLR 2026
- Meta-Learners for Partially-Identified Treatment Effects Across Multiple EnvironmentsJonas Schweisthal, Dennis Frauen, Mihaela van der Schaar, Stefan FeuerriegelICML 2024 · 10 citations
