arXiv:2506.05316cs.LGcs.AI2025-06NeurIPS被引 57

通过智能选题和重用推演,让大模型强化学习更省资源。

Improving Data Efficiency for LLM Reinforcement Fine-tuning Through Difficulty-targeted Online Data Selection and Rollout Replay

  • 按题目难易动态选题,聚焦有学习价值的问题
  • 减少23%至62%训练时间,性能持平原方法
  • 适合追求高效微调的大模型研究者

强化学习(RL)已成为提升大语言模型(LLM)推理能力的有效方法,但其微调过程仍高度依赖资源,现有工作对数据效率问题关注不足。本文提出两种提升数据效率的技术:难度导向的在线数据选择与推演重放。引入自适应难度概念,优先选择中等难度问题以获取更丰富的学习信号;通过注意力机制仅对少量参考问题进行推演,其余问题的难度基于相似性估算。为降低推演开销,提出受传统RL经验回放启发的推演重放机制,复用近期推演结果,在保持稳定更新的同时减少每步计算。在6个LLM-数据集组合上的实验表明,该方法将强化学习微调时间减少23%至62%,且性能与原始GRPO算法相当。代码已开源。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has become an effective approach for fine-tuning large language models (LLMs), particularly to enhance their reasoning capabilities. However, RL fine-tuning remains highly resource-intensive, and existing work has largely overlooked the problem of data efficiency. In this paper, we propose two techniques to improve data efficiency in LLM RL fine-tuning: difficulty-targeted online data selection and rollout replay. We introduce the notion of adaptive difficulty to guide online data selection, prioritizing questions of moderate difficulty that are more likely to yield informative learning signals. To estimate adaptive difficulty efficiently, we develop an attention-based framework that requires rollouts for only a small reference set of questions. The adaptive difficulty of the remaining questions is then estimated based on their similarity to this set. To further reduce rollout cost, we introduce a rollout replay mechanism inspired by experience replay in traditional RL. This technique reuses recent rollouts, lowering per-step computation while maintaining stable updates. Experiments across 6 LLM-dataset combinations show that our method reduces RL fine-tuning time by 23% to 62% while reaching the same level of performance as the original GRPO algorithm. Our code is available at https://github.com/ASTRAL-Group/data-efficient-llm-rl.

强化学习大模型微调数据效率推理优化

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