- This repository provides official PyTorch implementations for DMR.
- This work has been accepted to the CVPR 2024 workshop on VAND 2.0: Visual Anomaly and Novelty Detection (VAND).
Authors: Dasol Choi, Dongbin Na
Out-of-distribution (OOD) detection is crucial for the reliable deployment of deep-learning applications. When a given input image does not belong to any categories of the deployed classification model, the classification model is expected to alert the user that the predicted outputs might be unreliable. Recent studies have shown that utilizing a large amount of explicit OOD training data helps improve OOD detection performance. However, collecting explicit real-world OOD data is burdensome, and pre-defining all out-of-distribution labels is fundamentally difficult. In this work, we present a novel method, Disentangling Marginal Representations (DMR), that generates artificial OOD training data by extracting marginal features from images of an In-Distribution (ID) training dataset and manipulating these extracted marginal representations. DMR is intuitive and can be used as a realistic solution that does not require any extra real-world OOD data. Moreover, our method can be simply applied to pre-trained classifier networks without affecting the original classification performance. We demonstrate that a shallow rejection network that is trained on the small subset of synthesized OOD training data generated from our method and attachable to the classifier network achieves superior OOD detection performance. With extensive experiments, we show that our proposed method significantly outperforms the state-of-the-art OOD detection methods on the broadly used CIFAR-10 and CIFAR-100 detection benchmark datasets. We also demonstrate that our proposed method can be further improved when combined with existing methods.
- We use all the 50,000
$\mathcal{X}^{ID}$ images from CIFAR-10 and CIFAR-100 as the ID dataset. - The synthesized 50,000
$\mathcal{X}^{OOD}_{train}$ images are used as the OOD training dataset for KIRBY and DMR. - OOD dataset for evaluation: (1) SVHN, (2) Textures, (3) LSUN-Crop, (4) Tiny-ImageNet, (5) Places-365, (6) Gaussian Noise.
- MSP (Maximum Softmax Probability) (ICLR 2017) [Paper]
- ODIN (ICLR 2018) [Paper]
- Mahalanobis (NeurIPS 2018) [Paper]
- Energy-based (NeurIPS 2020) [Paper]
- Entropy (CVPR 2021) [Paper]
- Maximum Logit (ICML 2022) [Paper]
- KL-Matching (ICML 2022) [Paper]
- ViM (CVPR 2022) [Paper]
- OE (Outlier Exposure) (ICLR 2019) [Paper]
- KIRBY (AAAI 2023) [Paper]
- OOD training images generation codes: [CIFAR-10]
- Generated OOD training images (.zip): [CIFAR-10]
- Ours
- OOD training image generation codes:
- Multiple Latent Mix-up without DMR: [CIFAR-10]
- Single Latent Inversion with DMR: [CIFAR-10]
- Multiple Latent Mix-up with DMR: [CIFAR-10]
- Generated OOD training images (.zip):
- Multiple Latent Mix-up without DMR: [CIFAR-10]
- Single Latent Inversion with DMR: [CIFAR-10]
- Multiple Latent Mix-up with DMR: [CIFAR-10]
- OOD training image generation codes:
- Baseline Evaluations: [CIFAR-10], [CIFAR-100]
- DMR Evaluations: [CIFAR-10], [CIFAR-100]
@inproceedings{choi2024dmr, title={DMR: Disentangling Marginal Representations for Out-of-Distribution Detection}, author={Choi, Dasol and Na, Dongbin}, booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition}, pages={4032--4041}, year={2024} }