stat.APMay 22, 2026

Distributionally Robust Transfer Learning with Structurally Missing Covariates, with Application to Cross-National Cardiac Arrest Prediction

Authors: Siqi LiChuan HongZiye TianBenjamin Sieu-Hon LeongKoshi NakagawaHideharu TanakaSang Do ShinKhuong Quoc Dai+4 more

Organizations: Centre for Biomedical Data Science, Duke-NUS Medical School, Singapore · 2Duke-NUS AI + Medical Sciences Initiative, Duke-NUS Medical School, Singapore · Department of Biostatistics and Bioinformatics, Duke University, Durham, NC, USA · 4Duke Clinical Research Institute, Durham, NC, USA · 5Emergency Medicine Department, National University Hospital, Singapore · Department of Sport and Medical Science, Faculty of Physical Education, Kokushikan University, Tokyo, Japan · 7Graduate School of Emergency Medical System, Kokushikan University, Tokyo, Japan · Department of Emergency Medicine, Seoul National University College of Medicine, Seoul, Republic of Korea · Center for Emergency Medicine, Bach Mai Hospital, Hanoi, Vietnam · Center for Critical Care Medicine, Bach Mai Hospital, Hanoi, Vietnam · 11Health Services Research Centre, Singapore Health Services, Singapore · Department of Emergency Medicine, Singapore General Hospital, Singapore · 13Pre-hospital & Emergency Research Centre, Health Services Research and Population Health, Duke-NUS Medical School, Singapore · 14NUS Artificial Intelligence Institute, National University of Singapore, Singapore · Department of Biostatistics, Peking University Health Science Center, Peking University, Beijing, China · 16Beijing International Center for Mathematical Research, Peking University, Beijing, China

Abstract

Deploying clinical prediction models across healthcare systems often fails when key training covariates are unavailable at deployment and labeled outcomes are limited in the target domain. For example, high-performing models for out-of-hospital cardiac arrest (OHCA) rely on detailed prehospital measurements routinely collected in high-resource settings but unavailable in many international registries. Existing methods either discard missing covariates, sacrificing predictive information, or rely on untestable assumptions about their target distribution. We propose DRUM (\underline{D}istributionally \underline{R}obust \underline{U}nsupervised transfer learning with structurally \underline{M}issing covariates), a framework that transfers prediction models to target populations where certain covariates are structurally absent and outcome labels are unavailable. DRUM partitions covariates into shared components (XX), observed across all settings, and missing components (AA), observed only in the source. Rather than imputing missing covariates, DRUM optimizes worst-case predictive performance over the unknown target distribution of AXA \mid X using a neural network generator, with a robustness parameter controlling allowable deviation from the source conditional. We further develop a bias correction procedure that reduces sensitivity to nuisance estimation error. Simulations show substantial improvements in both mean and worst-case prediction error under distribution shift. Applied to cross-national OHCA prediction, transferring models from a US registry to multiple Asian registries where prehospital variables are unrecorded, DRUM yields better-calibrated predictions and improved clinical classification performance across sites.

Explore similar work

CardsList