arXiv:2512.20145cs.CLcs.AI2025-12被引 1

用检索增强提示学习,让大模型少靠死记、多靠推理。

Retrieval-augmented Prompt Learning for Pre-trained Foundation Models

  • 引入外部知识库,动态检索上下文信息补全提示
  • 在零样本和少样本任务中显著提升泛化能力
  • 适合需要强泛化、避免过拟合的复杂场景

预训练基础模型(PFMs)已成为大规模多模态学习的核心。研究者通过“预训练、提示、预测”范式利用提示学习提升了少样本性能。然而,现有提示学习方法仍依赖参数化学习,导致记忆与泛化之间的稳定性下降。传统方法在有限数据下难以充分利用异常样本,易陷入浅层模式过拟合。为此,我们提出RetroPrompt,通过解耦知识与单纯记忆,实现记忆与泛化的平衡。不同于传统方法,RetroPrompt利用公开训练数据构建的知识库,在输入、训练和推理阶段均引入检索机制,使模型能主动从语料库中获取相关上下文信息,增强可用线索。我们在自然语言处理和计算机视觉多个数据集上进行实验,结果表明RetroPrompt在零样本和少样本场景下均表现更优。通过分析记忆模式发现,RetroPrompt有效降低了对机械记忆的依赖,显著提升了泛化能力。

原文摘要 · Abstract (English)

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.

提示学习检索增强泛化能力

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。