Delving into Out-of-Distribution Detection with Vision-Language Representations
Yifei Ming, Ziyang Cai, Jiuxiang Gu, Yiyou Sun, Wei Li, Yixuan Li
Code Available — Be the first to reproduce this paper.
ReproduceCode
- github.com/deeplearning-wisc/mcmOfficialIn paperpytorch★ 97
- github.com/HHU-MMBS/plp-official-tmlr2024pytorch★ 4
Abstract
Recognizing out-of-distribution (OOD) samples is critical for machine learning systems deployed in the open world. The vast majority of OOD detection methods are driven by a single modality (e.g., either vision or language), leaving the rich information in multi-modal representations untapped. Inspired by the recent success of vision-language pre-training, this paper enriches the landscape of OOD detection from a single-modal to a multi-modal regime. Particularly, we propose Maximum Concept Matching (MCM), a simple yet effective zero-shot OOD detection method based on aligning visual features with textual concepts. We contribute in-depth analysis and theoretical insights to understand the effectiveness of MCM. Extensive experiments demonstrate that MCM achieves superior performance on a wide variety of real-world tasks. MCM with vision-language features outperforms a common baseline with pure visual features on a hard OOD task with semantically similar classes by 13.1% (AUROC). Code is available at https://github.com/deeplearning-wisc/MCM.
Tasks
Benchmark Results
| Dataset | Model | Metric | Claimed | Verified | Status |
|---|---|---|---|---|---|
| ImageNet-1k vs Curated OODs (avg.) | MCM (CLIP-L) | FPR95 | 38.17 | — | Unverified |
| ImageNet-1k vs iNaturalist | MCM (CLIP-B) | AUROC | 94.61 | — | Unverified |
| ImageNet-1k vs iNaturalist | MCM (CLIP-L) | AUROC | 94.95 | — | Unverified |
| ImageNet-1k vs Places | MCM (CLIP-L) | FPR95 | 35.42 | — | Unverified |
| ImageNet-1k vs Places | MCM (CLIP-B) | FPR95 | 44.69 | — | Unverified |
| ImageNet-1k vs SUN | MCM (CLIP-L) | FPR95 | 29 | — | Unverified |
| ImageNet-1k vs SUN | MCM (CLIP-B) | FPR95 | 37.59 | — | Unverified |
| ImageNet-1k vs Textures | MCM (CLIP-L) | AUROC | 84.88 | — | Unverified |
| ImageNet-1k vs Textures | MCM (CLIP-B) | AUROC | 86.11 | — | Unverified |