Repository accompanying the paper "Uncertainty estimation for deep learning-based automated analysis of 12-lead electrocardiograms" by Vranken and Van de Leur et al. It contains all the uncertainty estimation methods described in the paper.
First, install dependencies
# clone project
git clone https://github.com/rutgervandeleur/uncertainty
# install project
cd uncertainty
pip install -e .
pip install -r requirements.txt
Next, navigate to any file and run it.
# module folder
cd project
# run module (example: Variational inference with bayesian decomposition)
python main.py --epistemic_method varinf --aleatoric_method bayesdecomp
@article{VrankenUncertainty2021,
title={Uncertainty estimation for deep learning-based automated analysis of 12-lead electrocardiograms},
author={Jeroen F. Vranken, Rutger R. van de Leur, Deepak K. Gupta, Luis E. Juarez Orozco, Rutger J. Hassink, Pim van der Harst, Pieter A. Doevendans, Sadaf Gulshad, René van Es},
journal={submitted},
year={2021}
}