mlsanityData to 14 Oct 2026

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

arXiv 2605.24212 ↗First posted 22 May 2026stat.AP cs.AI cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

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 (underlineDistributionally underlineRobust underlineUnsupervised transfer learning with structurally underlineMissing 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 (X), observed across all settings, and missing components (A), observed only in the source. Rather than imputing missing covariates, DRUM optimizes worst-case predictive performance over the unknown target distribution of A 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.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 13 of its 40 references that are in the map. First posted on arXiv 22 May 2026.

Builds on

13 of 40 references.

References · 13

  1. 2015Domain-Adversarial Training of Neural Networkscited by 3,044
  2. 2015Learning Transferable Features with Deep Adaptation Networkscited by 1,329
  3. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
  4. 2017CyCADA: Cycle-Consistent Adversarial Domain Adaptationcited by 1,082
  5. 2018Fairness Without Demographics in Repeated Loss Minimizationcited by 364
  6. 2018Learning Models with Uniform Performance via Distributionally Robust Optimizationcited by 335
  7. 2015Data-driven Distributionally Robust Optimization Using the Wasserstein Metric: Performance Guarantees and Tractable Reformulationscited by 291
  8. 2023Engression: Extrapolation through the Lens of Distributional Regressioncited by 51
  9. 2019Integrating multi-source block-wise missing data in model selectioncited by 11
  10. 2023Federated Learning for Clinical Structured Data: A Benchmark Comparison of Engineering and Statistical Approachescited by 8
  11. 2024Bridging Data Gaps in Healthcare: A Scoping Review of Transfer Learning in Biomedical Data Analysiscited by 6
  12. 2022Modular Regression: Improving Linear Models by Incorporating Auxiliary Datacited by 5
1 more
  1. 2025Robust Mixture Models for Algorithmic Fairness Under Latent Heterogeneitycited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.