TorchSurv: A Lightweight Package for Deep Survival Analysis

  • 2024-04-16 18:41:17
  • Melodie Monod, Peter Krusche, Qian Cao, Berkman Sahiner, Nicholas Petrick, David Ohlssen, Thibaud Coroller
  • 0

Abstract

TorchSurv is a Python package that serves as a companion tool to perform deepsurvival modeling within the PyTorch environment. Unlike existing librariesthat impose specific parametric forms, TorchSurv enables the use of customPyTorch-based deep survival mod- els. With its lightweight design, minimalinput requirements, full PyTorch backend, and freedom from restrictive survivalmodel parameterizations, TorchSurv facilitates efficient deep survival modelimplementation and is particularly beneficial for high-dimensional and complexinput data scenarios

 

Quick Read (beta)

loading the full paper ...