Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Evidential time-to-event prediction with calibrated uncertainty quantification

About

Time-to-event analysis provides insights into clinical prognosis and treatment recommendations. However, this task is more challenging than standard regression problems due to the presence of censored observations. Additionally, the lack of confidence assessment, model robustness, and prediction calibration raises concerns about the reliability of predictions. To address these challenges, we propose an evidential regression model specifically designed for time-to-event prediction. The proposed model quantifies both epistemic and aleatory uncertainties using Gaussian Random Fuzzy Numbers and belief functions, providing clinicians with uncertainty-aware survival time predictions. The model is trained by minimizing a generalized negative log-likelihood function accounting for data censoring. Experimental evaluations using simulated datasets with different data distributions and censoring conditions, as well as real-world datasets across diverse clinical applications, demonstrate that our model delivers both accurate and reliable performance, outperforming state-of-the-art methods. These results highlight the potential of our approach for enhancing clinical decision-making in survival analysis.

Ling Huang, Yucheng Xing, Swapnil Mishra, Thierry Denoeux, Mengling Feng• 2024

Related benchmarks

TaskDatasetResultRank
Survival PredictionTCGA-LUAD
C-index0.565
195
Survival PredictionTCGA-UCEC
C-index0.657
179
Survival PredictionTCGA-BLCA
C-index0.587
116
Showing 3 of 3 rows

Other info

Follow for update