Learning Representations for Counterfactual Inference
Not on the map. Why?
Abstract (arXiv)
Observational studies are rising in importance due to the widespread accumulation of data in fields such as healthcare, education, employment and ecology. We consider the task of answering counterfactual questions such as, "Would this patient have lower blood sugar had she received a different medication?". We propose a new algorithmic framework for counterfactual inference which brings together ideas from domain adaptation and representation learning. In addition to a theoretical justification, we perform an empirical comparison with previous approaches to causal inference from observational data. Our deep learning algorithm significantly outperforms the previous state-of-the-art.
Key facts. Cited by 327 papers in the map (data through 7 Oct 2026). First posted on arXiv 12 May 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 56; 2022: 62; 2023: 69; 2024: 70; 2025: 44; 2026: 26.
327 papers.
Sort:
By area:
- 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
- 2022Causal Transformer for Estimating Counterfactual Outcomescited by 121
- 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inferencecited by 97
- 2022Diffusion Causal Models for Counterfactual Estimationcited by 87
- 2021Size-Invariant Graph Representations for Graph Classification Extrapolationscited by 82
- 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 74
- 2021Learning from Counterfactual Links for Link Predictioncited by 69
- 2021VCNet and Functional Targeted Regularization For Learning Causal Effects of Continuous Treatmentscited by 59
- 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
- 2021Causally motivated Shortcut Removal Using Auxiliary Labelscited by 55
- 2022Neural Causal Models for Counterfactual Identification and Estimationcited by 54
- 2022A Causal Lens for Controllable Text Generationcited by 45
38 more
- 2023A Survey on Causal Reinforcement Learningcited by 43
- 2023Counterfactual Identifiability of Bijective Causal Modelscited by 42
- 2021Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regressioncited by 40
- 2021Deep Proxy Causal Learning and its Application to Confounded Bandit Policy Evaluationcited by 39
- 2021Causal Effect Inference for Structured Treatmentscited by 38
- 2022Combining Observational and Randomized Data for Estimating Heterogeneous Treatment Effectscited by 32
- 2023Optimal Transport for Treatment Effect Estimationcited by 32
- 2022Data-Driven Offline Decision-Making via Invariant Representation Learningcited by 31
- 2025Foundation Models for Causal Inference via Prior-Data Fitted Networkscited by 30
- 2023In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect Estimationcited by 29
- 2023Bayesian Neural Controlled Differential Equations for Treatment Effect Estimationcited by 26
- 2021Estimating Average Treatment Effects via Orthogonal Regularizationcited by 25
- 2021SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event Datacited by 25
- 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimationcited by 25
- 2021Sequential Deconfounding for Causal Inference with Unobserved Confounderscited by 24
- 2024Causality for Large Language Modelscited by 24
- 2023Conformal Meta-learners for Predictive Inference of Individual Treatment Effectscited by 23
- 2021E-Commerce Promotions Personalization via Online Multiple-Choice Knapsack with Uplift Modelingcited by 21
- 2022Normalizing Flows for Interventional Density Estimationcited by 21
- 2022Direct Heterogeneous Causal Learning for Resource Allocation Problems in Marketingcited by 21
- 2023Bounds on Representation-Induced Confounding Bias for Treatment Effect Estimationcited by 20
- 2024Neural Causal Abstractionscited by 20
- 2024Doubly Robust Causal Effect Estimation under Networked Interference via Targeted Learningcited by 19
- 2021Auto IV: Counterfactual Prediction via Automatic Instrumental Variable Decompositioncited by 18
- 2021Estimating Categorical Counterfactuals via Deep Twin Networkscited by 18
- 2021A Theoretical Analysis on Independence-driven Importance Weighting for Covariate-shift Generalizationcited by 18
- 2022Robust and Agnostic Learning of Conditional Distributional Treatment Effectscited by 18
- 2022Learning Disentangled Representations for Counterfactual Regression via Mutual Information Minimizationcited by 18
- 2024Model-agnostic meta-learners for estimating heterogeneous treatment effects over timecited by 18
- 2023Zero-shot causal learningcited by 17
- 2023Estimating Treatment Effects from Irregular Time Series Observations with Hidden Confounderscited by 17
- 2023Towards Causal Foundation Model: on Duality between Causal Inference and Attentioncited by 17
- 2023Causal Inference with Conditional Front-Door Adjustment and Identifiable Variational Autoencodercited by 17
- 2023A Neural Framework for Generalized Causal Sensitivity Analysiscited by 17
- 2024DiffPO: A causal diffusion model for learning distributions of potential outcomescited by 17
- 2021Deconfounding Temporal Autoencoder: Estimating Treatment Effects over Time Using Noisy Proxiescited by 16
- 2022DESCN: Deep Entire Space Cross Networks for Individual Treatment Effect Estimationcited by 16
- 2023CF-GODE: Continuous-Time Causal Inference for Multi-Agent Dynamical Systemscited by 16
50 most cited of 327 All 327 (JSON)
- 2026When AI Generates Covariates: Causal Typing and Estimand Drift in Sequential Experiments15 Sep 2026
- 2026Vision-Language Grounded Task-Context-Aware Imitation Learning for Robotic Disassembly15 Sep 2026
- 2026Causal multi-modal AI for personalized chemosensitivity prediction11 Sep 2026
- 2026Representation Learning for Sample-Efficient CATE Estimation by Leveraging Multiple Outcomes5 Sep 2026
- 2026CASTANET: Causality-Aware Spatio-Temporal Adversarial Network Using Traffic Incident Effects28 Aug 2026
- 2026From Multimodal Observation to Interpretable Suggestions: Counterfactual Time-Expanded Relational Modeling of Surgical Teams24 Aug 2026
- 2026Doubly Robust Estimation of Causal Effect on CVR with Targeted Regularization13 Aug 2026
- 2026RAGuard: A Layered Defense Framework for Retrieval-Augmented Generation Systems Against Data Poisoning28 Jul 2026
- 2026CausalGraphX: A Counterfactual Graph Neural Network Framework for Explainable Systemic Risk Assessment15 Jul 2026
- 2026Counterfactual Optimal Action Trees (COAT): Interpretable Prescriptive Policies from Observational Data15 Jul 2026
- 2026Representation Learning for Semiparametric Causal Mediation Analysis under No Essential Heterogeneity12 Jul 2026
- 2026GUSH3R: Everyone Everywhere All at Once as Gaussians6 Jul 2026
38 more
- 2026Balanced Twins: Causal Inference on Time Series with Hidden Confounding17 Jun 2026
- 2026Disentangled Double Machine Learning for Accurate Causal Effect Estimation24 May 2026
- 2026Treatment Effect Estimation with Differentiated Networked Effect on Graph Data23 May 2026
- 2026Decision-Aware Proximal Bridge Learning for Optimal Treatment Selection16 May 2026
- 2026Real vs. Semi-Simulated: Rethinking Evaluation for Treatment Effect Estimation11 May 2026
- 2026PerCaM-Health: Personalized Dynamic Causal Graphs for Healthcare Reasoning8 May 2026
- 2026BGM-IV: AI-Powered Bayesian Generative Modeling for Instrumental Variable Regression with High-Dimensional Covariates7 May 2026
- 2026MOCA: A Transformer-based Modular Causal Inference Framework with One-way Cross-attention and Cutting Feedback25 Apr 2026
- 2026Retrospective Counterfactual Prediction by Conditioning on the Factual Outcome: A Cross-World Approach28 Mar 2026
- 2026Evaluating Uplift Modeling under Structural Biases: Insights into Metric Stability and Model Robustness21 Mar 2026
- 2026Improving RCT-Based Treatment Effect Estimation Under Covariate Mismatch via Calibrated Alignment19 Mar 2026
- 2026Causal Representation Learning with Optimal Compression under Complex Treatments12 Mar 2026
- 2026Estimating Visual Attribute Effects in Advertising from Observational Data: A Deepfake-Informed Double Machine Learning Approach2 Mar 2026
- 2026Knowledge-Guided Time-Varying Causal Inference for Arctic Sea Ice Dynamics25 Jan 2026
- 2025PIPCFR: Pseudo-outcome Imputation with Post-treatment Variables for Individual Treatment Effect Estimation21 Dec 2025
- 2025Synergizing Deconfounding and Temporal Generalization For Time-series Counterfactual Outcome Estimation20 Nov 2025
- 2025Multiple Treatments Causal Effects Estimation with Task Embeddings and Balanced Representation Learning12 Nov 2025
- 2025Cross-Treatment Effect Estimation for Multi-Category, Multi-Valued Causal Inference via Dynamic Neural Masking3 Nov 2025
- 2025A Comparative Study of Model Adaptation Strategies for Multi-Treatment Uplift Modeling3 Nov 2025
- 2025Estimating Treatment Effects in Networks under Unknown Exposure Mappings24 Oct 2025
- 2025Bi-Level Decision-Focused Causal Learning for Large-Scale Marketing Optimization: Bridging Observational and Experimental Data22 Oct 2025
- 2025Adversary-Free Counterfactual Prediction via Information-Regularized Representations17 Oct 2025
- 2025Data-driven simulator of multi-animal behavior with unknown dynamics via offline and online reinforcement learning12 Oct 2025
- 2025Overlap-Adaptive Regularization for Conditional Average Treatment Effect Estimation29 Sep 2025
- 2025GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomes26 Sep 2025
- 2025CausalKANs: interpretable treatment effect estimation with Kolmogorov-Arnold networks26 Sep 2025
- 2025Causal Machine Learning for Surgical Interventions24 Sep 2025
- 2025Augmenting Limited and Biased RCTs through Pseudo-Sample Matching-Based Observational Data Fusion Method16 Sep 2025
- 2025Synthetic Counterfactual Labels for Efficient Conformal Counterfactual Inference4 Sep 2025
- 2025LLM-based Agents for Automated Confounder Discovery and Subgroup Analysis in Causal Inference10 Aug 2025
- 2025SACO: Sequence-Aware Constrained Optimization Framework for Coupon Distribution in E-commerce8 Aug 2025
- 2025Structure Maintained Representation Learning Neural Network for Causal Inference3 Aug 2025
- 2025Annotation-Assisted Learning of Treatment Policies From Multimodal Electronic Health Records28 Jul 2025
- 2025Causality-aligned Prompt Learning via Diffusion-based Counterfactual Generation26 Jul 2025
- 2025Consistent Labeling Across Group Assignments: Variance Reduction in Conditional Average Treatment Effect Estimation6 Jul 2025
- 2025LLM-Driven Treatment Effect Estimation Under Inference Time Text Confounding3 Jul 2025
- 2025Hybrid Meta-learners for Estimating Heterogeneous Treatment Effects16 Jun 2025
- 2025Foundation Models for Causal Inference via Prior-Data Fitted Networks12 Jun 2025
50 newest of 327 All 327 (JSON)
- 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
- 2022Causal Transformer for Estimating Counterfactual Outcomescited by 121
- 2021The Causal-Neural Connection: Expressiveness, Learnability, and Inferencecited by 97
- 2022Diffusion Causal Models for Counterfactual Estimationcited by 87
- 2021Size-Invariant Graph Representations for Graph Classification Extrapolationscited by 82
- 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 74
- 2021Learning from Counterfactual Links for Link Predictioncited by 69
- 2021VCNet and Functional Targeted Regularization For Learning Causal Effects of Continuous Treatmentscited by 59
- 2022Continuous-Time Modeling of Counterfactual Outcomes Using Neural Controlled Differential Equationscited by 59
- 2021Causally motivated Shortcut Removal Using Auxiliary Labelscited by 55
- 2022Neural Causal Models for Counterfactual Identification and Estimationcited by 54
- 2022A Causal Lens for Controllable Text Generationcited by 45
12 most cited of 322 All 322 (JSON)
- 2026When AI Generates Covariates: Causal Typing and Estimand Drift in Sequential Experiments15 Sep 2026
- 2026Causal multi-modal AI for personalized chemosensitivity prediction11 Sep 2026
- 2026Representation Learning for Sample-Efficient CATE Estimation by Leveraging Multiple Outcomes5 Sep 2026
- 2026CASTANET: Causality-Aware Spatio-Temporal Adversarial Network Using Traffic Incident Effects28 Aug 2026
- 2026From Multimodal Observation to Interpretable Suggestions: Counterfactual Time-Expanded Relational Modeling of Surgical Teams24 Aug 2026
- 2026Doubly Robust Estimation of Causal Effect on CVR with Targeted Regularization13 Aug 2026
- 2026CausalGraphX: A Counterfactual Graph Neural Network Framework for Explainable Systemic Risk Assessment15 Jul 2026
- 2026Counterfactual Optimal Action Trees (COAT): Interpretable Prescriptive Policies from Observational Data15 Jul 2026
- 2026Representation Learning for Semiparametric Causal Mediation Analysis under No Essential Heterogeneity12 Jul 2026
- 2026GUSH3R: Everyone Everywhere All at Once as Gaussians6 Jul 2026
- 2026Balanced Twins: Causal Inference on Time Series with Hidden Confounding17 Jun 2026
- 2026Disentangled Double Machine Learning for Accurate Causal Effect Estimation24 May 2026
12 newest of 322 All 322 (JSON)