Bayesian LSTMs in medicine
arXiv:1706.01242
Abstract
The medical field stands to see significant benefits from the recent advances in deep learning. Knowing the uncertainty in the decision made by any machine learning algorithm is of utmost importance for medical practitioners. This study demonstrates the utility of using Bayesian LSTMs for classification of medical time series. Four medical time series datasets are used to show the accuracy improvement Bayesian LSTMs provide over standard LSTMs. Moreover, we show cherry-picked examples of confident and uncertain classifications of the medical time series. With simple modifications of the common practice for deep learning, significant improvements can be made for the medical practitioner and patient.
11 pages, 8 figures
References in corpus (9)
- Adam: A Method for Stochastic Optimization
- A Theoretically Grounded Application of Dropout in Recurrent Neural Networks
- Bayesian Convolutional Neural Networks with Bernoulli Approximate Variational Inference
- Learning to Diagnose with LSTM Recurrent Neural Networks
- Doctor AI: Predicting Clinical Events via Recurrent Neural Networks
- Phased LSTM: Accelerating Recurrent Network Training for Long or Event-based Sequences
- Bayesian Recurrent Neural Networks
- Architectural Complexity Measures of Recurrent Neural Networks
- Recurrent Batch Normalization
Cited by in corpus (4)
- Uncertainty-Aware Attention for Reliable Interpretation and Prediction
- Mitigating Uncertainty in Document Classification
- Stein Variational Online Changepoint Detection with Applications to Hawkes Processes and Neural Networks
- Efficient Computation Reduction in Bayesian Neural Networks Through Feature Decomposition and Memorization