Debiased Machine Learning for Conformal Prediction of Counterfactual Outcomes Under Runtime Confounding
Authors: Keith Barnatchez, Kevin P. Josey, Rachel C. Nethery, Giovanni Parmigiani
Organizations: Department of Biostatistics, Harvard T.H. Chan School of Public Health · Department of Biostatistics and Informatics, Colorado School of Public Health · Department of Data Science, Dana Farber Cancer Institute
Data-driven decision making frequently relies on predicting counterfactual outcomes. In practice, researchers commonly train counterfactual prediction models on a source dataset to inform decisions on a possibly separate target population. Conformal prediction has arisen as a popular method for producing assumption-lean prediction intervals for counterfactual outcomes that would arise under different treatment decisions in the target population of interest. However, existing methods require that every confounding factor of the treatment-outcome relationship used for training on the source data is additionally measured in the target population, risking miscoverage if important confounders are unmeasured in the target population. In this paper, we introduce a computationally efficient debiased machine learning framework that allows for valid prediction intervals when only a subset of confounders is measured in the target population, a common challenge referred to as runtime confounding. Grounded in semiparametric efficiency theory, we show the resulting prediction intervals achieve desired coverage rates with faster convergence compared to standard methods. Through numerous synthetic and semi-synthetic experiments, we demonstrate the utility of our proposed method.
Figures & tables
Y
A
V
U
S
Y1
A1
V1
U1
1
⋮
⋮
⋮
⋮
⋮
Yn1
An1
Vn1
Un1
1
NA
NA
Vn1+1
NA
0
⋮
⋮
⋮
⋮
⋮
NA
NA
Vn1+n0
NA
0
Table 1: Observed data structure.
Figure 1: Two possible directed acyclic graphs consistent with Assumptions 3 and 4 . Runtime confounding is induced when U is unobserved when S=0 .
Figure 2: Experiment results, varying n and fixing number of runtime confounders at 10.
Figure 3: Performance of proposed methods on semi-synthetic ACIC data.
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 4: Performance of proposed methods stratified by counterfactual outcome.
Figure 5: Performance of proposed methods, varying n and fixing the number of runtime confounders at 5.
Figure 6: Performance of proposed methods, varying n and fixing the number of runtime confounders at 15.
Figure 7: Performance of proposed methods when varying P(S=1) , fixing n=5000 and the number of runtime confounders at 10.
Figure 8: Performance of proposed methods on semi-synthetic ACIC data, varying n under the baseline moderate runtime confounding scenario.
Figure 9: Performance of proposed methods on semi-synthetic ACIC data under the mild runtime confounding scenario.
Figure 10: Performance of proposed methods on semi-synthetic ACIC data under the severe runtime confounding scenario.
Figure 11: Two possible directed acyclic graphs consistent where Assumption 3 is satisfied but 4 is violated.
Online conformal prediction can fail when predictions shape actions and actions determine which outcomes enter calibration. Standard adaptive methods may retain marginal coverage while systematically miscovering the counterfactual outcomes of rarely selected actions. This paper formalizes the failure through counterfactual coverage and introduces Propensity-Weighted Online Conformal Prediction, an inverse-propensity-weighted recursion that debiases calibration. A doubly robust variant further reduces nuisance bias to the product of outcome-model and propensity errors. Under positivity, the resulting coverage rate matches an information-theoretic lower bound up to logarithmic factors. Experiments on synthetic decision tasks, open bandit data, and financial rebalancing show that PW-OCP and DR-OCP improve counterfactual coverage and downstream regret without sacrificing prediction-set sharpness.
Xinyu Qiao, Yichen Lin, Kaihong Ji +2
Shanghai Jiao Tong University, Shanghai, China · Alibaba Group
Conformal counterfactual prediction constructs prediction sets with finite-sample coverage guarantees for counterfactual outcomes and individual treatment effects under the no-interference assumption. In this work, we relax this assumption by allowing each unit's potential outcomes to depend on other units' treatments and covariates. In this setting, propensity-score reweighting does not restore weighted exchangeability, and existing methods may fail to achieve valid coverage. To address this issue, we develop interference-adjusted weighted conformal prediction that accounts for interference by constructing an observable upper bound on the ideal and unobserved conformal p-value under the target intervention. The resulting prediction sets provide finite-sample marginal coverage guarantees for counterfactual outcomes and individual treatment effects in both transductive and inductive settings. We also derive a sharper construction when intervention-induced changes in nonconformity scores are bounded. Numerical experiments show that our methods preserve nominal coverage, whereas existing methods may not.
Matteo Zecchin, Osvaldo Simeone
Communication Systems Department, EURECOM, Sophia Antipolis, France · Institute for Intelligent Networked Systems (INSI), Northeastern University London, London, UK
Predictions are increasingly used to guide high-stakes decisions, from treatment selection to policy making. To ensure reliability with imperfect predictions, uncertainty quantification methods such as conformal prediction build prediction sets with coverage guarantees. However, statistical validity alone does not immediately determine the decisions to take, nor the optimality thereof. This gap is especially delicate in counterfactual settings where the outcome that materializes depends on the action taken, so uncertainty cannot be specified independently of the decision rule. We develop a decision-theoretic framework for uncertainty-informed counterfactual decisions. We identify a novel notion of policy-coupled coverage -- namely, coverage of the realized outcome under the action induced by the prediction sets themselves -- as the optimal and lossless interface between uncertainty and action. It plays three roles. First, it justifies acting via a natural max-min rule as minimax-optimal under distributional ambiguity. Second, optimizing prediction sets under policy-coupled coverage is equivalent both to a stronger universal-coverage formulation and to the direct risk-averse optimization over policies and utility certificates; this equivalence yields the explicit form of the population-optimal prediction sets. Third, it admits a two-stage procedure, Policy-Coupled Risk-Averse Conformal Prediction (PC-RACP), that approximates these optimal sets with rigorous finite-sample coverage. Simulations and a real email-marketing experiment confirm that PC-RACP delivers higher utility than existing approaches while maintaining valid coverage, and that ignoring the counterfactual structure of the decision problem is suboptimal for both validity and utility.
Yurui Zheng, Ying Jin
School of Mathematical Sciences, Peking University · Department of Statistics and Data Science, University of Pennsylvania