FA: Forced Prompt Learning of Vision-Language Models for Out-of-Distribution Detection
By: Xinhua Lu , Runhe Lai , Yanqi Wu and more
Potential Business Impact:
Helps computers tell normal from weird pictures.
Pre-trained vision-language models (VLMs) have advanced out-of-distribution (OOD) detection recently. However, existing CLIP-based methods often focus on learning OOD-related knowledge to improve OOD detection, showing limited generalization or reliance on external large-scale auxiliary datasets. In this study, instead of delving into the intricate OOD-related knowledge, we propose an innovative CLIP-based framework based on Forced prompt leArning (FA), designed to make full use of the In-Distribution (ID) knowledge and ultimately boost the effectiveness of OOD detection. Our key insight is to learn a prompt (i.e., forced prompt) that contains more diversified and richer descriptions of the ID classes beyond the textual semantics of class labels. Specifically, it promotes better discernment for ID images, by forcing more notable semantic similarity between ID images and the learnable forced prompt. Moreover, we introduce a forced coefficient, encouraging the forced prompt to learn more comprehensive and nuanced descriptions of the ID classes. In this way, FA is capable of achieving notable improvements in OOD detection, even when trained without any external auxiliary datasets, while maintaining an identical number of trainable parameters as CoOp. Extensive empirical evaluations confirm our method consistently outperforms current state-of-the-art methods. Code is available at https://github.com/0xFAFA/FA.
Similar Papers
Harnessing Large Language and Vision-Language Models for Robust Out-of-Distribution Detection
CV and Pattern Recognition
Helps computers spot weird things in pictures.
Prompt Optimization Meets Subspace Representation Learning for Few-shot Out-of-Distribution Detection
Machine Learning (CS)
AI spots new things it hasn't seen before.
FOCoOp: Enhancing Out-of-Distribution Robustness in Federated Prompt Learning for Vision-Language Models
CV and Pattern Recognition
Helps AI learn from many computers without sharing private data.