evidential-deep-learning

Learn fast, scalable, and calibrated measures of uncertainty using neural networks!


Keywords
deep-learning, neural-network, tensorflow, pytorch, uncertainty, evidence, confidence, neurips-2020
License
Apache-2.0
Install
pip install evidential-deep-learning==0.4.0

Documentation

Evidential Deep Learning

This repository contains the code to reproduce the recent NeurIPS submissions Deep Evidential Regression as well as more general code to leverage evidential learning to train neural networks to learn their own measures of uncertainty directly from data!

Code is coming soon! and will be published when the paper is presented during the NeurIPS 2020 conference.

Citation

If you use this code for evidential learning as part of your project or paper, please cite the following work:

@article{amini2020deep,
  title={Deep evidential regression},
  author={Amini, Alexander and Schwarting, Wilko and Soleimany, Ava and Rus, Daniela},
  journal={Advances in Neural Information Processing Systems},
  volume={33},
  year={2020}
}