DeepAI AI Chat
Log In Sign Up

Learning to Predict with Supporting Evidence: Applications to Clinical Risk Prediction

by   Aniruddh Raghu, et al.

The impact of machine learning models on healthcare will depend on the degree of trust that healthcare professionals place in the predictions made by these models. In this paper, we present a method to provide people with clinical expertise with domain-relevant evidence about why a prediction should be trusted. We first design a probabilistic model that relates meaningful latent concepts to prediction targets and observed data. Inference of latent variables in this model corresponds to both making a prediction and providing supporting evidence for that prediction. We present a two-step process to efficiently approximate inference: (i) estimating model parameters using variational learning, and (ii) approximating maximum a posteriori estimation of latent variables in the model using a neural network, trained with an objective derived from the probabilistic model. We demonstrate the method on the task of predicting mortality risk for patients with cardiovascular disease. Specifically, using electrocardiogram and tabular data as input, we show that our approach provides appropriate domain-relevant supporting evidence for accurate predictions.


Image Classification with Consistent Supporting Evidence

Adoption of machine learning models in healthcare requires end users' tr...

A Variational Bayesian Approach to Learning Latent Variables for Acoustic Knowledge Transfer

We propose a variational Bayesian (VB) approach to learning distribution...

Probabilistic Machine Learning for Healthcare

Machine learning can be used to make sense of healthcare data. Probabili...

Learning Deep Bayesian Latent Variable Regression Models that Generalize: When Non-identifiability is a Problem

Bayesian Neural Networks with Latent Variables (BNN+LV's) provide uncert...

Probabilistic Adaptive Computation Time

We present a probabilistic model with discrete latent variables that con...

Single Index Latent Variable Models for Network Topology Inference

A semi-parametric, non-linear regression model in the presence of latent...

Deep learning of thermodynamics-aware reduced-order models from data

We present an algorithm to learn the relevant latent variables of a larg...