arXiv:2510.01135cs.LGcs.CL2025-10被引 36

用智能提示筛选提升大模型微调效率,速度最高快16倍

Prompt Curriculum Learning for Efficient LLM Post-Training

  • 基于价值模型动态选中等难度提示,实现在线优化
  • 在MATH和DeepScaleR上提速12.1倍至16.9倍,性能不降
  • 适合追求高效推理训练的开发者和研究者

我们提出提示课程学习(PCL),一种轻量级强化学习算法,通过学习到的价值模型选择中等难度提示来微调语言模型。由于强化学习微调对批量大小和提示选择策略敏感,我们系统实验发现:(1) 存在平衡生成效率与梯度质量的最优批大小;(2) 聚焦中等难度提示对策略至关重要。基于此,设计PCL,利用随策略同步更新的价值模型,在策略内识别中等难度提示。通过聚焦高有效率提示,PCL达到最高性能或显著更早达到相当性能。相比基于回放的筛选方法,PCL避免昂贵回放,训练时在MATH和DeepScaleR上分别提速12.1倍和16.9倍。进一步证明,该价值模型准确预测提示难度,使PCL在强化学习过程中逐步聚焦更难提示。结果展示了一种在上限性能与效率间取得更好权衡的新方法,适用于以推理为导向的强化学习。

原文摘要 · Abstract (English)

We introduce Prompt Curriculum Learning (PCL), a lightweight reinforcement learning (RL) algorithm that selects intermediate-difficulty prompts using a learned value model to post-train language models. Since post-training LLMs via RL remains sensitive to batching and prompt selection strategies, we first conduct a series of systematic experiments where we (1) determine the optimal training batch size that balances generation efficiency and gradient quality and (2) establish the importance of focusing on prompts of intermediate difficulty for the policy. We build upon these results to design PCL, which identifies prompts of intermediate difficulty for the current policy in an on-policy manner by using a value model that is concurrently updated based on the current policy. By focusing on informative prompts that yield high effective ratios, PCL achieves either the highest performance or requires significantly less time to reach comparable performance to its counterparts. Compared to rollout-based filtering methods, PCL avoids costly rollouts and achieves $12.1\times$ and $16.9\times$ faster speed on identifying intermediate-difficulty prompts when training on MATH and DeepScaleR, respectively. We further demonstrate that our value model accurately predicts prompt difficulty and allows PCL to focus on progressively more challenging prompts during RL. Our results present a new methodology that delivers improved tradeoff between upper-bound performance and efficiency for reasoning-focused RL.

强化学习大模型微调提示工程效率优化

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