通过难度估计动态分配采样预算,提升大模型自训练时的推理能力。
HS-STaR: Hierarchical Sampling for Self-Taught Reasoners via Difficulty Estimation and Budget Reallocation
- 基于奖励模型估算题目难度,识别模型能力边界处的问题
- 将采样预算动态分配给高价值问题,生成更优质训练数据
- 无需额外预算,显著提升多类推理任务表现
自训练推理者(STaRs)通过自生成回答实现大语言模型(LLMs)的数学推理能力增强。现有方法虽引入奖励模型指导响应选择或解码,但通常对所有问题采用均匀采样预算,忽视了不同难度问题的学习价值差异。本研究实证发现,位于模型推理能力边界的题目比简单题和过难题具有更高的学习效用。为此,我们提出HS-STaR——一种分层采样框架。在固定采样预算下,先以轻量级预采样结合奖励引导的难度估计策略,高效识别边界水平问题;随后在重采样阶段,将剩余预算动态分配至这些高价值问题,最大化高质量训练数据生成。跨多个推理基准与骨干模型的实验表明,HS-STaR显著优于其他基线,且无需增加采样预算。
原文摘要 · Abstract (English)
Self-taught reasoners (STaRs) enhance the mathematical reasoning abilities of large language models (LLMs) by leveraging self-generated responses for self-training. Recent studies have incorporated reward models to guide response selection or decoding, aiming to obtain higher-quality data. However, they typically allocate a uniform sampling budget across all problems, overlooking the varying utility of problems at different difficulty levels. In this work, we conduct an empirical study and find that problems near the boundary of the LLM's reasoning capability offer significantly greater learning utility than both easy and overly difficult ones. To identify and exploit such problems, we propose HS-STaR, a Hierarchical Sampling framework for Self-Taught Reasoners. Given a fixed sampling budget, HS-STaR first performs lightweight pre-sampling with a reward-guided difficulty estimation strategy to efficiently identify boundary-level problems. Subsequently, it dynamically reallocates the remaining budget toward these high-utility problems during a re-sampling phase, maximizing the generation of valuable training data. Extensive experiments across multiple reasoning benchmarks and backbone LLMs demonstrate that HS-STaR significantly outperforms other baselines without requiring additional sampling budget.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。