Treatment effect estimation with disentangled latent factors
Not on the map. Why?
Abstract (arXiv)
Much research has been devoted to the problem of estimating treatment effects from observational data; however, most methods assume that the observed variables only contain confounders, i.e., variables that affect both the treatment and the outcome. Unfortunately, this assumption is frequently violated in real-world applications, since some variables only affect the treatment but not the outcome, and vice versa. Moreover, in many cases only the proxy variables of the underlying confounding factors can be observed. In this work, we first show the importance of differentiating confounding factors from instrumental and risk factors for both average and conditional average treatment effect estimation, and then we propose a variational inference approach to simultaneously infer latent factors from the observed variables, disentangle the factors into three disjoint sets corresponding to the instrumental, confounding, and risk factors, and use the disentangled factors for treatment effect estimation. Experimental results demonstrate the effectiveness of the proposed method on a wide range of synthetic, benchmark, and real-world datasets.
Key facts. Cited by 63 papers in the map (data through 7 Oct 2026). First posted on arXiv 29 Jan 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 3; 2022: 12; 2023: 16; 2024: 16; 2025: 10; 2026: 6.
63 papers.
Sort:
By area:
- 2022Learning Disentangled Representations for Counterfactual Regression via Mutual Information Minimizationcited by 18
- 2023Causal Inference with Conditional Front-Door Adjustment and Identifiable Variational Autoencodercited by 17
- 2024DiffPO: A causal diffusion model for learning distributions of potential outcomescited by 17
- 2023Causal Effect Estimation: Recent Advances, Challenges, and Opportunitiescited by 14
- 2022Deep Multi-Modal Structural Equations For Causal Effect Estimation With Unstructured Proxiescited by 13
- 2022Long-term Causal Effects Estimation via Latent Surrogates Representation Learningcited by 11
- 2022Deep Causal Learning: Representation, Discovery and Inferencecited by 11
- 2023Dynamic Inter-treatment Information Sharing for Individualized Treatment Effects Estimationcited by 11
- 2022Instrumental Variables in Causal Inference and Machine Learning: A Surveycited by 10
- 2023Disentangled Representation for Causal Mediation Analysiscited by 10
- 2023Adversarially Balanced Representation for Continuous Treatment Effect Estimationcited by 10
- 2022Adversarial De-confounding in Individualised Treatment Effects Estimationcited by 9
38 more
- 2024Modular Learning of Deep Causal Generative Models for High-dimensional Causal Inferencecited by 9
- 2021β-Intact-VAE: Identifying and Estimating Causal Effects under Limited Overlapcited by 8
- 2022Causal Inference with Conditional Instruments using Deep Generative Modelscited by 6
- 2024Disentangled Representation Learning for Causal Inference with Instrumentscited by 6
- 2023Causal Inference from Text: Unveiling Interactions between Variablescited by 5
- 2021Towards Principled Causal Effect Estimation by Deep Identifiable Modelscited by 4
- 2024Conditional Generative Models are Sufficient to Sample from Any Causal Effect Estimandcited by 4
- 2024Causal Inference with Latent Variables: Recent Advances and Future Prospectivescited by 4
- 2025Counterfactual Probabilistic Diffusion with Expert Modelscited by 4
- 2023High Dimensional Causal Inference with Variational Backdoor Adjustmentcited by 3
- 2024Self-Distilled Disentangled Learning for Counterfactual Predictioncited by 3
- 2024Estimating Conditional Average Treatment Effects via Sufficient Representation Learningcited by 3
- 2025Long-Term Individual Causal Effect Estimation via Identifiable Latent Representation Learningcited by 3
- 2022Trying to Outrun Causality with Machine Learning: Limitations of Model Explainability Techniques for Identifying Predictive Variablescited by 2
- 2022A Causal Research Pipeline and Tutorial for Psychologists and Social Scientistscited by 2
- 2023DR-VIDAL -- Doubly Robust Variational Information-theoretic Deep Adversarial Learning for Counterfactual Prediction and Treatment Effect Estimation on Real World Datacited by 2
- 2023Towards Learning and Explaining Indirect Causal Effects in Neural Networkscited by 2
- 2024Causal GNNs: A GNN-Driven Instrumental Variable Approach for Causal Inference in Networkscited by 2
- 2021Variational Auto-Encoder Architectures that Excel at Causal Inferencecited by 1
- 2022NESTER: An Adaptive Neurosymbolic Method for Causal Effect Estimationcited by 1
- 2022Confounder Balancing for Instrumental Variable Regression with Latent Variablecited by 1
- 2023Causal Effect Estimation with Variational AutoEncoder and the Front Door Criterioncited by 1
- 2023SLEM: Machine Learning for Path Modeling and Causal Inference with Super Learner Equation Modelingcited by 1
- 2023Counterfactual Prediction Under Selective Confoundingcited by 1
- 2023Estimating treatment effects from single-arm trials via latent-variable modelingcited by 1
- 2024On the Effects of Irrelevant Variables in Treatment Effect Estimation with Deep Disentanglementcited by 1
- 2024Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variablescited by 1
- 2025Causal Effect Estimation under Networked Interference without Networked Unconfoundedness Assumptioncited by 1
- 2025Federated Instrumental Variable Analysis via Federated Generalized Method of Momentscited by 1
- 2025Deep Disentangled Representation Network for Treatment Effect Estimationcited by 1
- 2025GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomescited by 1
- 2025Text Rationalization for Robust Causal Effect Estimationcited by 1
- 2026Generalized Bayes for Causal Inferencecited by 1
- 2022Variational Temporal Deconfounder for Individualized Treatment Effect Estimation from Longitudinal Observational Datacited by 0
- 2023Linking a predictive model to causal effect estimationcited by 0
- 2023Variational Counterfactual Prediction under Runtime Domain Corruptioncited by 0
- 2023Disentangled Latent Representation Learning for Tackling the Confounding M-Bias Problem in Causal Inferencecited by 0
- 2024Disentangled Representation via Variational AutoEncoder for Continuous Treatment Effect Estimationcited by 0
50 most cited of 63 All 63 (JSON)
- 2026Representation Learning for Sample-Efficient CATE Estimation by Leveraging Multiple Outcomes5 Sep 2026
- 2026Uncertainty-Guided LLM Semantic Augmentation for Heterogeneous Treatment Effect Estimation29 Jul 2026
- 2026Multi-Task Anti-Causal Learning for Reconstructing Urban Events from Residents' Reports12 Mar 2026
- 2026Proxy-Guided Measurement Calibration10 Mar 2026
- 2026Generalized Bayes for Causal Inference3 Mar 2026
- 2026Towards causal effect estimation with learned instrument representations10 Feb 2026
- 2025Text Rationalization for Robust Causal Effect Estimation5 Dec 2025
- 2025GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomes26 Sep 2025
- 2025Counterfactual Probabilistic Diffusion with Expert Models18 Aug 2025
- 2025Multi-Treatment-DML: Causal Estimation for Multi-Dimensional Continuous Treatments with Monotonicity Constraints in Personal Loan Risk Optimization4 Aug 2025
- 2025Deep Disentangled Representation Network for Treatment Effect Estimation9 Jul 2025
- 2025Consistent Labeling Across Group Assignments: Variance Reduction in Conditional Average Treatment Effect Estimation6 Jul 2025
38 more
- 2025Federated Instrumental Variable Analysis via Federated Generalized Method of Moments27 May 2025
- 2025A Generative Framework for Causal Estimation via Importance-Weighted Diffusion Distillation16 May 2025
- 2025Long-Term Individual Causal Effect Estimation via Identifiable Latent Representation Learning8 May 2025
- 2025Causal Effect Estimation under Networked Interference without Networked Unconfoundedness Assumption27 Feb 2025
- 2024Disentangled Graph Autoencoder for Treatment Effect Estimation19 Dec 2024
- 2024Disentangled Representation Learning for Causal Inference with Instruments5 Dec 2024
- 2024Leaning Time-Varying Instruments for Identifying Causal Effects in Time-Series Data26 Nov 2024
- 2024CaTs and DAGs: Integrating Directed Acyclic Graphs with Transformers for Causally Constrained Predictions18 Oct 2024
- 2024DiffPO: A causal diffusion model for learning distributions of potential outcomes11 Oct 2024
- 2024Causal GNNs: A GNN-Driven Instrumental Variable Approach for Causal Inference in Networks13 Sep 2024
- 2024Estimating Conditional Average Treatment Effects via Sufficient Representation Learning30 Aug 2024
- 2024Causal Effect Estimation using identifiable Variational AutoEncoder with Latent Confounders and Post-Treatment Variables13 Aug 2024
- 2024Generalized Encouragement-Based Instrumental Variables for Counterfactual Regression10 Aug 2024
- 2024On the Effects of Irrelevant Variables in Treatment Effect Estimation with Deep Disentanglement29 Jul 2024
- 2024Causal Inference with Latent Variables: Recent Advances and Future Prospectives20 Jun 2024
- 2024Self-Distilled Disentangled Learning for Counterfactual Prediction9 Jun 2024
- 2024Enhancing predictive imaging biomarker discovery through treatment effect analysis4 Jun 2024
- 2024Disentangled Representation via Variational AutoEncoder for Continuous Treatment Effect Estimation4 Jun 2024
- 2024Conditional Generative Models are Sufficient to Sample from Any Causal Effect Estimand12 Feb 2024
- 2024Modular Learning of Deep Causal Generative Models for High-dimensional Causal Inference2 Jan 2024
- 2023Adversarially Balanced Representation for Continuous Treatment Effect Estimation17 Dec 2023
- 2023Disentangled Latent Representation Learning for Tackling the Confounding M-Bias Problem in Causal Inference8 Dec 2023
- 2023Causal Inference from Text: Unveiling Interactions between Variables9 Nov 2023
- 2023Estimating treatment effects from single-arm trials via latent-variable modeling6 Nov 2023
- 2023Counterfactual Prediction Under Selective Confounding21 Oct 2023
- 2023High Dimensional Causal Inference with Variational Backdoor Adjustment9 Oct 2023
- 2023Causal Inference with Conditional Front-Door Adjustment and Identifiable Variational Autoencoder3 Oct 2023
- 2023SLEM: Machine Learning for Path Modeling and Causal Inference with Super Learner Equation Modeling8 Aug 2023
- 2023Variational Counterfactual Prediction under Runtime Domain Corruption23 Jun 2023
- 2023Dynamic Inter-treatment Information Sharing for Individualized Treatment Effects Estimation25 May 2023
- 2023Causal Effect Estimation with Variational AutoEncoder and the Front Door Criterion24 Apr 2023
- 2023Linking a predictive model to causal effect estimation10 Apr 2023
- 2023Towards Learning and Explaining Indirect Causal Effects in Neural Networks24 Mar 2023
- 2023DR-VIDAL -- Doubly Robust Variational Information-theoretic Deep Adversarial Learning for Counterfactual Prediction and Treatment Effect Estimation on Real World Data7 Mar 2023
- 2023Disentangled Representation for Causal Mediation Analysis19 Feb 2023
- 2023Causal Effect Estimation: Recent Advances, Challenges, and Opportunities2 Feb 2023
- 2022Instrumental Variables in Causal Inference and Machine Learning: A Survey12 Dec 2022
- 2022Causal Inference with Conditional Instruments using Deep Generative Models29 Nov 2022
50 newest of 63 All 63 (JSON)
- 2022Learning Disentangled Representations for Counterfactual Regression via Mutual Information Minimizationcited by 18
- 2023Causal Inference with Conditional Front-Door Adjustment and Identifiable Variational Autoencodercited by 17
- 2023Causal Effect Estimation: Recent Advances, Challenges, and Opportunitiescited by 14
- 2022Deep Multi-Modal Structural Equations For Causal Effect Estimation With Unstructured Proxiescited by 13
- 2022Long-term Causal Effects Estimation via Latent Surrogates Representation Learningcited by 11
- 2022Deep Causal Learning: Representation, Discovery and Inferencecited by 11
- 2023Dynamic Inter-treatment Information Sharing for Individualized Treatment Effects Estimationcited by 11
- 2022Instrumental Variables in Causal Inference and Machine Learning: A Surveycited by 10
- 2023Disentangled Representation for Causal Mediation Analysiscited by 10
- 2023Adversarially Balanced Representation for Continuous Treatment Effect Estimationcited by 10
- 2022Adversarial De-confounding in Individualised Treatment Effects Estimationcited by 9
- 2024Modular Learning of Deep Causal Generative Models for High-dimensional Causal Inferencecited by 9
12 most cited of 61 All 61 (JSON)
- 2026Representation Learning for Sample-Efficient CATE Estimation by Leveraging Multiple Outcomes5 Sep 2026
- 2026Uncertainty-Guided LLM Semantic Augmentation for Heterogeneous Treatment Effect Estimation29 Jul 2026
- 2026Multi-Task Anti-Causal Learning for Reconstructing Urban Events from Residents' Reports12 Mar 2026
- 2026Proxy-Guided Measurement Calibration10 Mar 2026
- 2026Generalized Bayes for Causal Inference3 Mar 2026
- 2026Towards causal effect estimation with learned instrument representations10 Feb 2026
- 2025Text Rationalization for Robust Causal Effect Estimation5 Dec 2025
- 2025GDR-learners: Orthogonal Learning of Generative Models for Potential Outcomes26 Sep 2025
- 2025Counterfactual Probabilistic Diffusion with Expert Models18 Aug 2025
- 2025Multi-Treatment-DML: Causal Estimation for Multi-Dimensional Continuous Treatments with Monotonicity Constraints in Personal Loan Risk Optimization4 Aug 2025
- 2025Deep Disentangled Representation Network for Treatment Effect Estimation9 Jul 2025
- 2025Consistent Labeling Across Group Assignments: Variance Reduction in Conditional Average Treatment Effect Estimation6 Jul 2025
12 newest of 61 All 61 (JSON)