Get our free extension to see links to code for papers anywhere online!

Chrome logo Add to Chrome

Firefox logo Add to Firefox

Personalized Survival Prediction with Contextual Explanation Networks

Jan 30, 2018
Maruan Al-Shedivat, Avinava Dubey, Eric P. Xing

Share this with someone who'll enjoy it:

Accurate and transparent prediction of cancer survival times on the level of individual patients can inform and improve patient care and treatment practices. In this paper, we design a model that concurrently learns to accurately predict patient-specific survival distributions and to explain its predictions in terms of patient attributes such as clinical tests or assessments. Our model is flexible and based on a recurrent network, can handle various modalities of data including temporal measurements, and yet constructs and uses simple explanations in the form of patient- and time-specific linear regression. For analysis, we use two publicly available datasets and show that our networks outperform a number of baselines in prediction while providing a way to inspect the reasons behind each prediction.

* Machine Learning for Healthcare Workshop, NIPS 2017 

   Access Paper Source

Share this with someone who'll enjoy it: