arXiv:2506.06033cs.CL2025-06ICML被引 4

用智能筛选减少大模型演示数据量,提升效率不降性能。

Large Language Models are Demonstration Pre-Selectors for Themselves

  • 预选机制基于充分性与必要性指标,树状算法高效找出关键演示样本。
  • 数据量压缩超20%后,少样本学习与微调性能基本不变。
  • 适合需要高效推理的场景,尤其适用于资源受限的大模型应用。

在上下文学习中,大语言模型通过从海量训练数据中选取少量示范样本实现强少样本性能。然而,现有方法依赖相似性或多样性评分频繁检索大规模数据集,计算开销高。为此,本文提出FEEDER(Few yet Essential Demonstration pre-selector)框架,针对特定大模型识别出包含最具代表性示例的精炼示范子集。通过引入‘充分性’与‘必要性’度量,并设计树状算法,高效筛选核心样本。预选后的子集可替代完整训练数据,在保持上下文学习性能的同时显著提升效率。此外,该子集亦有助于模型微调,我们提出双层优化方法,在不牺牲性能的前提下提高训练效率。实验覆盖300M至8B参数的多类大模型,结果表明FEEDER可将训练数据规模压缩超过20%,且能无缝集成到多种下游示范选择策略中。

原文摘要 · Abstract (English)

In-context learning (ICL) with large language models (LLMs) delivers strong few-shot performance by choosing few-shot demonstrations from the entire training data. However, existing ICL methods, which rely on similarity or diversity scores to choose demonstrations, incur high computational costs due to repeatedly retrieval from large-scale datasets for each query. To this end, we propose FEEDER (FEw yet Essential Demonstration prE-selectoR), a novel pre-selection framework that identifies a representative subset of demonstrations containing the most representative examples in the training data, tailored to specific LLMs. To construct this subset, we introduce the "sufficiency" and "necessity" metrics in the pre-selection stage and design a tree-based algorithm to identify representative examples efficiently. Once pre-selected, this representative subset can effectively replace the full training data, improving efficiency while maintaining comparable performance in ICL. Additionally, our pre-selected subset also benefits fine-tuning LLMs, where we introduce a bi-level optimization method that enhances training efficiency without sacrificing performance. Experiments with LLMs ranging from 300M to 8B parameters show that FEEDER can reduce training data size by over 20% while maintaining performance and seamlessly integrating with various downstream demonstration selection strategies in ICL.

大模型少样本学习数据压缩预筛选

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