Large Language Models are Good Prompt Learners for Low-Shot Image Classification
About
Low-shot image classification, where training images are limited or inaccessible, has benefited from recent progress on pre-trained vision-language (VL) models with strong generalizability, e.g. CLIP. Prompt learning methods built with VL models generate text features from the class names that only have confined class-specific information. Large Language Models (LLMs), with their vast encyclopedic knowledge, emerge as the complement. Thus, in this paper, we discuss the integration of LLMs to enhance pre-trained VL models, specifically on low-shot classification. However, the domain gap between language and vision blocks the direct application of LLMs. Thus, we propose LLaMP, Large Language Models as Prompt learners, that produces adaptive prompts for the CLIP text encoder, establishing it as the connecting bridge. Experiments show that, compared with other state-of-the-art prompt learning methods, LLaMP yields better performance on both zero-shot generalization and few-shot image classification, over a spectrum of 11 datasets. Code will be made available at: https://github.com/zhaohengz/LLaMP.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Classification | Food101 | -- | 309 | |
| Image Classification | StanfordCars | Accuracy81.56 | 266 | |
| Base-to-New Generalization | DTD | Base Accuracy83.49 | 68 | |
| Image Classification | EuroSAT Base-to-New | Base Score91.93 | 65 | |
| Image Classification | Caltech101 Base and New Classes | Base Accuracy98.45 | 50 | |
| Image Classification | Average of 11 datasets (ImageNet, Caltech101, OxfordPets, StanfordCars, Flowers102, Food101, FGVCAircraft, SUN397, DTD, EuroSAT, UCF101) Base-to-Novel Generalization | Base Accuracy85.16 | 49 | |
| Image Classification | ImageNet Base-to-New | H Score74.48 | 40 | |
| Image Classification | Oxford Pets | Base Accuracy96.31 | 38 | |
| Base-to-novel generalization | Flowers102 | Base Accuracy97.82 | 29 | |
| Image Classification | StanfordCars novel classes | Accuracy74.54 | 18 |