Barnatchez et al. propose debiased ML for robust causal AI under missing data

A team of researchers including Keith Barnatchez, Kevin P. Josey, Rachel C. Nethery, and Giovanni Parmigiani has published a significant paper tackling a major hurdle in deploying causal AI models. The work, 'Debiased Machine Learning for Conformal Prediction of Counterfactual Outcomes Under Runtime Confounding,' addresses the common real-world problem where models trained on rich source data must be applied to target populations with incomplete measurements. Specifically, it solves 'runtime confounding'—the scenario where not all variables that influence both a treatment and an outcome are available at deployment time, which previously risked invalid and misleading prediction intervals.

Their proposed framework combines debiased machine learning (DML) with conformal prediction, a popular method for creating statistically rigorous prediction intervals. By leveraging semiparametric efficiency theory, the method produces intervals that maintain correct coverage rates (e.g., 95% confidence) even when a subset of confounders is unmeasured in the new data. The authors demonstrate through synthetic and semi-synthetic experiments that their approach not only ensures validity but also achieves faster statistical convergence compared to standard methods. This represents a crucial step toward more robust and trustworthy AI systems for decision-making in fields like healthcare and policy, where 'what-if' scenarios must be assessed with reliable uncertainty estimates despite imperfect data.