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

by   Zhiliang Wu, et al.

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.



There are no comments yet.


page 1

page 2

page 3

page 4


Quantifying Predictive Uncertainty in Medical Image Analysis with Deep Kernel Learning

Deep neural networks are increasingly being used for the analysis of med...

Mitigating Uncertainty in Document Classification

The uncertainty measurement of classifiers' predictions is especially im...

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 ...

Bayesian Neural Hawkes Process for Event Uncertainty Prediction

Many applications comprise of sequences of event data with the time of o...

The Peril of Popular Deep Learning Uncertainty Estimation Methods

Uncertainty estimation (UE) techniques – such as the Gaussian process (G...

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

Event ticket price prediction is important to marketing strategy for any...

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.