This repository is the official implementation of Energy-based Out-of-distribution Detection by Weitang Liu, Xiaoyun Wang, John Owens and Yixuan Li. This method is an effective and easy OOD detector with and without fine-tuning. Our code is modified from Outlier-Exposure.

Pretrained models are provided in folder

./CIFAR/snapshots/

Please download the datasets in folder

./data/

run energy score testing for cifar10 WRN

bash run.sh energy 0

run energy score testing for cifar100 WRN

bash run.sh energy 1

run energy score training and testing for cifar10 WRN

bash run.sh energy_ft 0

run energy score training and testing for cifar100 WRN

bash run.sh energy_ft 1

Our model achieves the following average performance on 6 OOD datasets:

CIFAR-10:

CIFAR-100:

These experiments make use of numerous outlier datasets. Links for less common datasets are as follows, 80 Million Tiny Images Textures, Places365, LSUN-C, LSUN-R, iSUN and SVHN.

 @article{liu2020energy,
      title={Energy-based Out-of-distribution Detection},
      author={Liu, Weitang and Wang, Xiaoyun and Owens, John and Li, Yixuan},
      journal={Advances in Neural Information Processing Systems},
      year={2020}
 }