SuperSurv: A Unified Framework for Machine Learning Ensembles in Survival Analysis

Journal: bioRxiv
Published Date:

Abstract

This paper introduces SuperSurv, a user-friendly R package for building, evaluating, and interpreting ensemble models for right-censored survival data. Although many survival modeling methods are available, existing tools are often model-specific and lack a unified platform for systematically integrating, comparing, and ensembling heterogeneous learners. SuperSurv addresses this gap by providing a unified interface for diverse survival learners, including models that return full survival curves as well as methods that produce only risk scores. All learner outputs are mapped to calibrated survival probability curves on a common evaluation time grid, enabling direct comparison and ensemble construction across heterogeneous model classes. SuperSurv implements stacking of survival models using inverse-probability-of-censoring weighted (IPCW) Brier risk to estimate ensemble weights in the presence of right censoring. The framework integrates hyperparameter tuning, time-dependent benchmarking metrics, and visualization tools for survival model evaluation. In addition, the package provides post-hoc interpretability utilities based on SHAP (SHapley Additive exPlanations) values and supports covariate-adjusted restricted mean survival time (RMST) contrasts through g-computation. By bridging the gap between theoretical rigor and clinical application, SuperSurv offers researchers a comprehensive ecosystem for modern survival analysis. The SuperSurv package is open-source and available on CRAN at https://CRAN.R-project.org/package=SuperSurv and on GitHub at https://github.com/yuelyu21/SuperSurv. An empirical example using the METABRIC breast cancer dataset demonstrates a complete workflow from model training and benchmarking to explainability and clinically interpretable survival contrasts.

Authors

  • Lyu
  • Y.; Lin
  • S. H.; Huang
  • X.; Li
  • Z.

Categories