arXiv:2501.04070cs.LGcs.AI2025-01ACL被引 12

提升大模型多示例提示学习效果,解决示例增多时性能下降问题

More is not always better? Enhancing Many-Shot In-Context Learning with Differentiated and Reweighting Objectives

  • 采用差异化与动态加权策略优化多示例学习目标
  • 在50个任务上验证,多示例性能显著超越零样本
  • 适合研究大模型提示学习与数据噪声处理的学者

大语言模型在少样本提示学习(ICL)中表现优异,无需参数更新。但当示例数量从少量增至大量时,性能趋于平缓甚至下降。我们发现主要原因是负对数似然(NLL)优化目标不优及数据噪声累积。为此提出DrICL方法,通过差异化学习优化全局NLL目标,确保多示例性能超过零样本;同时利用强化学习启发的累积优势,动态调整示例权重,缓解噪声影响。为填补多任务多示例分布数据集的空白,构建了包含50个任务、覆盖1至350个示例(最长8,000词元)的大型基准ICL-50,支持微调与评估。实验表明,使用DrICL增强的模型在多种任务(含域内与域外)中多示例表现显著提升。代码与数据已开源。

原文摘要 · Abstract (English)

Large language models (LLMs) excel at few-shot in-context learning (ICL) without requiring parameter updates. However, as ICL demonstrations increase from a few to many, performance tends to plateau and eventually decline. We identify two primary causes for this trend: the suboptimal negative log-likelihood (NLL) optimization objective and the incremental data noise. To address these issues, we introduce \textit{DrICL}, a novel optimization method that enhances model performance through \textit{Differentiated} and \textit{Reweighting} objectives. Globally, DrICL utilizes differentiated learning to optimize the NLL objective, ensuring that many-shot performance surpasses zero-shot levels. Locally, it dynamically adjusts the weighting of many-shot demonstrations by leveraging cumulative advantages inspired by reinforcement learning, thereby mitigating the impact of noisy data. Recognizing the lack of multi-task datasets with diverse many-shot distributions, we develop the \textit{Many-Shot ICL Benchmark} (ICL-50)-a large-scale benchmark of 50 tasks that cover shot numbers from 1 to 350 within sequences of up to 8,000 tokens-for both fine-tuning and evaluation purposes. Experimental results demonstrate that LLMs enhanced with DrICL achieve significant improvements in many-shot setups across various tasks, including both in-domain and out-of-domain scenarios. We release the code and dataset hoping to facilitate further research in many-shot ICL\footnote{https://github.com/xiaoqzhwhu/DrICL}.

提示学习大模型多示例

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