arXiv:2505.14970cs.AIcs.LG2025-05被引 111

让大模型自动设计学习路径,提升推理能力。

Self-Evolving Curriculum for LLM Reasoning

论文配图:Self-Evolving Curriculum for LLM Reasoning
图 1 · 摘自论文原文
  • 用强化学习动态选题,按难易程度分批训练。
  • 在数学、归纳推理等任务上显著提升准确率。
  • 适合多领域联合训练,避免偏科问题。

强化学习(RL)在微调大语言模型(LLM)方面已证明有效,显著提升了其在数学和代码生成等领域的推理能力。训练课程的安排——即问题呈现顺序——是影响微调效果的关键因素。虽然随机课程作为基准,但性能不佳;人工设计课程依赖经验,而在线筛选方法计算成本高。为此,我们提出自演化课程(SEC),一种在强化学习微调过程中同步学习课程策略的自动方法。将课程选择建模为非平稳多臂赌博机问题,每个问题类别(如难度等级或类型)视为一个独立“臂”。利用策略梯度方法的绝对优势作为即时学习收益的代理指标。每步训练中,课程策略通过最大化该奖励信号选择类别,并使用TD(0)方法更新。在规划、归纳推理和数学三个不同推理领域上的实验表明,SEC显著增强了模型的推理能力,使其对更难的分布外测试问题具备更强泛化性。此外,该方法在同时微调多个推理领域时表现出更好的技能均衡性。这些结果凸显了SEC作为大语言模型强化学习微调的有前景策略。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has proven effective for fine-tuning large language models (LLMs), significantly enhancing their reasoning abilities in domains such as mathematics and code generation. A crucial factor influencing RL fine-tuning success is the training curriculum: the order in which training problems are presented. While random curricula serve as common baselines, they remain suboptimal; manually designed curricula often rely heavily on heuristics, and online filtering methods can be computationally prohibitive. To address these limitations, we propose Self-Evolving Curriculum (SEC), an automatic curriculum learning method that learns a curriculum policy concurrently with the RL fine-tuning process. Our approach formulates curriculum selection as a non-stationary Multi-Armed Bandit problem, treating each problem category (e.g., difficulty level or problem type) as an individual arm. We leverage the absolute advantage from policy gradient methods as a proxy measure for immediate learning gain. At each training step, the curriculum policy selects categories to maximize this reward signal and is updated using the TD(0) method. Across three distinct reasoning domains: planning, inductive reasoning, and mathematics, our experiments demonstrate that SEC significantly improves models' reasoning capabilities, enabling better generalization to harder, out-of-distribution test problems. Additionally, our approach achieves better skill balance when fine-tuning simultaneously on multiple reasoning domains. These findings highlight SEC as a promising strategy for RL fine-tuning of LLMs.

强化学习课程学习大模型推理

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