用自适应选材让大模型用极少数据高效练出强推理能力
SPaCe: Unlocking Sample-Efficient Large Language Models Training With Self-Pace Curriculum Learning
- 根据模型能力动态选难易适中的训练数据,避免无效消耗
- 在多个推理任务上仅用1/100样本即达顶尖模型水平
- 适合资源有限但需强推理的大模型训练场景
大语言模型在强化学习微调后展现出强大推理能力,但此类方法依赖海量数据与算力,在现实训练预算下不切实际。现有流程均匀采样训练数据,忽略难度、冗余与学习价值差异,导致学习缓慢且计算浪费。本文提出SPaCe框架,通过优化数据选择时机与内容实现高效学习:首先采用基于语义和难度的聚类数据压缩,提取紧凑多样子集以减少冗余;随后利用多臂老虎机将数据聚类视为“手臂”,根据模型求解率与学习进度动态分配样本。跨多个推理基准的实验表明,SPaCe在使用最多减少100倍样本的情况下,达到或超越当前最优基线精度。消融实验与分析进一步凸显数据聚类与自适应选择的重要性。结果证明,精心设计、性能驱动的训练课程可使大模型以极小资源实现强推理能力。
原文摘要 · Abstract (English)
Large language models (LLMs) have shown strong reasoning capabilities when fine-tuned with reinforcement learning (RL). However, such methods require extensive data and compute, making them impractical under many realistic training budgets. Many existing pipelines sample training examples uniformly across steps or epochs, ignoring differences in difficulty, redundancy, and learning value, which slows learning and wastes computation. We propose \textbf{SPaCe}, a self-paced learning framework that enables efficient learning based on the capability of the model being trained through optimizing which data to use and when. First, we apply \emph{cluster-based data reduction} to partition training data by semantics and difficulty, extracting a compact yet diverse subset that reduces redundancy. Then, a \textit{multi-armed bandit} treats data clusters as arms, allocating training samples based on the model's solve rates and learning progress. Experiments across multiple reasoning benchmarks show that SPaCe achieves comparable or better accuracy than state-of-the-art baselines while using up to \(100\times\) fewer samples. Ablation studies and analyses further highlight the importance of both data clustering and adaptive selection. Our results demonstrate that carefully curated, performance-driven training curricula can unlock strong reasoning abilities in LLMs with minimal resources.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。