Multi-class Data Description (Deep-MCDD) for Out-of-distribution Detection

This is the author code of "Multi-class Data Description for Out-of-distribution Detection". Some codes are implemented based on Deep Mahalanobis Detector.

Overview

Downloading tabular datasets

The four multi-class tabular datsets reported in the paper can be downloaded from the below links. We provide the preprocessed version of the datasets, which are converted into the numpy array format, to allow users to easily load them. For more details of the datasets, please refer to the UCI repository. Place them in the directory ./table_data/.

GasSensor: [Numpy format] [Raw format]
Shuttle: [Numpy format] [Raw format]
DriveDiagnosis: [Numpy format] [Raw format]
MNIST: [Numpy format] [Raw format]

Downloading image datasets

The three in-distribution datasets (i.e., SVHN, CIFAR-10, and CIFAR-100) would be automatically downloaded via torchvision. We use the download links of two additional out-of-distributin datasets (i.e., TinyImageNet and LSUN) from Deep Mahalanobis Detector and ODIN Detector. Place them in the directory ./image_data/.

Running the codes

python
torch (GPU version only)

Training MLP with Deep-MCDD for tabular data

python train_deepmcdd_table.py --dataset gas --net_type mlp --oodclass_idx 0

Training CNN with Deep-MCDD for image data

python train_deepmcdd_image.py --dataset cifar10 --net_type resnet

Citation

@inproceedings{lee2020multi,
  author = {Lee, Dongha and Yu, Sehun and Yu, Hwanjo},
  title = {Multi-Class Data Description for Out-of-Distribution Detection},
  year = {2020},
  booktitle = {Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining},
  pages = {1362–1370}
}

donalee / DeepMCDD