TorchSurv: A Lightweight Package for Deep Survival Analysis

TorchSurv是一个Python包，作为PyTorch环境中进行深度生存建模的伴侣工具。与现有的库不同，TorchSurv允许使用基于PyTorch的自定义深度生存模型，而不是强制使用特定的参数形式。TorchSurv具有轻量级设计、最小输入要求、完整的PyTorch后端和不受限制的生存模型参数化自由，有助于高维度和复杂输入数据场景下高效地实现深度生存模型。

TorchSurv试图解决什么问题？

TorchSurv是一个轻量级的Python包，旨在在PyTorch环境中进行深度生存建模，与现有的库不同，它不强加特定的参数形式，而是允许使用基于PyTorch的自定义深度生存模型。

TorchSurv的设计轻量级，输入要求最少，完全使用PyTorch后端，并且不受限制的生存模型参数化，有助于高维和复杂输入数据场景下高效实现深度生存模型。实验使用了两个公共数据集，并开源了代码。

最近的相关研究包括DeepSurv、SurvivalNet、Cox-nnet和Cox-Time等。