OpenOOD icon indicating copy to clipboard operation
OpenOOD copied to clipboard

Benchmarking Generalized Out-of-Distribution Detection

OpenOOD: Benchmarking Generalized OOD Detection

paper     paper     paper

This repository reproduces representative methods within the Generalized Out-of-Distribution Detection Framework, aiming to make a fair comparison across methods that initially developed for anomaly detection, novelty detection, open set recognition, and out-of-distribution detection. This codebase is still under construction. Comments, issues, contributions, and collaborations are all welcomed!

timeline.jpg
Timeline of the methods that OpenOOD supports.

Updates

Get Started

To setup the environment, we use conda to manage our dependencies.

Our developers use CUDA 10.1 to do experiments.

You can specify the appropriate cudatoolkit version to install on your machine in the environment.yml file, and then run the following to create the conda environment:

conda env create -f environment.yml
conda activate openood

Datasets and pretrained models are provided here. Please unzip the files if necessary.

Our codebase accesses the datasets from ./data/ and pretrained models from ./results/checkpoints/ by default.

├── ...
├── data
│   ├── benchmark_imglist
│   ├── images_classic
│   ├── images_medical
│   └── images_largescale
├── openood
├── results
│   ├── checkpoints
│   └── ...
├── scripts
├── main.py
├── ...

The easiest hands-on script is to train LeNet-5 on MNIST and evaluate its OOD or FS-OOD performance with MSP baseline.

sh scripts/basics/mnist/train_mnist.sh
sh scripts/ood/msp/mnist_test_ood_msp.sh

Supported Benchmarks (10)

This part lists all the benchmarks we support. Feel free to include more.

Anomaly Detection (1)
Open Set Recognition (4)
  • [x] MNIST-4/6
  • [x] CIFAR-4/6
  • [x] CIFAR-40/60
  • [x] TinyImageNet-20/180
Out-of-Distribution Detection (5)
  • [x] BIMCV (A COVID X-Ray Dataset)

    Near-OOD: CT-SCAN, X-Ray-Bone;
    Far-OOD: MNIST, CIFAR-10, Texture, Tiny-ImageNet;
    Robust-ID: ActMed;

  • [x] MNIST

    Near-OOD: NotMNIST, FashionMNIST;
    Far-OOD: Texture, CIFAR-10, TinyImageNet, Places365;
    Robust-ID: SVHN;

  • [x] CIFAR-10

    Near-OOD: CIFAR-100, TinyImageNet;
    Far-OOD: MNIST, SVHN, Texture, Places365;
    Robust-ID: CINIC-10;

  • [x] CIFAR-100

    Near-OOD: CIFAR-10, TinyImageNet;
    Far-OOD: MNIST, SVHN, Texture, Places365;
    Robust-ID: CIFAR-100-C;

  • [x] ImageNet-1K

    Near-OOD: Species, iNaturalist, ImageNet-O, OpenImage-O;
    Far-OOD: Texture, MNIST;
    Robust-ID: ImageNet-v2;


Supported Backbones (6)

This part lists all the backbones we will support in our codebase, including CNN-based and Transformer-based models. Backbones like ResNet-50 and Transformer have ImageNet-1K/22K pretrained models.

CNN-based Backbones (4)
Transformer-based Architectures (2)

Supported Methods (34)

This part lists all the methods we include in this codebase. In v0.5, we totally support more than 32 popular methods for generalized OOD detection.

All the supported methodolgies can be placed in the following four categories.

density   reconstruction   classification   distance

We also note our supported methodolgies with the following tags if they have special designs in the corresponding steps, compared to the standard classifier training process.

preprocess   extradata   training   postprocess

Anomaly Detection (5)
  • [x] training postprocess
  • [x] training postprocess
  • [x] training postprocess
  • [x] training postprocess
  • [x] training postprocess
Open Set Recognition (3)

No Extra Data (2):

  • [x] training postprocess
  • [x] training postprocess

With Extra Data (1):

  • [x] training postprocess
Out-of-Distribution Detection (20)

No Extra Data (17):

  • [x] msp
  • [x] odin    postprocess
  • [x] mds    postprocess
  • [x] confbranch    preprocess   training
  • [x] godin    training   postprocess
  • [x] gram    postprocess
  • [x] duq    postprocess
  • [x] csi    preprocess   training   postprocess
  • [x] ebo    postprocess
  • [x] mos    training
  • [x] gradnorm    postprocess
  • [x] react    postprocess
  • [x] vos    training   postprocess
  • [x] vim    postprocess
  • [x] sem    preprocess   training   postprocess
  • [x] mls    postprocess
  • [x] kld    postprocess
  • [x] kld    training   preprocess

With Extra Data (3):

  • [x] oe    extradata
  • [x] mcd    extradata   training
  • [x] udg    extradata   training
Other Methods on Robustness and Uncertainty (6)
  • [x] mcdropout    training   postprocess
  • [x] deepensemble    training
  • [x] tempscale    postprocess
  • [x] mixup    preprocess
  • [x] cutmix    preprocess
  • [x] pixmix    preprocess

Contributing

We appreciate all contributions to improve OpenOOD. We sincerely welcome community users to participate in these projects. Please refer to CONTRIBUTING.md for the contributing guideline.

Contributors

Citation

If you find our repository useful for your research, please consider citing our paper:

@article{yang2022openood,
    author = {Yang, Jingkang and {\textit{et al.}}},
    title = {OpenOOD: Benchmarking Generalized Out-of-Distribution Detection},
    year = {2022}
}

@article{yang2022fsood,
    title = {Full-Spectrum Out-of-Distribution Detection},
    author = {Yang, Jingkang and Zhou, Kaiyang and Liu, Ziwei},
    journal={arXiv preprint arXiv:2204.05306},
    year = {2022}
}

@article{yang2021oodsurvey,
    title={Generalized Out-of-Distribution Detection: A Survey},
    author={Yang, Jingkang and Zhou, Kaiyang and Li, Yixuan and Liu, Ziwei},
    journal={arXiv preprint arXiv:2110.11334},
    year={2021}
}