arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.04098stat.MLcs.LG
收藏

低维数据上生存模型大规模中性比较研究

A Large-Scale Neutral Comparison Study of Survival Models on Low-Dimensional Data

Department of Statistics, LMU Munich(慕尼黑大学统计学系);Munich Center for Machine Learning(慕尼黑机器学习中心);Leibniz Institute for Prevention Research and Epidemiology – BIPS(预防研究与流行病学研究所 – BIPS);Faculty of Mathematics & Computer Science, University of Bremen(不莱梅大学数学与计算机科学学院);Department of Cancer Genetics, Institute for Cancer Research, Oslo University Hospital(癌症遗传学部,癌症研究所,奥斯陆大学医院);Department of Biostatistics, Oslo Centre for Biostatistics and Epidemiology (OCBE), University of Oslo (UiO)(生物统计学部,奥斯陆生物统计与流行病学中心(OCBE),奥斯陆大学);OSPO Now, London, UK(OSPO Now,伦敦,英国)

Lukas Burk, John Zobolas, Bernd Bischl, Andreas Bender, Marvin N. Wright, Raphael Sonabend

AI总结:

本研究对低维生存数据中的19种生存模型进行大规模中性比较,发现Cox模型在多数指标上表现稳健,仍为实用选择。

摘要:

本文首次提出了一个大规模中性基准实验,专注于单事件、右删失、低维生存数据。基准实验在方法学研究中至关重要,通过适当的实证评估科学地比较新旧模型类别。现有生存文献中的基准实验在所用数据集数量和实证评估范围上较小,往往缺乏适当的调优或评估程序,而其他比较研究则侧重于定性评论而非定量比较。本综合性研究旨在通过中性评估广泛的方法并提供可推广的指南来填补这一空白。我们对34个公开可用的数据集上基准测试了19种模型,从经典统计方法到许多常见机器学习方法。基准测试使用了判别度量(Harrell的C指数)和评分规则(整合生存Brier分数)来调优模型,并在涵盖判别、校准和总体预测性能的六个指标上评估它们。尽管从个体学习者如斜向随机生存森林和基于似然的提升方法在总体预测性能上平均排名更优,以及多个提升和树方法以及参数生存模型在判别排名上更优,但没有方法在任一调优度量上显著优于常用的Cox比例风险模型。我们得出结论,在低维、右删失数据的标准生存分析设置中,Cox比例风险模型仍然是简单且稳健的方法,对于大多数从业者来说是足够的。所有代码、数据和结果均在GitHub上公开可用:https://github.com/slds-lmu/paper_2023_survival_benchmark

英文摘要:

This work presents the first large-scale neutral benchmark experiment focused on single-event, right-censored, low-dimensional survival data. Benchmark experiments are essential in methodological research to scientifically compare new and existing model classes through proper empirical evaluation. Existing benchmarks in the survival literature are smaller in scale regarding the number of used datasets and extent of empirical evaluation. They often lack appropriate tuning or evaluation procedures, while other comparison studies focus on qualitative reviews rather than quantitative comparisons. This comprehensive study aims to fill the gap by neutrally evaluating a broad range of methods and providing generalizable guidelines for practitioners. We benchmark 19 models, ranging from classical statistical approaches to many common machine learning methods, on 34 publicly available datasets. The benchmark tunes models using both a discrimination measure (Harrell's C-index) and a scoring rule (Integrated Survival Brier Score), and evaluates them across six metrics covering discrimination, calibration, and overall predictive performance. Despite superior average ranks in overall predictive performance from individual learners like oblique random survival forests and likelihood-based boosting, and better discrimination rankings from multiple boosting- and tree-based methods as well as parametric survival models, no method significantly outperforms the commonly used Cox proportional hazards model for either tuning measure. We conclude that for predictive purposes in the standard survival analysis setting of low-dimensional, right-censored data, the Cox Proportional Hazards model remains a simple and robust method, sufficient for most practitioners. All code, data, and results are publicly available on GitHub https://github.com/slds-lmu/paper_2023_survival_benchmark

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑