Counterfactual Predictions under Runtime Confounding
Amanda Coston, Edward H. Kennedy, Alexandra Chouldechova
Abstract
Algorithms are commonly used to predict outcomes under a particular decision or intervention, such as predicting whether an offender will succeed on parole if placed under minimal supervision. Generally, to learn such counterfactual prediction models from observational data on historical decisions and corresponding outcomes, one must measure all factors that jointly affect the outcomes and the decision taken. Motivated by decision support applications, we study the counterfactual prediction task in the setting where all relevant factors are captured in the historical data, but it is either undesirable or impermissible to use some such factors in the prediction model. We refer to this setting as runtime confounding. We propose a doubly-robust procedure for learning counterfactual prediction models in this setting. Our theoretical analysis and experimental results suggest that our method often outperforms competing approaches. We also present a validation procedure for evaluating the performance of counterfactual prediction methods.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers11
- On Inductive Biases for Heterogeneous Treatment Effect EstimationAlicia Curth, Mihaela van der SchaarNeurIPS 2021 · 114 citations
- Bounds on Representation-Induced Confounding Bias for Treatment Effect EstimationValentyn Melnychuk, Dennis Frauen, Stefan FeuerriegelICLR 2024 · 23 citations
- Orthogonal Survival Learners for Estimating Heterogeneous Treatment Effects from Time-to-Event DataDennis Frauen, Maresa Schröder, Konstantin Hess, Stefan FeuerriegelNeurIPS 2025 · 13 citations
- Fair and Robust Estimation of Heterogeneous Treatment Effects for Policy LearningKwangho Kim, José R. ZubizarretaICML 2023 · 12 citations
- Doubly Robust Counterfactual ClassificationKwangho Kim, Edward H. Kennedy, José R. ZubizarretaNeurIPS 2022 · 9 citations
Builds on1
Related papers
- Towards Robust Off-Policy Learning for Runtime UncertaintyDa Xu, Yuting Ye, Chuanwei Ruan, Bo YangAAAI 2022 · 5 citations
- Off-policy Policy Evaluation For Sequential Decisions Under Unobserved ConfoundingHongseok Namkoong, Ramtin Keramati, Steve Yadlowsky, Emma BrunskillNeurIPS 2020 · 81 citations
- Model-Free and Model-Based Policy Evaluation when Causality is UncertainDavid Bruns-SmithICML 2021 · 14 citations
- Confounding-Robust Policy Evaluation in Infinite-Horizon Reinforcement LearningNathan Kallus, Angela ZhouNeurIPS 2020 · 78 citations
- What's the Harm? Sharp Bounds on the Fraction Negatively Affected by TreatmentNathan KallusNeurIPS 2022 · 40 citations
