提出模块化提示学习,提升视觉语言模型信息保留能力
Modular Prompt Learning Improves Vision-Language Models
- 将提示分模块插入各层,避免信息丢失
- 在11个数据集上平均提升0.7%,最高达10.7%
- 适合需高效迁移且保持原模型性能的场景
预训练视觉语言模型能理解视觉概念与语言语义。提示学习通过构建文本或图像编码器的提示,激发预训练模型潜力,并快速适配新任务。相比微调,提示学习用更少可训练参数实现相当或更优性能,且冻结预训练模型,避免灾难性遗忘。将连续提示插入每个Transformer层(深层提示)可提升下游任务表现,但第i层的提示会覆盖第(i-1)层的提示。尽管自注意力机制对当前层新提示进行上下文建模,但移除前层提示会导致其包含的信息丢失。本文提出模块化提示学习(MPL),旨在增强插入提示中信息的保留。我们在基础到新任务泛化和跨数据集任务上评估该方法,在11个数据集上的平均性能比当前最优方法提升0.7%,单个数据集最大提升达10.7%(EuroSAT)。
原文摘要 · Abstract (English)
Pre-trained vision-language models are able to interpret visual concepts and language semantics. Prompt learning, a method of constructing prompts for text encoders or image encoders, elicits the potentials of pre-trained models and readily adapts them to new scenarios. Compared to fine-tuning, prompt learning enables the model to achieve comparable or better performance using fewer trainable parameters. Besides, prompt learning freezes the pre-trained model and avoids the catastrophic forgetting issue in the fine-tuning. Continuous prompts inserted into the input of every transformer layer (i.e. deep prompts) can improve the performances of pre-trained models on downstream tasks. For i-th transformer layer, the inserted prompts replace previously inserted prompts in the $(i-1)$-th layer. Although the self-attention mechanism contextualizes newly inserted prompts for the current layer and embeddings from the previous layer's output, removing all inserted prompts from the previous layer inevitably loses information contained in the continuous prompts. In this work, we propose Modular Prompt Learning (MPL) that is designed to promote the preservation of information contained in the inserted prompts. We evaluate the proposed method on base-to-new generalization and cross-dataset tasks. On average of 11 datasets, our method achieves 0.7% performance gain on the base-to-new generalization task compared to the state-of-the-art method. The largest improvement on the individual dataset is 10.7% (EuroSAT dataset).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。