Out-of-distribution detection using the pNML regret. NeurIPS2021

Last update: Dec 02, 2022

Overview

OOD Detection

Load conda environment

conda env create -f environment.yml

or install requirements:

while read requirement; do conda install --yes $requirement || pip install $requirement; done < requirements.txt

# Download OOD data
cd bash_scripts
chmod 777 ./download_data.sh
./download_data.sh

# Download pretrained models
chmod 777 ./download_models.sh
./download_models.sh

Download imagenet30

Follow https://github.com/alinlab/CSI

Imagenet30 training set: https://drive.google.com/file/d/1B5c39Fc3haOPzlehzmpTLz6xLtGyKEy4/view

Imagenet30 testing set: https://drive.google.com/file/d/13xzVuQMEhSnBRZr-YaaO08coLU2dxAUq/view

Put and untar under ./data/Imagenet30

.
├── README.md
├── data
│   ├── Imagenet30
│   │   ├── one_class_test
│   │   ├── one_class_test.tar
│   │   ├── one_class_train
│   │   └── one_class_train.tar

Execute methods

Using the pretrained models, score ood detection

cd bash_scripts
chmod 777 ./execute_methods.sh
./execute_methods.sh

Create paper's tables

cd src
python main main_create_tables.py

Out-of-distribution detection using the pNML regret. NeurIPS2021

Related tags

Overview

OOD Detection

Download imagenet30

Execute methods

Owner

Koby Bibas

This repository provides the official implementation of 'Learning to ignore: rethinking attention in CNNs' accepted in BMVC 2021.

scikit-learn: machine learning in Python

Language Used: Python . Made in Jupyter(Anaconda) notebook.

World Models with TensorFlow 2

This repository contains the PyTorch implementation of the paper STaCK: Sentence Ordering with Temporal Commonsense Knowledge appearing at EMNLP 2021.

Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)

JAXDL: JAX (Flax) Deep Learning Library

Optimized code based on M2 for faster image captioning training

A lightweight tool to get an AI Infrastructure Stack up in minutes not days.

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

LSUN Dataset Documentation and Demo Code

PyTorch implementation of the cross-modality generative model that synthesizes dance from music.

This repository contains the code and models for the following paper.

Unity Propagation in Bayesian Networks Handling Inconsistency via Unity Smoothing

An end-to-end implementation of intent prediction with Metaflow and other cool tools

This repository contains several image-to-image translation models, whcih were tested for RGB to NIR image generation. The models are Pix2Pix, Pix2PixHD, CycleGAN and PointWise.

Answering Open-Domain Questions of Varying Reasoning Steps from Text

MagFace: A Universal Representation for Face Recognition and Quality Assessment

Learning Visual Words for Weakly-Supervised Semantic Segmentation

Refactoring dalle-pytorch and taming-transformers for TPU VM