mlsanityData to 7 Oct 2026

Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learning

arXiv 1706.03461 ↗First posted 12 Jun 2017math.ST stat.ME stat.THcited by 475

Not on the map. Why?

Abstract (arXiv)

There is growing interest in estimating and analyzing heterogeneous treatment effects in experimental and observational studies. We describe a number of meta-algorithms that can take advantage of any supervised learning or regression method in machine learning and statistics to estimate the Conditional Average Treatment Effect (CATE) function. Meta-algorithms build on base algorithms---such as Random Forests (RF), Bayesian Additive Regression Trees (BART) or neural networks---to estimate the CATE, a function that the base algorithms are not designed to estimate directly. We introduce a new meta-algorithm, the X-learner, that is provably efficient when the number of units in one treatment group is much larger than in the other, and can exploit structural properties of the CATE function. For example, if the CATE function is linear and the response functions in treatment and control are Lipschitz continuous, the X-learner can still achieve the parametric rate under regularity conditions. We then introduce versions of the X-learner that use RF and BART as base learners. In extensive simulation studies, the X-learner performs favorably, although none of the meta-learners is uniformly the best. In two persuasion field experiments from political science, we demonstrate how our new X-learner can be used to target treatment regimes and to shed light on underlying mechanisms. A software package is provided that implements our methods.

Key facts. Cited by 475 papers in the map (data through 7 Oct 2026). First posted on arXiv 12 Jun 2017.

Builds on

No reference list yet.

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2021: 32; 2022: 65; 2023: 92; 2024: 94; 2025: 110; 2026: 82.

475 papers.

Sort:

By area:

  1. 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
  2. 2025TabPFN-2.5: Advancing the State of the Art in Tabular Foundation Modelscited by 115
  3. 2022Deep End-to-end Causal Inferencecited by 93
  4. 2021On Inductive Biases for Heterogeneous Treatment Effect Estimationcited by 74
  5. 2024Causal machine learning for predicting treatment outcomescited by 73
  6. 2022Causal Machine Learning for Healthcare and Precision Medicinecited by 68
  7. 2022CEBaB: Estimating the Causal Effects of Real-World Concepts on NLP Model Behaviorcited by 54
  8. 2025Do-PFN: In-Context Learning for Causal Effect Estimationcited by 46
  9. 2023Faithful Explanations of Black-box NLP Models Using LLM-generated Counterfactualscited by 44
  10. 2023A Survey on Causal Reinforcement Learningcited by 43
  11. 2025CausalPFN: Amortized Causal Effect Estimation via In-Context Learningcited by 43
  12. 2023Spawrious: A Benchmark for Fine Control of Spurious Correlation Biasescited by 41
38 more
  1. 2021Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regressioncited by 40
  2. 2021Causal Effect Inference for Structured Treatmentscited by 38
  3. 2021Selective Regression Under Fairness Criteriacited by 34
  4. 2023Optimal Transport for Treatment Effect Estimationcited by 32
  5. 2025Foundation Models for Causal Inference via Prior-Data Fitted Networkscited by 30
  6. 2021Evaluating Treatment Prioritization Rules via Rank-Weighted Average Treatment Effectscited by 29
  7. 2022Causal Proxy Models for Concept-Based Model Explanationscited by 29
  8. 2023In Search of Insights, Not Magic Bullets: Towards Demystification of the Model Selection Dilemma in Heterogeneous Treatment Effect Estimationcited by 29
  9. 2023B-Learner: Quasi-Oracle Bounds on Heterogeneous Causal Effects Under Hidden Confoundingcited by 28
  10. 2025Causal LLM Routing: End-to-End Regret Minimization from Observational Datacited by 27
  11. 2021SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event Datacited by 25
  12. 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimationcited by 25
  13. 2022Empirical Analysis of Model Selection for Heterogeneous Causal Effect Estimationcited by 25
  14. 2023Explicit Feature Interaction-aware Uplift Network for Online Marketingcited by 25
  15. 2024Conformal Prediction: A Data Perspectivecited by 25
  16. 2023Sharp Bounds for Generalized Causal Sensitivity Analysiscited by 24
  17. 2023Conformal Meta-learners for Predictive Inference of Individual Treatment Effectscited by 23
  18. 2021E-Commerce Promotions Personalization via Online Multiple-Choice Knapsack with Uplift Modelingcited by 21
  19. 2022Scientific Inference With Interpretable Machine Learning: Analyzing Models to Learn About Real-World Phenomenacited by 21
  20. 2022Normalizing Flows for Interventional Density Estimationcited by 21
  21. 2022Transfer Learning on Heterogeneous Feature Spaces for Treatment Effects Estimationcited by 21
  22. 2022Direct Heterogeneous Causal Learning for Resource Allocation Problems in Marketingcited by 21
  23. 2022Validating Causal Inference Methodscited by 20
  24. 2022Scalable Sensitivity and Uncertainty Analysis for Causal-Effect Estimates of Continuous-Valued Interventionscited by 20
  25. 2022Image-based Treatment Effect Heterogeneitycited by 20
  26. 2023Bounds on Representation-Induced Confounding Bias for Treatment Effect Estimationcited by 20
  27. 2023An Adaptive Kernel Approach to Federated Learning of Heterogeneous Causal Effectscited by 19
  28. 2022Robust and Agnostic Learning of Conditional Distributional Treatment Effectscited by 18
  29. 2024Model-agnostic meta-learners for estimating heterogeneous treatment effects over timecited by 18
  30. 2022Estimating individual treatment effects under unobserved confounding using binary instrumentscited by 17
  31. 2023Zero-shot causal learningcited by 17
  32. 2023Causal Inference with Conditional Front-Door Adjustment and Identifiable Variational Autoencodercited by 17
  33. 2023A Neural Framework for Generalized Causal Sensitivity Analysiscited by 17
  34. 2024DiffPO: A causal diffusion model for learning distributions of potential outcomescited by 17
  35. 2022Personalized Public Policy Analysis in Social Sciences using Causal-Graphical Normalizing Flowscited by 16
  36. 2022DESCN: Deep Entire Space Cross Networks for Individual Treatment Effect Estimationcited by 16
  37. 2023Causal isotonic calibration for heterogeneous treatment effectscited by 16
  38. 2022Benchmarking Heterogeneous Treatment Effect Models through the Lens of Interpretabilitycited by 15

50 most cited of 475 All 475 (JSON)