# Conformal Meta-learners for Predictive Inference of Individual Treatment Effects

arXiv: https://arxiv.org/abs/2308.14895 · HTML: https://mlsanity.com/paper/2308.14895/

## Key facts

- Cited by 23 papers in the map (data through 14 Oct 2026).
- Builds on 21 of its 51 references that are in the map.
- First posted on arXiv 28 Aug 2023.

## Builds on (most cited first, 21 of 21)

- [Unsupervised Domain Adaptation by Backpropagation](https://mlsanity.com/paper/1409.7495/) (2014)
- [VisualBERT: A Simple and Performant Baseline for Vision and Language](https://mlsanity.com/paper/1908.03557/) (2019)
- [Distribution-Free Predictive Inference For Regression](https://mlsanity.com/paper/1604.04173/) (2016)
- [Conformalized Quantile Regression](https://mlsanity.com/paper/1905.03222/) (2019)
- [Estimation and Inference of Heterogeneous Treatment Effects using Random Forests](https://mlsanity.com/paper/1510.04342/) (2015)
- [Conformal Prediction Under Covariate Shift](https://mlsanity.com/paper/1904.06019/) (2019)
- [Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learning](https://mlsanity.com/paper/1706.03461/) (2017)
- [Adaptive Conformal Inference Under Distribution Shift](https://mlsanity.com/paper/2106.00170/) (2021)
- [Conformal prediction beyond exchangeability](https://mlsanity.com/paper/2202.13415/) (2022)
- [Learning Representations for Counterfactual Inference](https://mlsanity.com/paper/1605.03661/) (2016)
- [Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processes](https://mlsanity.com/paper/1704.02801/) (2017)
- [Conformal Inference of Counterfactuals and Individual Treatment Effects](https://mlsanity.com/paper/2006.06138/) (2020)
- [Automated versus do-it-yourself methods for causal inference: Lessons learned from a data analysis competition](https://mlsanity.com/paper/1707.02641/) (2017)
- [Discussion on "Bayesian Regression Tree Models for Causal Inference: Regularization, Confounding, and Heterogeneous Effects" by Hahn, Murray and Carvalho](https://mlsanity.com/paper/2108.02836/) (2021)
- [On Inductive Biases for Heterogeneous Treatment Effect Estimation](https://mlsanity.com/paper/2106.03765/) (2021)
- [Sensitivity Analysis of Individual Treatment Effects: A Robust Conformal Inference Approach](https://mlsanity.com/paper/2111.12161/) (2021)
- [Confidence Intervals for Random Forests: The Jackknife and the Infinitesimal Jackknife](https://mlsanity.com/paper/1311.4555/) (2013)
- [Conformal Sensitivity Analysis for Individual Treatment Effects](https://mlsanity.com/paper/2112.03493/) (2021)
- [Conformal Off-Policy Prediction in Contextual Bandits](https://mlsanity.com/paper/2206.04405/) (2022)
- [Conformal Off-policy Prediction](https://mlsanity.com/paper/2206.06711/) (2022)
- [Assessing Treatment Effect Variation in Observational Studies: Results from a Data Challenge](https://mlsanity.com/paper/1907.07592/) (2019)

## Built on by (most cited first, 23 of 23)

- [Causal machine learning for predicting treatment outcomes](https://mlsanity.com/paper/2410.08770/) (2024)
- [Conformal Prediction: A Data Perspective](https://mlsanity.com/paper/2410.06494/) (2024)
- [Conformal Prediction for Causal Effects of Continuous Treatments](https://mlsanity.com/paper/2407.03094/) (2024)
- [Conformal Counterfactual Inference under Hidden Confounding](https://mlsanity.com/paper/2405.12387/) (2024)
- [Quantifying Aleatoric Uncertainty of the Treatment Effect: A Novel Orthogonal Learner](https://mlsanity.com/paper/2411.03387/) (2024)
- [Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event Data](https://mlsanity.com/paper/2505.13072/) (2025)
- [Conformal Convolution and Monte Carlo Meta-learners for Predictive Inference of Individual Treatment Effects](https://mlsanity.com/paper/2402.04906/) (2024)
- [Synthetic Counterfactual Labels for Efficient Conformal Counterfactual Inference](https://mlsanity.com/paper/2509.04112/) (2025)
- [Conformal Diffusion Models for Individual Treatment Effect Estimation and Inference](https://mlsanity.com/paper/2408.01582/) (2024)
- [Cross-World Assumption and Refining Prediction Intervals for Individual Treatment Effects](https://mlsanity.com/paper/2507.12581/) (2025)
- [A Systematic Review of Conformal Inference Procedures for Treatment Effect Estimation: Methods and Challenges](https://mlsanity.com/paper/2509.21660/) (2025)
- [Retrospective Counterfactual Prediction by Conditioning on the Factual Outcome: A Cross-World Approach](https://mlsanity.com/paper/2603.27320/) (2026)
- [Bridging the gap: Towards an Expanded Toolkit for AI-driven Decision-Making in the Public Sector](https://mlsanity.com/paper/2310.19091/) (2023)
- [From Text to Treatment Effects: A Meta-Learning Approach to Handling Text-Based Confounding](https://mlsanity.com/paper/2409.15503/) (2024)
- [Prediction Intervals for Individual Treatment Effects in a Multiple Decision Point Framework using Conformal Inference](https://mlsanity.com/paper/2512.08828/) (2025)
- [Partial Causal Structure Learning for Valid Selective Conformal Inference under Interventions](https://mlsanity.com/paper/2603.02204/) (2026)
- [Debiased Machine Learning for Conformal Prediction of Counterfactual Outcomes Under Runtime Confounding](https://mlsanity.com/paper/2604.03772/) (2026)
- [Bayesian X-Learner: Calibrated Posterior Inference for Heterogeneous Treatment Effects under Heavy-Tailed Outcomes](https://mlsanity.com/paper/2604.27394/) (2026)
- [Causal EpiNets: Precision-corrected Bounds on Individual Treatment Effects using Epistemic Neural Networks](https://mlsanity.com/paper/2605.07065/) (2026)
- [Set-Valued Policy Learning](https://mlsanity.com/paper/2605.19830/) (2026)
- [CausalGuard: Conformal Inference under Graph Uncertainty](https://mlsanity.com/paper/2605.21928/) (2026)
- [Calibrated Inference for the Conditional Average Treatment Effect in the Few-Placebo Regime via Gaussian Processes](https://mlsanity.com/paper/2605.27473/) (2026)
- [Prediction Sets for Counterfactual Decisions: Coverage, Optimality, and Conformal Prediction](https://mlsanity.com/paper/2607.02206/) (2026)

## Abstract (arXiv)

We investigate the problem of machine learning-based (ML) predictive inference on individual treatment effects (ITEs). Previous work has focused primarily on developing ML-based meta-learners that can provide point estimates of the conditional average treatment effect (CATE); these are model-agnostic approaches for combining intermediate nuisance estimates to produce estimates of CATE. In this paper, we develop conformal meta-learners, a general framework for issuing predictive intervals for ITEs by applying the standard conformal prediction (CP) procedure on top of CATE meta-learners. We focus on a broad class of meta-learners based on two-stage pseudo-outcome regression and develop a stochastic ordering framework to study their validity. We show that inference with conformal meta-learners is marginally valid if their (pseudo outcome) conformity scores stochastically dominate oracle conformity scores evaluated on the unobserved ITEs. Additionally, we prove that commonly used CATE meta-learners, such as the doubly-robust learner, satisfy a model- and distribution-free stochastic (or convex) dominance condition, making their conformal inferences valid for practically-relevant levels of target coverage. Whereas existing procedures conduct inference on nuisance parameters (i.e., potential outcomes) via weighted CP, conformal meta-learners enable direct inference on the target parameter (ITE). Numerical experiments show that conformal meta-learners provide valid intervals with competitive efficiency while retaining the favorable point estimation properties of CATE meta-learners.