arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.20164cs.LGcs.CV
收藏

Prompt Diffusion 增强任意模态提示学习的鲁棒性

Prompt Diffusion Robustifies Any-Modality Prompt Learning

University of Amsterdam(阿姆斯特丹大学);Cisco Research(思科研究院);Illinois Institute of Technology(伊利诺伊理工学院);Michigan State University(密歇根州立大学)

Yingjun Du, Gaowen Liu, Yuzhang Shang, Yuguang Yao, Ramana Kompella, Cees G. M. Snoek

AI总结:

针对固定提示在分布偏移下泛化不足的问题,本文提出模态无关的 prompt diffusion 即插即用模块,在提示空间中五步生成样本定制提示,显著提升多种提示学习方法在 15 个数据集上的三类泛化鲁棒性。

摘要:

基础模型使基于提示的分类器能够进行零样本和少样本学习。然而,使用固定提示的传统方法会受到分布偏移的影响,这会对其在未见样本上的泛化能力产生不利影响。本文介绍了 prompt diffusion,它使用扩散模型逐步细化提示,从而为每个样本获得定制化提示。具体而言,我们首先优化一组提示,以获得每个样本的过拟合提示。随后,我们在提示空间中提出一个 prompt diffusion 模型,从而能够训练从随机提示到其过拟合提示的生成式转换过程。由于在推理期间无法获取测试图像的标签,我们的模型仅从随机提示出发,利用训练好的 prompt diffusion 逐步生成定制化提示。我们的 prompt diffusion 具有通用性、灵活性和模态无关性,使其成为一个简单的即插即用模块,可无缝嵌入现有的文本、视觉或多模态提示学习方法中。我们的扩散模型采用基于 ODE 的快速采样策略,仅需五步即可优化测试样本提示,在性能提升与计算效率之间取得良好平衡。对于所有测试的提示学习方法,加入 prompt diffusion 后,在涵盖 15 个多样化数据集的分类任务中,均在基类到新类泛化、跨数据集泛化和域泛化方面产生了更鲁棒的结果。

英文摘要:

Foundation models enable prompt-based classifiers for zero-shot and few-shot learning. Nonetheless, the conventional method of employing fixed prompts suffers from distributional shifts that negatively impact generalizability to unseen samples. This paper introduces prompt diffusion, which uses a diffusion model to gradually refine the prompts to obtain a customized prompt for each sample. Specifically, we first optimize a collection of prompts to obtain over-fitted prompts per sample. Then, we propose a prompt diffusion model within the prompt space, enabling the training of a generative transition process from a random prompt to its overfitted prompt. As we cannot access the label of a test image during inference, our model gradually generates customized prompts solely from random prompts using our trained, prompt diffusion. Our prompt diffusion is generic, flexible, and modality-agnostic, making it a simple plug-and-play module seamlessly embedded into existing prompt learning methods for textual, visual, or multi-modal prompt learning. Our diffusion model uses a fast ODE-based sampling strategy to optimize test sample prompts in just five steps, offering a good trade-off between performance improvement and computational efficiency. For all prompt learning methods tested, adding prompt diffusion yields more robust results for base-to-new generalization, cross-dataset generalization, and domain generalization in classification tasks tested over 15 diverse datasets.

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

↑