Causal Outcome Learning with Noisy Proxies

원문 초록

Modern observational biomedical datasets contain increasingly extensive baseline and outcome measurements, yet detecting a clinically meaningful treatment effect remains difficult because investigators must identify an outcome that captures treatment response while adjusting for confounding distributed across numerous noisy baseline variables. We introduce SCOUT (Score-based Causal Outcome Learning Using Treatment signals), a framework that learns a clinically interpretable, treatment-responsive outcome while adjusting for confounding throughout the learning process. SCOUT constructs low-dimensional adjustment representations of high-dimensional baseline measurements to estimate the average treatment effect (ATE) and conditional average treatment effect (CATE) for the learned outcome. Importantly, the final adjustment space is at most two-dimensional, allowing treatment-effect heterogeneity to be estimated in this space rather than in the full high-dimensional covariate space. SCOUT also introduces a fourth-order statistic for testing treatment-effect heterogeneity that emphasizes unusually large deviations from the ATE. In synthetic and semi-synthetic experiments, the algorithm achieved competitive ATE estimation accuracy and substantially lower CATE estimation error. In an observational study of patients with bipolar disorder, SCOUT identified a clinically recognizable lithium-associated adverse-effect profile, detecting both a significant ATE (Holm-adjusted $p$ = 0.0026) and significant treatment-effect heterogeneity (Holm-adjusted $p$ = 0.033).These results demonstrate how jointly learning outcomes and controlling for confounding can facilitate the discovery of treatment effects in high-dimensional observational data.