让大模型强化学习更高效:动态调整问题池,精准挑难题训练。
HeaPA: Difficulty-Aware Heap Sampling and On-Policy Query Augmentation for LLM Reinforcement Learning
- 用堆结构动态筛选难题,实时跟踪模型能力边界。
- 通过在线增补新问题,在不增加教师成本下提升训练效率。
- 适合大规模模型的推理任务训练,尤其在中大型模型上效果显著。
强化学习用于训练大语言模型解决可验证推理任务已成为标准方法,但当生成回滚(rollout)成为主要开销时,采样策略直接影响效率。现有方法多依赖固定问题池,难以支持稳定的问题池增长,或引入额外教师模型带来的成本与延迟。本文提出HeaPA(堆采样与在线查询增强),维护一个有限且持续演进的问题池,利用堆结构进行边界采样以追踪能力前沿,通过轻量级异步验证下的在线策略增补扩展问题池,并基于拓扑感知的统计重估与可控重插入稳定相关查询。在两个训练语料、两种训练范式和七个基准测试中,HeaPA始终提升准确率,以更少计算量达到目标性能,且耗时相当。分析表明,优势源于聚焦前沿的采样与在线问题池增长,尤其在中大型模型上表现更优。代码已公开于https://github.com/horizon-llm/HeaPA。
原文摘要 · Abstract (English)
RLVR has become a standard recipe for training LLMs on reasoning tasks with verifiable outcomes, but when rollout generation dominates the cost, efficiency hinges on which prompts are sampled and when. In practice, prompt pools are often static or only weakly coupled to policy progress, so uniform sampling fails to track the moving capability frontier and wastes rollouts on regions that are already solved or still unreachable. Prior methods improve efficiency via filtering, curricula, adaptive rollout allocation, or teacher guidance, but they often assume a fixed pool, which does not support stable on-policy pool growth, or they introduce additional teacher cost and latency. In this work, we propose HeaPA (Heap Sampling and On-Policy Query Augmentation), which maintains a bounded, evolving pool, tracks the frontier with heap-based boundary sampling, grows the pool via on-policy augmentation under lightweight asynchronous validation, and stabilizes correlated queries via topology-aware pool statistics re-estimation and controlled reinsertion. Across two training corpora, two training recipes, and seven benchmarks, HeaPA consistently improves accuracy and reaches target performance with fewer computations at comparable wall-clock time. Analyses attribute the gains to frontier-focused sampling and on-policy pool growth, with more pronounced improvements at mid-to-large model scales. Our training code is publicly available at https://github.com/horizon-llm/HeaPA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。