Skip to content

Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-Counts (Neurips 2020)

License

Notifications You must be signed in to change notification settings

sharpenb/Posterior-Network

Repository files navigation

Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-Counts

This repository presents the experiments of the paper:

Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-Counts
Bertrand Charpentier, Daniel Zügner, Stephan Günnemann
Conference on Neural Information Processing Systems (NeurIPS), 2020.

[Paper|Video]

Diagram

Requirements

To install requirements:

conda env create -f environment.yaml
conda activate posterior-network
conda env list

Training & Evaluation

To train the model(s) in the paper, run one jupyter notebook in the folder notebooks. All parameter are described.

To dowload the datasets, follow the following links:

Pre-trained Models

You can find pre-trained models in the folder saved_models. Models in saved_models/MNIST-postnet and saved_models/CIFAR10-postnet are trained on classic MNIST and CIFAR10 splits.

Cite

Please cite our paper if you use the model or this code in your own work:

@incollection{postnet,
title = {Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-Counts},
author = {Charpentier, Bertrand, Daniel Z\"{u}gner and G\"{u}nnemann, Stephan},
booktitle = {Advances in Neural Information Processing Systems 33},
year = {2020},
publisher = {Curran Associates, Inc.},
}

About

Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-Counts (Neurips 2020)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published