A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch
-
Updated
Feb 10, 2024 - Python
A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch
This repo contains a PyTorch implementation of the paper: "Evidential Deep Learning to Quantify Classification Uncertainty"
This repository provides the code used to implement the framework to provide deep learning models with total uncertainty estimates as described in "A General Framework for Uncertainty Estimation in Deep Learning" (Loquercio, Segù, Scaramuzza. RA-L 2020).
My implementation of the paper "Simple and Scalable Predictive Uncertainty estimation using Deep Ensembles"
Official repository for the paper "Masksembles for Uncertainty Estimation" (CVPR 2021).
Code to accompany the paper 'Improving model calibration with accuracy versus uncertainty optimization'.
Model zoo for different kinds of uncertainty quantification methods used in Natural Language Processing, implemented in PyTorch.
Uncertainty-Wizard is a plugin on top of tensorflow.keras, allowing to easily and efficiently create uncertainty-aware deep neural networks. Also useful if you want to train multiple small models in parallel.
Inferring distributions over depth from a single image, IROS 2019
Implementation of the MNIST experiment for Monte Carlo Dropout from http://mlg.eng.cam.ac.uk/yarin/PDFs/NIPS_2015_bayesian_convnets.pdf
[WACV'22] Official implementation of "HHP-Net: A light Heteroscedastic neural network for Head Pose estimation with uncertainty"
Code for "Deal: Deep Evidential Active Learning for Image Classification" (ICMLA 2020)
PyTorch implementation of Probabilistic MIMO U-Net
A project to train your model from scratch or fine-tune a pretrained model using the losses provided in this library to improve out-of-distribution detection and uncertainty estimation performances. Calibrate your model to produce enhanced uncertainty estimations. Detect out-of-distribution data using the defined score type and threshold.
A CNN based Depth, Optical Flow, Flow Uncertainty and Camera Pose Prediction pipeline
NOMU: Neural Optimization-based Model Uncertainty
Official Code: Trust Your Robots! Predictive Uncertainty Estimation of Neural Networks with Sparse Gaussian Processes
Experiments from Efficient Training of Interval Neural Networks for Imprecise Training Data
Code and supporting materials for the ICLR 2020 RIO paper
This repository provides the official implementation of "Robust channel-wise illumination estimation." accepted in BMVC (2021).
Add a description, image, and links to the uncertainty-neural-networks topic page so that developers can more easily learn about it.
To associate your repository with the uncertainty-neural-networks topic, visit your repo's landing page and select "manage topics."