Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

LoCoOp: Few-Shot Out-of-Distribution Detection via Prompt Learning

About

We present a novel vision-language prompt learning approach for few-shot out-of-distribution (OOD) detection. Few-shot OOD detection aims to detect OOD images from classes that are unseen during training using only a few labeled in-distribution (ID) images. While prompt learning methods such as CoOp have shown effectiveness and efficiency in few-shot ID classification, they still face limitations in OOD detection due to the potential presence of ID-irrelevant information in text embeddings. To address this issue, we introduce a new approach called Local regularized Context Optimization (LoCoOp), which performs OOD regularization that utilizes the portions of CLIP local features as OOD features during training. CLIP's local features have a lot of ID-irrelevant nuisances (e.g., backgrounds), and by learning to push them away from the ID class text embeddings, we can remove the nuisances in the ID class text embeddings and enhance the separation between ID and OOD. Experiments on the large-scale ImageNet OOD detection benchmarks demonstrate the superiority of our LoCoOp over zero-shot, fully supervised detection methods and prompt learning methods. Notably, even in a one-shot setting -- just one label per class, LoCoOp outperforms existing zero-shot and fully supervised detection methods. The code will be available via https://github.com/AtsuMiyai/LoCoOp.

Atsuyuki Miyai, Qing Yu, Go Irie, Kiyoharu Aizawa• 2023

Related benchmarks

TaskDatasetResultRank
Image ClassificationImageNet-1K--
524
Out-of-Distribution DetectionTextures
AUROC0.8886
141
OOD DetectionPlaces (OOD)
AUROC92.03
76
OOD DetectionSUN (OOD)
AUROC95.2
72
OOD DetectionImageNet-1K OOD (Average: OpenImage-O, Texture, iNaturalist, ImageNet-O) 1.0 (test)
AUROC93.52
61
Out-of-Distribution DetectionImageNet-1k Textures ID OOD
AUROC90.19
59
Out-of-Distribution DetectionImageNet 1k (test)
Average AUROC92.73
58
OOD DetectionCIFAR-100
FPR@9564.66
47
OOD DetectionImageNet SUN
FPR@9523.44
43
OOD DetectioniNaturalist
AUROC96.3
40
Showing 10 of 30 rows

Other info

Follow for update