Zero-Shot Out-of-Distribution Detection Based on the Pre-trained Model CLIP
Обнаружение вне распределения в режиме zero-shot на основе предобученной модели CLIP
2022-06-28
SCID: 54.1/kgtnpfn8
Discuss with AI
CLIPcandidate unknown class namesmulti-modal representation learningtext-based image description generatorzero-shot out-of-distribution detection
Figures from the paper
Abstract (AI)
In an out-of-distribution (OOD) detection problem, samples of known classes (also called in-distribution classes) are used to train a special classifier. In testing, the classifier can (1) classify the test samples of known classes to their respective classes and also (2) detect samples that do not belong to any of the known classes (i.e., they belong to some unknown or OOD classes). This paper studies the problem of zero-shot out-of-distribution (OOD) detection, which still performs the same two tasks in testing but has no training except using the given known class names. This paper proposes a novel and yet simple method (called ZOC) to solve the problem. ZOC builds on top of the recent advances in zero-shot classification through multi-modal representation learning. It first extends the pre-trained language-vision model CLIP by training a text-based image description generator on top of CLIP. In testing, it uses the extended model to generate candidate unknown class names for each test sample and computes a confidence score based on both the known class names and candidate unknown class names for zero-shot OOD detection. Experimental results on 5 benchmark datasets for OOD detection demonstrate that ZOC outperforms the baselines by a large margin.
Key Findings
1
During testing, computes confidence scores using both known class names and generated candidate unknown names to detect OOD samples.
2
Evaluated on 5 benchmark OOD datasets, ZOC outperforms baseline methods by a large margin.
3
Extends pre-trained CLIP by training a text-based image description generator on top of CLIP to produce candidate unknown class names.
4
Introduces ZOC, a simple zero-shot OOD detection method that requires no training data besides known class names.
Research Object
Zero-shot out-of-distribution (OOD) detection using a CLIP-based multi-modal model (extended CLIP with a text-based image description generator)
Research Subject
Detecting unknown/OOD samples and classifying known-class samples without training data by generating candidate unknown class names from the extended CLIP model and computing confidence scores for zero-shot OOD detection
Publication Details
Publication Date
2022-06-28
Journal
Publisher
ISSN
Cited by
113
Open access PDF
Access Type
Author Information
Download PDF
Subscribe to digest