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

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]

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/.

• python
• torch (GPU version only)

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

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

@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}
}