arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.18231cs.CV

TCP:面向视觉-语言模型的基于文本的类别感知提示调优

TCP:Textual-based Class-aware Prompt tuning for Visual-Language Model

Institute of Automation, CAS(中国科学院自动化研究所);State Key Laboratory of Multimodal Artificial Intelligence Systems(多模态人工智能系统国家重点实验室);State Key Lab of Processors(处理器国家重点实验室);Institute of Computing Technology, CAS(中国科学院计算技术研究所);University of Chinese Academy of Sciences(中国科学院大学)

Hantao Yao, Rui Zhang, Changsheng Xu

AI总结:

提出TCP方法,利用文本知识嵌入将类别先验知识融入提示调优,生成动态类别感知分类器,提升视觉-语言模型在未见领域的判别性,且训练效率高。

摘要:

提示调优是一种将预训练视觉-语言模型(VLM)适配到各种下游任务的有价值技术。近期基于CoOp的方法提出了一组可学习的领域共享或图像条件文本令牌,以促进任务特定文本分类器的生成。然而,这些文本令牌在未见领域上的泛化能力有限,因为它们无法动态调整以适应测试类别的分布。为解决此问题,我们提出了一种新颖的基于文本的类别感知提示调优(TCP),其显式地融入关于类别的先验知识以增强其判别性。TCP的关键概念在于利用文本知识嵌入(TKE)将类别级文本知识的高泛化性映射为类别感知的文本令牌。通过将这些类别感知提示无缝集成到文本编码器中,生成一个动态的类别感知分类器,以增强对未见领域的判别性。在推理期间,TKE动态生成与未见类别相关的类别感知提示。全面评估表明,TKE作为一个即插即用模块,可轻松与现有方法结合。此外,TCP在要求更少训练时间的同时,持续实现优越性能。代码:https://github.com/htyao89/Textual-based_Class-aware_prompt_tuning/

英文摘要:

Prompt tuning represents a valuable technique for adapting pre-trained visual-language models (VLM) to various downstream tasks. Recent advancements in CoOp-based methods propose a set of learnable domain-shared or image-conditional textual tokens to facilitate the generation of task-specific textual classifiers. However, those textual tokens have a limited generalization ability regarding unseen domains, as they cannot dynamically adjust to the distribution of testing classes. To tackle this issue, we present a novel Textual-based Class-aware Prompt tuning(TCP) that explicitly incorporates prior knowledge about classes to enhance their discriminability. The critical concept of TCP involves leveraging Textual Knowledge Embedding (TKE) to map the high generalizability of class-level textual knowledge into class-aware textual tokens. By seamlessly integrating these class-aware prompts into the Text Encoder, a dynamic class-aware classifier is generated to enhance discriminability for unseen domains. During inference, TKE dynamically generates class-aware prompts related to the unseen classes. Comprehensive evaluations demonstrate that TKE serves as a plug-and-play module effortlessly combinable with existing methods. Furthermore, TCP consistently achieves superior performance while demanding less training time. Code:https://github.com/htyao89/Textual-based_Class-aware_prompt_tuning/

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

↑