Uncertainty-Aware Time-to-Event Prediction using Deep Kernel Accelerated Failure Time Models

07/26/2021
by   Zhiliang Wu, et al.
0

Recurrent neural network based solutions are increasingly being used in the analysis of longitudinal Electronic Health Record data. However, most works focus on prediction accuracy and neglect prediction uncertainty. We propose Deep Kernel Accelerated Failure Time models for the time-to-event prediction task, enabling uncertainty-awareness of the prediction by a pipeline of a recurrent neural network and a sparse Gaussian Process. Furthermore, a deep metric learning based pre-training step is adapted to enhance the proposed model. Our model shows better point estimate performance than recurrent neural network based baselines in experiments on two real-world datasets. More importantly, the predictive variance from our model can be used to quantify the uncertainty estimates of the time-to-event prediction: Our model delivers better performance when it is more confident in its prediction. Compared to related methods, such as Monte Carlo Dropout, our model offers better uncertainty estimates by leveraging an analytical solution and is more computationally efficient.

READ FULL TEXT
POST COMMENT

Comments

There are no comments yet.

Authors

page 1

page 2

page 3

page 4

06/01/2021

Quantifying Predictive Uncertainty in Medical Image Analysis with Deep Kernel Learning

Deep neural networks are increasingly being used for the analysis of med...
07/17/2019

Mitigating Uncertainty in Document Classification

The uncertainty measurement of classifiers' predictions is especially im...
06/03/2019

Quantifying Point-Prediction Uncertainty in Neural Networks via Residual Estimation with an I/O Kernel

Neural Networks (NNs) have been extensively used for a wide spectrum of ...
12/29/2021

Bayesian Neural Hawkes Process for Event Uncertainty Prediction

Many applications comprise of sequences of event data with the time of o...
12/09/2021

The Peril of Popular Deep Learning Uncertainty Estimation Methods

Uncertainty estimation (UE) techniques – such as the Gaussian process (G...
12/03/2019

Event Ticket Price Prediction with Deep Neural Network on Spatial-Temporal Sparse Data

Event ticket price prediction is important to marketing strategy for any...
02/25/2020

Learning Queuing Networks by Recurrent Neural Networks

It is well known that building analytical performance models in practice...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.