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

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.

Zhaoheng Zheng, Jingmin Wei, Xuefeng Hu, Haidong Zhu, Ram Nevatia• 2023

Related benchmarks

TaskDatasetResultRank
Image ClassificationFood101--
309
Image ClassificationStanfordCars
Accuracy81.56
266
Base-to-New GeneralizationDTD
Base Accuracy83.49
68
Image ClassificationEuroSAT Base-to-New
Base Score91.93
65
Image ClassificationCaltech101 Base and New Classes
Base Accuracy98.45
50
Image ClassificationAverage of 11 datasets (ImageNet, Caltech101, OxfordPets, StanfordCars, Flowers102, Food101, FGVCAircraft, SUN397, DTD, EuroSAT, UCF101) Base-to-Novel Generalization
Base Accuracy85.16
49
Image ClassificationImageNet Base-to-New
H Score74.48
40
Image ClassificationOxford Pets
Base Accuracy96.31
38
Base-to-novel generalizationFlowers102
Base Accuracy97.82
29
Image ClassificationStanfordCars novel classes
Accuracy74.54
18
Showing 10 of 15 rows

Other info

Follow for update