DR-Learners for Generalizing CATEs from a Randomized Controlled Trial

Abstract

Randomized clinical trials are essential for evaluating interventions, yet average treatment effects can obscure treatment effect heterogeneity, making conditional average treatment effects (CATEs) a natural target. However, restrictive eligibility criteria and differential trial participation can limit the generalizability of these effects to broader target populations. Building on existing DR-Learners, we develop the gDR-Learner for generalizing RCT CATEs to a target population, using cross-fitted machine-learning nuisance estimation. We derive excess-risk bounds in which nuisance errors enter only as products, showing that the estimator achieves oracle-rate convergence despite slower individual nuisance estimation. The variables needed to generalize are rarely measured across the entire target population. We therefore extend the methods to two settings with incomplete covariate information: when certain covariates are measured only in the trial (e.g., biomarkers), and covariates driving trial participation measured only in a subsample of the target population, such as a representative cohort or registry. We construct weighted conformal prediction intervals for individual treatment effects, accounting for covariate shift between the trial and target population. In simulations, the gDR-Learner achieved the lowest estimation error when all covariates were observed relative to standard T- and DR-Learners and an inverse-probability-weighted T-Learner, and also outperformed the latter under incomplete covariate observation. We illustrate the methods by reanalyzing the Coronary Artery Surgery Study, examining treatment effect heterogeneity by prior myocardial infarction and baseline ejection fraction.