arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.08573cs.LGstat.APstat.ML

Uncertainty Estimation in Cancer Survival Prediction

Indian Institute of Technology Bombay(印度理工学院孟买分校);Case Western Reserve University(凯斯西储大学);University of Illinois at Chicago(伊利诺伊大学芝加哥分校)

Hrushikesh Loya, Pranav Poduval, Deepak Anand, Neeraj Kumar, Amit Sethi

英文摘要:

Survival models are used in various fields, such as the development of cancer treatment protocols. Although many statistical and machine learning models have been proposed to achieve accurate survival predictions, little attention has been paid to obtain well-calibrated uncertainty estimates associated with each prediction. The currently popular models are opaque and untrustworthy in that they often express high confidence even on those test cases that are not similar to the training samples, and even when their predictions are wrong. We propose a Bayesian framework for survival models that not only gives more accurate survival predictions but also quantifies the survival uncertainty better. Our approach is a novel combination of variational inference for uncertainty estimation, neural multi-task logistic regression for estimating nonlinear and time-varying risk models, and an additional sparsity-inducing prior to work with high dimensional data.

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

↑