Saved in:
Bibliographic Details
Main Authors: Portela, Gerard T., Gibbons, Jason B., Schneeweiss, Sebastian, Desai, Rishi J.
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2510.26700
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917050603012096
author Portela, Gerard T.
Gibbons, Jason B.
Schneeweiss, Sebastian
Desai, Rishi J.
author_facet Portela, Gerard T.
Gibbons, Jason B.
Schneeweiss, Sebastian
Desai, Rishi J.
contents Observational studies developing causal machine learning (ML) models for the prediction of individualized treatment effects (ITEs) seldom conduct empirical evaluations to assess the conditional exchangeability assumption. We aimed to evaluate the performance of these models under conditional exchangeability violations and the utility of negative control outcomes (NCOs) as a diagnostic. We conducted a simulation study to examine confounding bias in ITE estimates generated by causal forest and X-learner models under varying conditions, including the presence or absence of true heterogeneity. We simulated data to reflect real-world scenarios with differing levels of confounding, sample size, and NCO confounding structures. We then estimated and compared subgroup-level treatment effects on the primary outcome and NCOs across settings with and without unmeasured confounding. When conditional exchangeability was violated, causal forest and X-learner models failed to recover true treatment effect heterogeneity and, in some cases, falsely indicated heterogeneity when there was none. NCOs successfully identified subgroups affected by unmeasured confounding. Even when NCOs did not perfectly satisfy its ideal assumptions, it remained informative, flagging potential bias in subgroup level estimates, though not always pinpointing the subgroup with the largest confounding. Violations of conditional exchangeability substantially limit the validity of ITE estimates from causal ML models in routinely collected observational data. NCOs serve a useful empirical diagnostic tool for detecting subgroup-specific unmeasured confounding and should be incorporated into causal ML workflows to support the credibility of individualized inference.
format Preprint
id arxiv_https___arxiv_org_abs_2510_26700
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Assessment of the conditional exchangeability assumption in causal machine learning models: a simulation study
Portela, Gerard T.
Gibbons, Jason B.
Schneeweiss, Sebastian
Desai, Rishi J.
Machine Learning
Observational studies developing causal machine learning (ML) models for the prediction of individualized treatment effects (ITEs) seldom conduct empirical evaluations to assess the conditional exchangeability assumption. We aimed to evaluate the performance of these models under conditional exchangeability violations and the utility of negative control outcomes (NCOs) as a diagnostic. We conducted a simulation study to examine confounding bias in ITE estimates generated by causal forest and X-learner models under varying conditions, including the presence or absence of true heterogeneity. We simulated data to reflect real-world scenarios with differing levels of confounding, sample size, and NCO confounding structures. We then estimated and compared subgroup-level treatment effects on the primary outcome and NCOs across settings with and without unmeasured confounding. When conditional exchangeability was violated, causal forest and X-learner models failed to recover true treatment effect heterogeneity and, in some cases, falsely indicated heterogeneity when there was none. NCOs successfully identified subgroups affected by unmeasured confounding. Even when NCOs did not perfectly satisfy its ideal assumptions, it remained informative, flagging potential bias in subgroup level estimates, though not always pinpointing the subgroup with the largest confounding. Violations of conditional exchangeability substantially limit the validity of ITE estimates from causal ML models in routinely collected observational data. NCOs serve a useful empirical diagnostic tool for detecting subgroup-specific unmeasured confounding and should be incorporated into causal ML workflows to support the credibility of individualized inference.
title Assessment of the conditional exchangeability assumption in causal machine learning models: a simulation study
topic Machine Learning
url https://arxiv.org/abs/2510.26700