Abstract:Motivation: Sparse survival models are statistical models that select a subset of predictor variables while modeling the time until an event occurs, which can subsequently help interpretability and transportability. The subset of important features is often obtained with regularized models, such as the Cox Proportional Hazards model with Lasso regularization, which limit the number of non-zero coefficients. However, such models can be sensitive to the choice of regularization hyperparameter. Results: In this work, we develop a software package and demonstrate how knowledge distillation, a powerful technique in machine learning that aims to transfer knowledge from a complex teacher model to a simpler student model, can be leveraged to learn sparse survival models while mitigating this challenge. For this purpose, we present sparsesurv, a Python package that contains a set of teacher-student model pairs, including the semi-parametric accelerated failure time and the extended hazards models as teachers, which currently do not have Python implementations. It also contains in-house survival function estimators, removing the need for external packages. Sparsesurv is validated against R-based Elastic Net regularized linear Cox proportional hazards models as implemented in the commonly used glmnet package. Our results reveal that knowledge distillation-based approaches achieve competitive discriminative performance relative to glmnet across the regularization path while making the choice of the regularization hyperparameter significantly easier. All of these features, combined with an sklearn-like API, make sparsesurv an easy-to-use Python package that enables survival analysis for high-dimensional datasets through fitting sparse survival models via knowledge distillation. Availability: sparsesurv is freely available under a BSD 3 license on GitHub (https://github.com/BoevaLab/sparsesurv) and The Python Package Index (PyPi) (https://pypi.org/project/sparsesurv/). Supplementary information: Supplementary data are available at Bioinformatics online.

mlr3proba: an R package for machine learning in survival analysis

Unbiased Prediction and Feature Selection in High-Dimensional Survival Regression

Bayesian Survival Analysis Using the rstanarm R Package

Multi-Parameter Regression Survival Modelling: An Alternative to Proportional Hazards

CaseCohortCoxSurvival: an R Package for Case-Cohort Inference for Relative Hazard and Pure Risk under the Cox Model

Interpretable Machine Learning for Survival Analysis

Parmsurv: a SAS Macro for Flexible Parametric Survival Analysis with Long-Term Predictions

pencal: an R Package for the Dynamic Prediction of Survival with Many Longitudinal Predictors

Subgroup identification in clinical trials: an overview of available methods and their implementations with R

spBayesSurv: Fitting Bayesian Spatial Survival Models Using R

survextrap: a package for flexible and transparent survival extrapolation

A review on statistical and machine learning competing risks methods

mlr3summary: Concise and interpretable summaries for machine learning models

sparsesurv: A Python package for fitting sparse survival models via knowledge distillation

ggRandomForests: Exploring Random Forest Survival

Survival Analysis Using a 5-Step Stratified Testing and Amalgamation Routine in Randomized Clinical Trials

Smooth Transformation Models for Survival Analysis: A Tutorial Using R

Web-Based Survival Analysis Tool Tailored for Medical Research (KMplot): Development and Implementation

OpenML: An R Package to Connect to the Machine Learning Platform OpenML

CPSM: R-package of an Automated Machine Learning Pipeline for Predicting the Survival Probability of Single Cancer Patient

Avoiding C-hacking when evaluating survival distribution predictions with discrimination measures