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

多样性感知的Meta视觉提示

Diversity-Aware Meta Visual Prompting

University of Science and Technology of China(中国科学技术大学);Microsoft Cloud AI(微软云人工智能部门);Wormpex AI Research(Wormpex人工智能研究院)

Qidong Huang, Xiaoyi Dong, Dongdong Chen, Weiming Zhang, Feifei Wang, Gang Hua, Nenghai Yu

摘要:

提出了Diversity-Aware Meta Visual Proximating ~(DAM-VP),它是一种高效的提示方法,用于将预先训练好的模型转移到具有冻结主干的下游任务中。视觉提示中的一个具有挑战性的问题是,图像数据集有时具有很大的数据多样性,而每个数据集的通用提示很难处理复杂的分布偏移向原始预训练数据分布正确。为了解决这个问题,我们提出了一个数据集的多样性意识的提示策略,其初始化是由一个元提示。具体来说,我们以多样性自适应的方式将下游数据集聚类成小的同质子集,每个子集都有自己的提示单独优化。这种分而治之的设计大大降低了优化的难度,并显着提高了提示性能。此外,所有提示都使用元提示进行初始化,该元提示是在多个数据集上学习的。这是一个自举的范例,关键的观察是,从以前的数据集学习的提示知识可以帮助提示更快地收敛,并在新的数据集上表现得更好。在推理过程中,我们根据输入和每个子集之间的特征距离为每个输入动态选择适当的提示。通过大量的实验,我们的DAM-VP表现出了卓越的效率和有效性,在不同预训练模型的一系列下游数据集中明显超过了以前的提示方法。我们的代码可从以下网址获得:\url{https://github.com/shikiw/DAM-VP}。

英文摘要:

We present Diversity-Aware Meta Visual Prompting~(DAM-VP), an efficient and effective prompting method for transferring pre-trained models to downstream tasks with frozen backbone. A challenging issue in visual prompting is that image datasets sometimes have a large data diversity whereas a per-dataset generic prompt can hardly handle the complex distribution shift toward the original pretraining data distribution properly. To address this issue, we propose a dataset Diversity-Aware prompting strategy whose initialization is realized by a Meta-prompt. Specifically, we cluster the downstream dataset into small homogeneity subsets in a diversity-adaptive way, with each subset has its own prompt optimized separately. Such a divide-and-conquer design reduces the optimization difficulty greatly and significantly boosts the prompting performance. Furthermore, all the prompts are initialized with a meta-prompt, which is learned across several datasets. It is a bootstrapped paradigm, with the key observation that the prompting knowledge learned from previous datasets could help the prompt to converge faster and perform better on a new dataset. During inference, we dynamically select a proper prompt for each input, based on the feature distance between the input and each subset. Through extensive experiments, our DAM-VP demonstrates superior efficiency and effectiveness, clearly surpassing previous prompting methods in a series of downstream datasets for different pretraining models. Our code is available at: \url{https://github.com/shikiw/DAM-VP}.

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

↑