用无标签图像提升零样本分类,融合LLM与自监督视觉模型优势。
CLIP meets DINO for Tuning Zero-Shot Classifier using Unlabeled Image Collections
- 用LLM生成更精准的类别描述,替代CLIP默认提示词。
- 通过伪标签训练对齐模块,融合DINO视觉特征与文本嵌入。
- 无需标注数据,性能超越现有方法3.6%平均提升。
在基础模型时代,CLIP虽能对齐文本与视觉模态,但其训练目标导致细粒度任务下视觉特征表现不佳。而像DINO这样的自监督模型因特定训练范式,可提取丰富视觉特征,却需依赖全标注数据进行线性探测。本文提出一种无标签提示调优方法(NoLA),利用自监督模型(DINO)的强视觉特征和大语言模型(LLM)的广泛文本知识,提升基于CLIP的图像分类性能。流程包括:(1) 通过LLM生成更具代表性的类别描述,获得更优文本嵌入;(2) 利用该文本嵌入生成伪标签,训练一个整合文本与视觉特征的对齐模块;(3) 通过训练好的对齐模块,以DINO辅助监督方式提示调优CLIP的视觉编码器。该三步法融合了文本与视觉基础模型的优势,在11个多样化图像分类数据集上,相比当前最优的LaFTer方法平均提升3.6%。代码与模型已开源。
原文摘要 · Abstract (English)
In the era of foundation models, CLIP has emerged as a powerful tool for aligning text & visual modalities into a common embedding space. However, the alignment objective used to train CLIP often results in subpar visual features for fine-grained tasks. In contrast, SSL-pretrained models like DINO excel at extracting rich visual features due to their specialized training paradigm. Yet, these SSL models require an additional supervised linear probing step, which relies on fully labeled data which is often expensive and difficult to obtain at scale. In this paper, we propose a label-free prompt-tuning method that leverages the rich visual features of self-supervised learning models (DINO) and the broad textual knowledge of large language models (LLMs) to largely enhance CLIP-based image classification performance using unlabeled images. Our approach unfolds in three key steps: (1) We generate robust textual feature embeddings that more accurately represent object classes by leveraging class-specific descriptions from LLMs, enabling more effective zero-shot classification compared to CLIP's default name-specific prompts. (2) These textual embeddings are then used to produce pseudo-labels to train an alignment module that integrates the complementary strengths of LLM description-based textual embeddings & DINO's visual features. (3) Finally, we prompt-tune CLIP's vision encoder through DINO-assisted supervision using the trained alignment module. This three-step process allows us to harness the best of visual & textual foundation models, resulting in a powerful and efficient approach that surpasses state-of-the-art label-free classification methods. Notably, our framework, NoLA (No Labels Attached), achieves an average absolute gain of 3.6% over the state-of-the-art LaFTer across 11 diverse image classification datasets. Our code & models can be found at https://github.com/fazliimam/NoLA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。