arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2312.02191cs.CVcs.AI
收藏

面向零样本组合学习的提示调优

Prompt Tuning for Zero-shot Compositional Learning

Samsung Research America(三星美国研究院)

Lingyu Zhang, Ting Hua, Yilin Shen, Hongxia Jin

AI总结:

针对开放世界组合零样本学习缺乏开放世界常识的问题,本文提出 MMPT 框架,通过多模态提示调优继承大型预训练视觉-语言模型知识,并在 UT-Zappos 和 MIT-States 上刷新最先进性能。

摘要:

开放世界组合零样本学习(OW-CZSL)被公认为一项极具挑战性的任务,其目标是在不对输出空间作任何先验假设的情况下,识别由已见属性和物体构成的未见组合。为实现这一目标,模型必须既“聪明”又“有知识”。所谓“聪明”,是指模型应善于从已见组合中推断属性与物体之间的交互;而“有知识”则意味着模型拥有面向开放世界的“常识”,能够“预见”未见组合的某些特征。以往大多数工作聚焦于“聪明”这一部分,却鲜有工作为实现“有知识”的目标提供有效方案。本文提出了一个名为 Multi-Modal Prompt Tuning(MMPT,多模态提示调优)的框架,以从大型预训练视觉-语言模型中继承“有知识”的特性。大量实验表明,所提出的 MMPT 在 OW-CZSL 任务上取得了新的最先进结果。在 UT-Zappos 数据集上,MMPT 将 AUC 分数提升至 29.8,而此前最佳分数为 26.5。在更具挑战性的 MIT-States 数据集上,MMPT 的 AUC 分数比当前最先进方法好 1.5 倍。

英文摘要:

Open World Compositional Zero-Shot Learning (OW-CZSL) is known to be an extremely challenging task, which aims to recognize unseen compositions formed from seen attributes and objects without any prior assumption of the output space. In order to achieve this goal, a model has to be "smart" and "knowledgeable". To be smart, a model should be good at reasoning the interactions between attributes and objects from the seen compositions. While "knowledgeable" means the model owns "common sense" to the open world that can "foresee" some features of the unseen compositions. Most previous work focuses on the "smart" part, while few of them provided an effective solution to achieve the "knowledgeable" goal. In this paper, we proposed a framework named Multi-Modal Prompt Tuning (MMPT) to inherit the "knowledgeable" property from the large pre-trained vision-language model. Extensive experiments show that our proposed MMPT obtains new state-of-the-art results in OW-CZSL task. On the UT-Zappos dataset, MMPT pushes the AUC score to $29.8$, while the previous best score is $26.5$. On the more challenging MIT-States dataset, the AUC score of MMPT is 1.5 times better than the current state-of-the-art.

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

↑