DeepAI AI Chat
Log In Sign Up

Uncertainty Estimation in Cancer Survival Prediction

by   Hrushikesh Loya, et al.
IIT Bombay

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.


page 1

page 2

page 3

page 4


Deep-CR MTLR: a Multi-Modal Approach for Cancer Survival Prediction with Competing Risks

Accurate survival prediction is crucial for development of precision can...

High-Dimensional Survival Analysis: Methods and Applications

In the era of precision medicine, time-to-event outcomes such as time to...

Predicting Survival of Tongue Cancer Patients by Machine Learning Models

Tongue cancer is a common oral cavity malignancy that originates in the ...

Effective Ways to Build and Evaluate Individual Survival Distributions

An accurate model of a patient's individual survival distribution can he...

Computing the Hazard Ratios Associated with Explanatory Variables Using Machine Learning Models of Survival Data

Purpose: The application of Cox Proportional Hazards (CoxPH) models to s...

Survival analysis as a classification problem

In this paper, we explore a method for treating survival analysis as a c...