Source projection for calibrated transfer learning in survival prediction
Abstract
Transfer learning can improve survival prediction in cohorts with few events. An external model may capture useful covariate associations yet systematically overestimate or underestimate risk in the target population, creating a need for selective borrowing. We propose Source Projection Transfer (SPT) for fixed-horizon prediction with a target generalized additive model. SPT requires only source predictions at target covariates, supporting generalized additive, Cox and neural-network source models without individual-level source data. It projects source log-odds scores onto the target basis and borrows through two separately adapted channels: a projection-centered penalty transfers effect shapes, while intercept-recalibrated soft labels inform probabilities at the target risk level. Under local asymptotics, projection avoids the strictly positive risk cost of estimating an uninformative additional source direction. Each channel asymptotically switches off under its corresponding persistent-mismatch condition. In log-logistic and Weibull simulations with covariate shift, SPT improves mean Brier score and discrimination over target-only modeling in all 24 setting-distribution-horizon combinations. Matched comparisons show improvements over borrowing through source probabilities alone across settings, and over borrowing the complete source score under structural mismatch. In UK Biobank-to-SPRINT mortality prediction among standard-treatment participants with reduced kidney function, SPT improves mean discrimination at 24, 30 and 36 months, with Brier improvements at 24 and 36 months. SPT enables flexible source learners to inform a parsimonious target model while adapting to differences between populations.
Related
- Calibrated Transfer Learning for Survival Prediction and Treatment Inference
- A novel Weighted Exponential-Lindley distribution with Bayesian estimation under progressive Type-II censoring and a Bayesian-Calibrated Variational Autoencoder for survival prediction.
- RECaST-Surv: A Calibrated Borrowing Method for Survival Endpoints in Unequal Randomized Trials
- A Cosine-Projection Framework for Choosing Among Cox HR, D-RMST, and Average-Hazard Tests in Survival Trials
- Covariance-Calibrated Prediction-Powered Inference
BibTeX
@misc{cao2026source,
title = {{Source projection for calibrated transfer learning in survival prediction}},
author = {Fangzhou Cao and Ting Tian},
howpublished = {arXiv},
year = {2026},
month = oct,
doi = {10.48550/arxiv.2610.41230},
eprint = {2610.41230},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2610.41230},
}
RIS
TY - UNPB
TI - Source projection for calibrated transfer learning in survival prediction
AU - Fangzhou Cao
AU - Ting Tian
PY - 2026
DA - 2026/10/30
PB - arXiv
DO - 10.48550/arxiv.2610.41230
UR - https://arxiv.org/abs/2610.41230
AB - Transfer learning can improve survival prediction in cohorts with few events. An external model may capture useful covariate associations yet systematically overestimate or underestimate risk in the target population, creating a need for selective borrowing. We propose Source Projection Transfer (SPT) for fixed-horizon prediction with a target generalized additive model. SPT requires only source predictions at target covariates, supporting generalized additive, Cox and neural-network source models without individual-level source data. It projects source log-odds scores onto the target basis and borrows through two separately adapted channels: a projection-centered penalty transfers effect shapes, while intercept-recalibrated soft labels inform probabilities at the target risk level. Under local asymptotics, projection avoids the strictly positive risk cost of estimating an uninformative additional source direction. Each channel asymptotically switches off under its corresponding persistent-mismatch condition. In log-logistic and Weibull simulations with covariate shift, SPT improves mean Brier score and discrimination over target-only modeling in all 24 setting-distribution-horizon combinations. Matched comparisons show improvements over borrowing through source probabilities alone across settings, and over borrowing the complete source score under structural mismatch. In UK Biobank-to-SPRINT mortality prediction among standard-treatment participants with reduced kidney function, SPT improves mean discrimination at 24, 30 and 36 months, with Brier improvements at 24 and 36 months. SPT enables flexible source learners to inform a parsimonious target model while adapting to differences between populations.
ER -