arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.04076cs.CV

大语言模型是低样本图像分类的良好提示学习器

Large Language Models are Good Prompt Learners for Low-Shot Image Classification

Viterbi School of Engineering(维特比工程学院);University of Southern California(南加州大学)

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

AI总结:

本文提出LLaMP方法,利用大语言模型为CLIP文本编码器生成自适应提示,以弥合语言与视觉的领域差距,在11个数据集上提升了低样本图像分类和零样本泛化性能。

摘要:

低样本图像分类,即训练图像有限或难以获取的情况,已受益于近期具有强泛化能力的预训练视觉-语言(VL)模型的进展,例如CLIP。基于VL模型构建的提示学习方法从类别名称生成文本特征,而这些特征仅包含有限的类别特定信息。大语言模型(LLMs)凭借其广博的百科知识,成为补充。因此,在本文中,我们讨论将LLMs集成以增强预训练VL模型,特别是在低样本分类上。然而,语言与视觉之间的领域差距阻碍了LLMs的直接应用。因此,我们提出LLaMP,即大语言模型作为提示学习器,为CLIP文本编码器生成自适应提示,使其成为连接桥梁。实验表明,与其他最先进的提示学习方法相比,LLaMP在11个数据集上的零样本泛化和少样本图像分类均取得了更好的性能。代码将发布于:https://github.com/zhaohengz/LLaMP。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑