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

利用结构化语言知识为视觉-语言模型学习分层提示

Learning Hierarchical Prompt with Structured Linguistic Knowledge for Vision-Language Models

Tongji University(同济大学);Microsoft Research Asia(微软亚洲研究院);Xidian University(西安电子科技大学)

Yubin Wang, Xinyang Jiang, De Cheng, Dongsheng Li, Cairong Zhao

AI总结:

针对视觉-语言模型提示学习缺乏结构化知识建模的问题,本文提出 HPT,利用 LLM 构建类别描述图,并通过关系引导注意力与分层提示融合结构化和全局语义,显著提升泛化性能。

摘要:

提示学习已成为将视觉-语言基础模型适配到下游任务的一种主流策略。随着大型语言模型(LLM)的出现,近期研究探索了使用与类别相关的描述作为输入,以增强提示的有效性。然而,传统描述缺乏能够有效表示与特定类别相关的实体或属性之间相互联系的结构化信息。为解决这一局限并优先利用结构化知识,本文主张利用 LLM 为每条描述构建一个图,以建模描述该类别的实体和属性及其相互关系。既有的提示调优方法在处理这种结构化知识方面存在不足。因此,我们提出了一种名为 Hierarchical Prompt Tuning(HPT,分层提示调优)的新方法,可同时建模结构化语言知识和传统语言知识。具体而言,我们引入了一个关系引导注意力模块,用于捕获实体与属性之间的成对关联,以进行低层提示学习。此外,通过融合建模整体语义的高层和全局层提示,所提出的分层结构形成了跨层连接,使模型能够处理更复杂、更长期的关系。大量实验表明,我们的 HPT 表现出强大的有效性,并且比现有 SOTA 方法具有更好的泛化能力。我们的代码可在 https://github.com/Vill-Lab/2024-AAAI-HPT 获取。

英文摘要:

Prompt learning has become a prevalent strategy for adapting vision-language foundation models to downstream tasks. As large language models (LLMs) have emerged, recent studies have explored the use of category-related descriptions as input to enhance prompt effectiveness. Nevertheless, conventional descriptions fall short of structured information that effectively represents the interconnections among entities or attributes linked to a particular category. To address this limitation and prioritize harnessing structured knowledge, this paper advocates for leveraging LLMs to build a graph for each description to model the entities and attributes describing the category, as well as their correlations. Preexisting prompt tuning methods exhibit inadequacies in managing this structured knowledge. Consequently, we propose a novel approach called Hierarchical Prompt Tuning (HPT), which enables simultaneous modeling of both structured and conventional linguistic knowledge. Specifically, we introduce a relationship-guided attention module to capture pair-wise associations among entities and attributes for low-level prompt learning. In addition, by incorporating high-level and global-level prompts modeling overall semantics, the proposed hierarchical structure forges cross-level interlinks and empowers the model to handle more complex and long-term relationships. Extensive experiments demonstrate that our HPT shows strong effectiveness and generalizes much better than existing SOTA methods. Our code is available at https://github.com/Vill-Lab/2024-AAAI-HPT.

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

↑