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

基于检索的提示学习用于预训练基础模型

Retrieval-augmented Prompt Learning for Pre-trained Foundation Models

MIIT Key Laboratory of Pattern Analysis and Machine Intelligence, College of Computer Science and Technology, Nanjing University of Aeronautics and Astronautics(信息产业部模式分析与机器智能重点实验室,计算机科学与技术学院,南京航空航天大学);Zhejiang University(浙江大学);Hunan Vanguard Group Corporation Limited(湖南先锋集团有限公司);National University of Singapore(新加坡国立大学)

Xiang Chen, Yixin Ou, Quan Feng, Lei Li, Piji Li, Haibo Ye, Sheng-Jun Huang, Shuofei Qiao, Shumin Deng, Huajun Chen, Ningyu Zhang

AI总结:

RetroPrompt通过整合检索机制和知识库,提升预训练基础模型在少样本和零样本场景下的泛化能力与记忆平衡。

摘要:

预训练基础模型(PFMs)已成为促进大规模多模态学习的关键。研究人员通过提示学习有效地采用了“预训练、提示、预测”范式,以诱导改进的少样本性能。然而,针对PFMs的提示学习方法仍然遵循参数学习范式。因此,记忆和机械学习的泛化稳定性可能会受到破坏。更具体地说,传统提示学习可能在完全监督训练过程中面临难以充分利用非典型实例和避免过拟合浅层模式的困难,尤其是在数据有限的情况下。为克服这些限制,我们提出了我们的方法RetroPrompt,旨在通过解耦知识与单纯记忆来实现记忆与泛化的平衡。与传统提示方法不同,RetroPrompt利用从训练数据生成的公开知识库,并在整个输入、训练和推理阶段中整合检索机制。这使模型能够主动从语料库中检索相关上下文信息,从而增强可用的提示。我们对自然语言处理和计算机视觉任务中的各种数据集进行了全面实验,以展示我们提出的方法RetroPrompt在零样本和少样本场景中的优越性能。通过详细分析记忆模式,我们观察到RetroPrompt有效减少了对机械记忆的依赖,从而提高了泛化能力。

英文摘要:

The pre-trained foundation models (PFMs) have become essential for facilitating large-scale multimodal learning. Researchers have effectively employed the ``pre-train, prompt, and predict'' paradigm through prompt learning to induce improved few-shot performance. However, prompt learning approaches for PFMs still follow a parametric learning paradigm. As such, the stability of generalization in memorization and rote learning can be compromised. More specifically, conventional prompt learning might face difficulties in fully utilizing atypical instances and avoiding overfitting to shallow patterns with limited data during the process of fully-supervised training. To overcome these constraints, we present our approach, named RetroPrompt, which aims to achieve a balance between memorization and generalization by decoupling knowledge from mere memorization. Unlike traditional prompting methods, RetroPrompt leverages a publicly accessible knowledge base generated from the training data and incorporates a retrieval mechanism throughout the input, training, and inference stages. This enables the model to actively retrieve relevant contextual information from the corpus, thereby enhancing the available cues. We conduct comprehensive experiments on a variety of datasets across natural language processing and computer vision tasks to demonstrate the superior performance of our proposed approach, RetroPrompt, in both zero-shot and few-shot scenarios. Through detailed analysis of memorization patterns, we observe that RetroPrompt effectively reduces the reliance on rote memorization, leading to enhanced generalization.

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

↑