A PyTorch Library for benchmarking and leveraging efficient predictive uncertainty quantification techniques.
Project description
TorchUncertainty is a package designed to help you leverage uncertainty quantification techniques and make your neural networks more reliable. It is based on PyTorch Lightning to handle multi-GPU training and inference and automatic logging through tensorboard.
This package provides a multi-level API, including:
- ready-to-train baselines on research datasets, such as ImageNet and CIFAR
- baselines available for training on your datasets
- pretrained weights for these baselines on ImageNet and CIFAR.
- layers available for use in your networks
Installation
The package can be installed from PyPI:
pip install torch-uncertainty
To contribute, install the package from source following the instructions of the dedicated page in the documentation and activate the pre-commit hooks with pre-commit install
.
Getting Started and Documentation
Please find the documentation at torch-uncertainty.github.io.
A quickstart is available at torch-uncertainty.github.io/quickstart.
Implemented baselines
To date, the following baselines are implemented:
- Deep Ensembles
- Masksembles
- Packed-Ensembles
Tutorials
Awesome Uncertainty repositories
You may find a lot of information about modern uncertainty estimation techniques on the Awesome Uncertainty in Deep Learning.
Other References
This package also contains the official implementation of Packed-Ensembles.
If you find the corresponding models interesting, please consider citing our paper:
@inproceedings{laurent2023packed,
title={Packed-Ensembles for Efficient Uncertainty Estimation},
author={Laurent, Olivier and Lafage, Adrien and Tartaglione, Enzo and Daniel, Geoffrey and Martinez, Jean-Marc and Bursuc, Andrei and Franchi, Gianni},
booktitle={ICLR},
year={2023}
}
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for torch_uncertainty-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 1241722720b34fd302bc641baa6ac6bc434905a453e6e0991167d3ab00713273 |
|
MD5 | ee5667d1749bc70d2e63e571be1580a5 |
|
BLAKE2b-256 | 6901f3dc43b7b6472ae2fb5a4f0486b8f689fe9f88334866f0dbd62f8b590a67 |