An Effective Meaningful Way to Evaluate Survival Models
Shiang Qi, Neeraj Kumar, Mahtab Farrokh, Weijie Sun, Li-Hao Kuan, Rajesh Ranganath, Ricardo Henao, Russell Greiner
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 5a6e65d5-5274-4e3b-89d1-67855aa4acc8Cited by top-tier papers7
- Conformalized Survival Distributions: A Generic Post-Process to Increase CalibrationShiang Qi, Yakun Yu, Russell GreinerICML 2024 · 10 citations
- Toward Conditional Distribution Calibration in Survival PredictionShiang Qi, Yakun Yu, Russell GreinerNeurIPS 2024 · 5 citations
- Fair Federated Survival AnalysisMd Mahmudur Rahman, Sanjay PurushothamAAAI 2025 · 1 citation
- SurvHTE-Bench: A Benchmark for Heterogeneous Treatment Effect Estimation in Survival AnalysisShahriar Noroozizadeh, Xiaobin Shen, Jeremy C. Weiss, George H. ChenICLR 2026
- Subgroup Discovery with the Cox ModelZachary Izzo, Iain MelvinICML 2026
Builds on1
Related papers
- When Can We Trust Survival Model Evaluation ?Ghanem BAHRINI, Sebastien Razakarivony, Jean-François Dupuy, Valerie Gares et al.ICML 2026 · 27 citations
- Take Control of Censoring, Generate Real-World Like Synthetic DataGhanem Bahrini, Morgane Barbet-Massin, Sébastien Razakarivony, Valérie Garès et al.KDD 2026
- Fair and Interpretable Models for Survival AnalysisMd. Mahmudur Rahman, Sanjay PurushothamKDD 2022 · 11 citations
- Multi-Source Survival Domain AdaptationAmmar Shaker, Carolin LawrenceAAAI 2023 · 4 citations
- SurvDiff: A Diffusion Model for Generating Synthetic Data in Survival AnalysisMarie Brockschmidt, Maresa Schröder, Stefan FeuerriegelICML 2026 · 2 citations
