arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2303.09100cs.CVcs.CLcs.LG
收藏

面向视觉-语言模型的Patch-Prompt对齐贝叶斯提示调优

Patch-Prompt Aligned Bayesian Prompt Tuning for Vision-Language Models

National Key Laboratory of Radar Signal Processing, Xidian University(西安电子科技大学雷达信号处理国家重点实验室);McCombs School of Business, The University of Texas at Austin(德克萨斯大学奥斯汀分校麦库姆斯商学院)

Xinyang Liu, Dongsheng Wang, Bowei Fang, Miaoge Li, Zhibin Duan, Yishi Xu, Bo Chen, Mingyuan Zhou

AI总结:

本文提出Patch-Prompt对齐贝叶斯提示调优方法,通过分层随机提示生成和视觉-语言统计距离正则化,提升视觉-语言模型在少样本识别、泛化及迁移等任务中的表现。

摘要:

对于视觉-语言预训练模型的下游应用,构建有效的提示(prompt)引起了广泛关注。现有的提示工程工作要么需要耗费大量人工设计,要么将提示调优作为点估计问题进行优化,可能无法描述类别的多样化特征,从而限制了其应用。我们提出了一种贝叶斯概率方法来解决提示调优问题,其中标签特定的随机提示通过分层方式生成:首先从底层分布中采样一个潜在向量,然后使用一个轻量级生成模型。重要的是,我们通过最小化视觉patch与语言提示之间的统计距离来对调优过程进行语义正则化,从而推动随机标签表示忠实地捕获多样化的视觉概念,而不是过拟合训练类别。我们在四个任务上评估了方法的有效性:少样本图像识别、基类到新类泛化、数据集迁移学习和领域偏移。在15个数据集上的大量结果表明,我们提出的模型在迁移性和泛化性能上均表现出良好的前景,无论是定量还是定性评估。

英文摘要:

For downstream applications of vision-language pre-trained models, there has been significant interest in constructing effective prompts. Existing works on prompt engineering, which either require laborious manual designs or optimize the prompt tuning as a point estimation problem, may fail to describe diverse characteristics of categories and limit their applications. We introduce a Bayesian probabilistic resolution to prompt tuning, where the label-specific stochastic prompts are generated hierarchically by first sampling a latent vector from an underlying distribution and then employing a lightweight generative model. Importantly, we semantically regularize the tuning process by minimizing the statistical distance between the visual patches and linguistic prompts, which pushes the stochastic label representations to faithfully capture diverse visual concepts, instead of overfitting the training categories. We evaluate the effectiveness of our approach on four tasks: few-shot image recognition, base-to-new generalization, dataset transfer learning, and domain shifts. Extensive results over 15 datasets show promising transferability and generalization performance of our proposed model, both quantitatively and qualitatively.

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

↑