Skip to main content

An Effective Meaningful Way to Evaluate Survival Models

Publication ,  Conference
Qi, SA; Kumar, N; Farrokh, M; Sun, W; Kuan, LH; Ranganath, R; Henao, R; Greiner, R
Published in: Proceedings of Machine Learning Research
January 1, 2023

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.

Duke Scholars

Published In

Proceedings of Machine Learning Research

EISSN

2640-3498

Publication Date

January 1, 2023

Volume

202

Start / End Page

28244 / 28276
 

Citation

APA
Chicago
ICMJE
MLA
NLM
Qi, S. A., Kumar, N., Farrokh, M., Sun, W., Kuan, L. H., Ranganath, R., … Greiner, R. (2023). An Effective Meaningful Way to Evaluate Survival Models. In Proceedings of Machine Learning Research (Vol. 202, pp. 28244–28276).
Qi, S. A., N. Kumar, M. Farrokh, W. Sun, L. H. Kuan, R. Ranganath, R. Henao, and R. Greiner. “An Effective Meaningful Way to Evaluate Survival Models.” In Proceedings of Machine Learning Research, 202:28244–76, 2023.
Qi SA, Kumar N, Farrokh M, Sun W, Kuan LH, Ranganath R, et al. An Effective Meaningful Way to Evaluate Survival Models. In: Proceedings of Machine Learning Research. 2023. p. 28244–76.
Qi, S. A., et al. “An Effective Meaningful Way to Evaluate Survival Models.” Proceedings of Machine Learning Research, vol. 202, 2023, pp. 28244–76.
Qi SA, Kumar N, Farrokh M, Sun W, Kuan LH, Ranganath R, Henao R, Greiner R. An Effective Meaningful Way to Evaluate Survival Models. Proceedings of Machine Learning Research. 2023. p. 28244–28276.

Published In

Proceedings of Machine Learning Research

EISSN

2640-3498

Publication Date

January 1, 2023

Volume

202

Start / End Page

28244 / 28276