arXiv:2502.02223q-bio.QMcs.LG2025-02被引 1

统一生存分析工具的Python框架,让机器学习新手也能轻松建模时间事件数据。

SurvHive: a package to consistently access multiple survival-analysis packages

  • 基于scikit-learn风格接口,整合经典统计与深度学习生存模型。
  • 支持时变风险评估、超参调优和右删失数据交叉验证。
  • 适合生物医学、医疗数据分析等需要处理删失数据的研究者。

生存分析是建模时间至事件数据的基础工具,近年来与机器学习方法结合以应对删失数据和时变风险的复杂性。尽管如此,使用最先进生存模型仍面临挑战,因现有实现分散且缺乏标准化接口,需大量预处理。我们提出SurvHive,一个基于Python的框架,旨在通过类scikit-learn的统一可扩展接口整合生存分析方法。SurvHive融合经典统计模型与前沿深度学习方法,包括基于Transformer的架构和参数化生存模型。通过一致API,简化模型训练、评估与优化,显著降低机器学习从业者进入生存分析的门槛。该包包含增强的超参数调优支持、时变风险评估指标及针对删失数据设计的交叉验证策略。凭借可扩展性与易用性,SurvHive架起了生存分析与更广泛机器学习社区之间的桥梁,推动跨领域时间至事件建模的发展。代码与文档已开源:https://github.com/compbiomed-unito/survhive。

原文摘要 · Abstract (English)

Survival analysis, a foundational tool for modeling time-to-event data, has seen growing integration with machine learning (ML) approaches to handle the complexities of censored data and time-varying risks. Despite these advances, leveraging state-of-the-art survival models remains a challenge due to the fragmented nature of existing implementations, which lack standardized interfaces and require extensive preprocessing. We introduce SurvHive, a Python-based framework designed to unify survival analysis methods within a coherent and extensible interface modeled on scikit-learn. SurvHive integrates classical statistical models with cutting-edge deep learning approaches, including transformer-based architectures and parametric survival models. Using a consistent API, SurvHive simplifies model training, evaluation, and optimization, significantly reducing the barrier to entry for ML practitioners exploring survival analysis. The package includes enhanced support for hyper-parameter tuning, time-dependent risk evaluation metrics, and cross-validation strategies tailored to censored data. With its extensibility and focus on usability, SurvHive provides a bridge between survival analysis and the broader ML community, facilitating advancements in time-to-event modeling across domains. The SurvHive code and documentation are available freely at https://github.com/compbiomed-unito/survhive.

生存分析机器学习Python工具删失数据

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。