Generalizability of AI Survival Models in the Context of Preterm Birth Prediction
摘要
This study investigates whether integrating a Variational Information Bottleneck (VIB) into deep survival models improves generalization across heterogeneous populations. We evaluate performance on preterm birth prediction using three distinct datasets, training on one and testing on the others. Compared to classical and non-VIB deep models, the VIB-enhanced models achieved higher time-dependent AUC in several cases, suggesting potential benefits despite limited sample sizes.