A Linear Algebraic Framework for Counterfactual Generation
Jong-Hoon Ahn, Akshay Vashist
Abstract
Estimating individual treatment effects in clinical data is essential for understanding how different patients uniquely respond to treatments and identifying the most effective interventions for specific patient subgroups, thereby enhancing the precision and personalization of healthcare. However, counterfactual data are not accessible, and the true calculation of causal effects cannot be performed at the individual level. This paper proposes a linear algebraic framework to generate counterfactual longitudinal data that exactly matches pre-treatment factual data. Because causation travels forward in time, not in reverse, counterfactual predictability is further strengthened by blocking causal effects from flowing back to the past, thus limiting counterfactual dependence on the future. Using simulated LDL cholesterol datasets, we show that our method significantly outperforms the most cited methods of counterfactual generation. We also provide a formula that can estimate the time-varying variance of individual treatment effects, interpreted as a confidence level in the generated counterfactuals compared to true values.
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 papers1
Ask how each one uses itBuilds on3
- Estimating counterfactual treatment outcomes over time through adversarially balanced representationsIoana Bica, Ahmed M. Alaa, James Jordon, Mihaela van der SchaarICLR 2020 · 224 citations
- SyncTwin: Treatment Effect Estimation with Longitudinal OutcomesZhaozhi Qian, Yao Zhang, Ioana Bica, Angela M. Wood et al.NeurIPS 2021 · 46 citations
- Synthetic Design: An Optimization Approach to Experimental Design with Synthetic ControlsNick Doudchenko, Khashayar Khosravi, Jean Pouget-Abadie, Sébastien Lahaie et al.NeurIPS 2021 · 13 citations
Related papers
- Counterfactual Generative Models for Time-Varying TreatmentsShenghao Wu, Wenbin Zhou, Minshuo Chen, Shixiang ZhuKDD 2024 · 2 citations
- Causal Transformer for Estimating Counterfactual OutcomesValentyn Melnychuk, Dennis Frauen, Stefan FeuerriegelICML 2022 · 146 citations
- Zero-shot causal learningHamed Nilforoshan, Michael Moor, Yusuf H. Roohani, Yining Chen et al.NeurIPS 2023 · 25 citations
- Counterfactual Contrastive Learning with Normalizing Flows for Robust Treatment Effect EstimationJiaxuan Zhang, Emadeldeen Eldele, Fuyuan Cao, Yang Wang et al.ICML 2025
- Policy Analysis using Synthetic Controls in Continuous-TimeAlexis Bellot, Mihaela van der SchaarICML 2021 · 19 citations
