通过自改进提升大模型推理能力,解决数据不均与冗余思考问题。
Better, Faster: Harnessing Self-Improvement in Large Reasoning Models

- 设计验证后退出采样缓解简单样本过多问题
- 引入内在多样性评分,减少冗余推理步骤,性能最高提升10.9%
- 适合追求高效精准推理的AI研发人员
自改进训练使大型推理模型(LRMs)能自主生成推理轨迹作为训练数据,无需外部监督。然而我们发现该方法在复杂任务中常表现不佳,甚至导致模型崩溃。分析揭示两个问题:(1) 数据不平衡,多数样本为简单任务,而关键难题样本稀缺;(2) 过度思考,大量包含冗余步骤的不良样本被用于自训练。为此,我们提出HSIR,通过两种简单但有效的方法提升自改进效果:一是采用验证后退出采样策略,高效收集困难问题的准确解法以缓解数据不平衡;二是设计内在多样性分数,量化并过滤掉冗余推理样本。我们将HSIR应用于多种后训练范式,进一步提出H-GRPO——一种增强型GRPO算法,利用内在多样性作为外在奖励,通过强化学习鼓励简洁且多样化的推理。实验表明,HSIR不仅显著提升推理性能(平均性能最高提升10.9%),还大幅降低推理开销,最多减少42.4%相对延迟。
原文摘要 · Abstract (English)
Self-improvement training enables the large reasoning models (LRMs) to improve themselves by self-generating reasoning trajectories as training data without external supervision. However, we find that this method often falls short in complex reasoning tasks and even leads to model collapse. Through a series of preliminary analyses, we reveal two problems: (1) data imbalance, where most training samples are simple, but the challenging yet crucial samples are scarce; (2) overthinking, where many undesired samples with redundant reasoning steps are used for self-training. To this end, we propose HSIR, which effectively Harnesses Self-Improvement in large Reasoning models via two simple-yet-effective approaches. Specifically, HSIR introduces a verify-then-exit sampling strategy to mitigate data imbalance by efficiently collecting more accurate solutions for difficult queries, and designs an Intrinsic Diversity score to quantify overthinking and filter out the undesired solutions. We apply HSIR to various post-training paradigms, among which we further propose H-GRPO, an enhanced GRPO algorithm that leverages the intrinsic diversity as an external reward to encourage concise and diverse reasoning via reinforcement learning. Extensive results show that HSIR not only effectively enhances the reasoning performance, i.e., bringing up to +10.9% average performance gains, but also significantly improves the reasoning efficiency by reducing up to 42.4% relative inference overhead.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。