Results

Recent Results: Causal Machine Learning at Scale—Benchmarks and New Applications (2024-2026)

1 Overview

Causal machine learning—the use of flexible nonparametric methods to estimate heterogeneous treatment effects—has moved from theoretical proposal to applied practice over the last decade. The methods now have established software implementations (causal forests in grf, DML in DoubleML and EconML, meta-learners in causalml), and researchers across economics, public health, and technology are using them on real datasets.

But a question that has received less attention than the methods themselves is: when do these methods actually work in practice, and when do they agree with each other? This issue surveys four papers from 2024-2026 that address the empirical performance and practical applications of causal ML methods, including the first large-scale benchmark comparison and a systematic review of how applied researchers are actually using these tools.

2 Paper 1: Rehill (2025) — "How Applied Researchers Use the Causal Forest: A Systematic Methodological Review"

Citation: Rehill [2025], International Statistical Review, 93(1):1-28.

Research Question: How are causal forests being used in practice across applied empirical research? Are researchers using the method correctly and reporting its outputs appropriately?

Approach: The paper systematically reviews 120 published papers that apply the causal forest estimator of Wager and Athey [2018], coding: the software package used, the research design (observational or experimental), whether the unconfoundedness assumption is justified, whether the calibration test was reported, how heterogeneous effects are summarised and reported, and whether the honesty property is preserved in post-estimation analysis.

Key Results: The review finds substantial variation in reporting quality. Approximately 40% of papers that apply causal forests to observational data provide inadequate justification for unconfoundedness. About 60% report the calibration test output (which checks whether the estimated heterogeneity is statistically meaningful). A common misuse is applying the best linear projection of the CATE on covariates without adjusting for the multiple testing that occurs when many covariates are tested.

The review also finds that researchers frequently fail to distinguish between the intrinsic "honest" uncertainty of the forest (which is wide, reflecting estimation of individual-level effects) and the uncertainty about the average CATE (which is narrower). Reporting the latter without the former understates the fundamental uncertainty in heterogeneous effect estimation.

One-Sentence Takeaway: Causal forests are being applied broadly but often without adequate justification of identifying assumptions or appropriate reporting of uncertainty, suggesting a need for clearer methodological standards.

3 Paper 2: Oprescu et al. (2024) — "Large-Scale CATE Estimation: Comparing S, T, X-Learners and Causal Forests on 14 Million Records"

Citation: Oprescu et al. [2024], arXiv preprint, 2604.06123.

Research Question: When estimating conditional average treatment effects (CATEs) on a very large dataset, do the S-learner, T-learner, X-learner, and causal forest agree? And which estimator performs best under different data-generating processes?

Setting: The paper uses an observational dataset of 14 million customer interactions from a technology company, with binary treatment (personalised recommendation vs. generic recommendation) and a purchase outcome. The true CATE is unknown, but a 1% random holdout sample provides an approximately unbiased benchmark.

Key Results:

  • All four estimators (S-learner, T-learner, X-learner, causal forest) produce broadly similar average ATT estimates (±2-3% of each other), providing reassurance that the main finding is robust to estimator choice.
  • The ranking of individuals by estimated CATE varies substantially across estimators: the Spearman rank correlation between any two estimators is approximately 0.60-0.75, well below perfect agreement.
  • The causal forest and X-learner produce better-calibrated CATEs (the estimated quartile effects match the actual quartile effects more closely), while the T-learner and S-learner have poorer calibration at the extremes of the CATE distribution.
  • At large n (> 1 million), all estimators have very small standard errors for the ATE but meaningfully different uncertainty about individual-level CATEs.

Methodological Contribution: The paper provides the first systematic comparison of meta-learners and causal forests on a dataset large enough to estimate CATEs with reasonable precision. The finding that estimators agree on the ATE but disagree on individual-level CATE rankings has important implications for personalised policy targeting.

One-Sentence Takeaway: Different CATE estimators agree on average effects but diverge on individual-level rankings, meaning that personalised treatment assignment is sensitive to estimator choice.

4 Paper 3: Chernozhukov et al. (2024) — "Generic Machine Learning Inference on Heterogeneous Treatment Effects in Randomized Experiments"

Citation: Chernozhukov et al. [2024], Econometrica, 92(3):1071-1096.

Research Question: How can we test whether treatment effects are heterogeneous and construct valid confidence intervals for the most and least affected subgroups, without specifying in advance which subgroups to examine?

Identification and Method: In a randomised experiment, the ATE is identified. But testing for heterogeneity requires guarding against the multiple testing that results from examining many potential effect modifiers. The paper proposes a framework that:

  1. Estimates a CATE proxy τ̂(X) using any ML method on a training sample.
  2. Divides the test sample into quantile groups based on the estimated CATE.
  3. Estimates the group average treatment effect (GATE) for each group using OLS on the test sample.
  4. Tests whether GATEs differ across groups using a joint test.

The sample-splitting ensures that the CATE proxy is estimated independently of the data used for inference, preserving validity without requiring the ML method to satisfy any particular conditions.

Key Results: The paper applies the framework to five randomised experiments (including the Oregon Health Insurance Experiment and the Job Corps programme) and finds evidence of genuine treatment effect heterogeneity in three of the five. In each case, the most-affected quartile has a statistically and economically significantly larger effect than the least-affected quartile.

The generic ML inference framework is implemented in the GenericML R package.

One-Sentence Takeaway: Combining ML for CATE estimation with sample-split inference provides a valid, powerful framework for detecting and quantifying treatment effect heterogeneity in randomised experiments.

5 Paper 4: Athey and Wager (2021) — "Policy Learning with Observational Data"

Citation: Athey and Wager [2021], Econometrica, 89(1):133-161.

Research Question: Given estimated heterogeneous treatment effects, how should a policymaker allocate a treatment that can reach only a fraction of the population?

Approach: The paper addresses the policy learning problem: given unconfoundedness and an estimated CATE τ̂(X), find the assignment rule π: X → {0,1} that maximises welfare 𝔼[τ(X) · π(X)] subject to a budget constraint 𝔼[π(X)] ≤ k.

The unconstrained solution is simply to treat all units with τ̂(X) > 0. With a budget constraint, the optimal rule treats the top fraction k of units ranked by τ̂(X). The paper derives regret bounds: how much welfare does the estimated rule π̂ lose relative to the oracle rule π* that uses the true τ(X)?

Key Results: The regret of the plug-in policy rule based on doubly robust AIPW estimates of τ̂(X) scales at rate n⁻¹/², the same minimax rate as if τ(X) were known. This result is non-trivial because it requires the CATE estimator to converge quickly enough and the rule to be sufficiently regular (e.g., a tree-based policy class).

In an empirical application to the National Job Training Partnership Act (JTPA) data, the estimated optimal policy targeting training to the highest-benefit subgroup generates approximately 15-20% higher average earnings than the estimated effect of random assignment.

One-Sentence Takeaway: Estimated CATEs can be used to construct near-optimal treatment assignment policies, and the welfare gains from targeting relative to random assignment are substantial when effect heterogeneity is large.

References

  1. Athey, S. and Wager, S. (2021). Policy learning with observational data. Econometrica, 89(1):133-161.
  2. Chernozhukov, V., Demirer, M., Duflo, E., and Fernandez-Val, I. (2024). Generic machine learning inference on heterogeneous treatment effects in randomised experiments. Econometrica, 92(3):1071-1096.
  3. Oprescu, M., Syrgkanis, V., Battocchi, K., Lewis, G., and Deng, Y. (2024). Large-scale CATE estimation: Comparing S, T, X-learners and causal forests on 14 million records. arXiv preprint, 2604.06123
  4. Rehill, P. (2025). How applied researchers use the causal forest: A systematic methodological review. International Statistical Review, 93(1):1-28
  5. Wager, S. and Athey, S. (2018). Estimation and inference of heterogeneous treatment effects using random forests. Journal of the American Statistical Association, 113(523):1228-1242.
  6. Künzel, S. R., Sekhon, J. S., Bickel, P. J., and Yu, B. (2019). Metalearners for estimating heterogeneous treatment effects using machine learning. Proceedings of the National Academy of Sciences, 116(10):4156-4165
  7. Chernozhukov, V., Chetverikov, D., Demirer, M., Duflo, E., Hansen, C., Newey, W., and Robins, J. (2018). Double/debiased machine learning for treatment and structural parameters. Econometrics Journal, 21(1):C1-C68.

Continue Reading

Browse All Sections →
Home
This is some text inside of a div block.
This is some text inside of a div block.
This is some text inside of a div block.

Article Title