Suppr超能文献

一种评估生存模型的有效且有意义的方法。

An Effective Meaningful Way to Evaluate Survival Models.

作者信息

Qi Shi-Ang, Kumar Neeraj, Farrokh Mahtab, Sun Weijie, Kuan Li-Hao, Ranganath Rajesh, Henao Ricardo, Greiner Russell

机构信息

Computing Science, University of Alberta, Edmonton, Canada.

Alberta Machine Intelligence Institute, Edmonton, Canada.

出版信息

Proc Mach Learn Res. 2023 Jul;202:28244-28276.

Abstract

One straightforward metric to evaluate a survival prediction model is based on the Mean Absolute Error (MAE) - the average of the absolute difference between the time predicted by the model and the true event time, over all subjects. Unfortunately, this is challenging because, in practice, the test set includes (right) censored individuals, meaning we do not know when a censored individual actually experienced the event. In this paper, we explore various metrics to estimate MAE for survival datasets that include (many) censored individuals. Moreover, we introduce a novel and effective approach for generating realistic semi-synthetic survival datasets to facilitate the evaluation of metrics. Our findings, based on the analysis of the semi-synthetic datasets, reveal that our proposed metric (MAE using pseudo-observations) is able to rank models accurately based on their performance, and often closely matches the true MAE - in particular, is better than several alternative methods.

摘要

评估生存预测模型的一个直接指标是基于平均绝对误差(MAE)——模型预测时间与真实事件时间之间绝对差值的平均值,涵盖所有受试者。不幸的是,这具有挑战性,因为在实际中,测试集包含(右)删失个体,这意味着我们不知道删失个体实际何时经历该事件。在本文中,我们探索了各种指标来估计包含(众多)删失个体的生存数据集的MAE。此外,我们引入了一种新颖且有效的方法来生成逼真的半合成生存数据集,以促进指标评估。基于对半合成数据集的分析,我们的研究结果表明,我们提出的指标(使用伪观测值的MAE)能够根据模型性能准确地对模型进行排名,并且通常与真实MAE非常接近——特别是,优于几种替代方法。

相似文献

本文引用的文献

1
Survival Mixture Density Networks.
Proc Mach Learn Res. 2022 Aug;182:224-248.
2
Personalized breast cancer onset prediction from lifestyle and health history information.
PLoS One. 2022 Dec 19;17(12):e0279174. doi: 10.1371/journal.pone.0279174. eCollection 2022.
3
Inverse-Weighted Survival Games.
Adv Neural Inf Process Syst. 2021 Dec;34:2160-2172.
5
Variational Learning of Individual Survival Distributions.
Proc ACM Conf Health Inference Learn (2020). 2020 Apr;2020:10-18. doi: 10.1145/3368555.3384454. Epub 2020 Apr 2.
6
X-CAL: Explicit Calibration for Survival Analysis.
Adv Neural Inf Process Syst. 2020 Dec;33:18296-18307.
7
Using survival prediction techniques to learn consumer-specific reservation price distributions.
PLoS One. 2021 Apr 29;16(4):e0249182. doi: 10.1371/journal.pone.0249182. eCollection 2021.
9
The Cancer Genome Atlas Pan-Cancer analysis project.
Nat Genet. 2013 Oct;45(10):1113-20. doi: 10.1038/ng.2764.
10
Pseudo-observations for competing risks with covariate dependent censoring.
Lifetime Data Anal. 2014 Apr;20(2):303-15. doi: 10.1007/s10985-013-9247-7. Epub 2013 Feb 22.

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍。

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索

文档翻译

学术文献翻译模型,支持多种主流文档格式。

立即体验