arXiv:2510.02172cs.CL2025-10被引 11

无需标注数据,模型自罚纠错提升推理能力

RESTRAIN: From Spurious Votes to Signals -- Self-Driven RL with Self-Penalization

  • 通过自惩罚机制,从无标签数据中提取有效信号
  • 在多个难题上性能提升超140%,接近有标注训练效果
  • 适合想低成本提升大模型推理能力的研究者

基于人类标注数据的强化学习虽提升了大模型的链式思维推理能力,但代价高昂且在难题上表现不佳。下一步自然方向是经验驱动学习,即模型在无标注数据中自我改进。我们提出RESTRAIN(REinforcement learning with Self-restraint),一种自惩罚强化学习框架,将缺乏真实标签转化为有用的学习信号。不同于盲目信任多数投票,RESTRAIN利用模型答案分布的整体信息:惩罚高置信度的错误推演和一致性低的样本,同时保留有潜力的推理路径。该自惩罚机制可无缝集成到GRPO等策略优化方法中,实现无监督下的持续自我提升。在多个挑战性推理基准测试中,使用Qwen3-4B-Base和OctoThinker Hybrid-8B-Base模型,RESTRAIN在AIME25上使Pass@1提升达+140.7%,MMLU_STEM上+36.2%,GPQA-Diamond上+19.6%,几乎达到有标注训练效果,而全程未使用任何黄金标签。这证明RESTRAIN为无需黄金标签的更强推理提供了可扩展路径。

原文摘要 · Abstract (English)

Reinforcement learning with human-annotated data has boosted chain-of-thought reasoning in large reasoning models, but these gains come at high costs in labeled data while faltering on harder tasks. A natural next step is experience-driven learning, where models improve without curated labels by adapting to unlabeled data. We introduce RESTRAIN (REinforcement learning with Self-restraint), a self-penalizing RL framework that converts the absence of gold labels into a useful learning signal. Instead of overcommitting to spurious majority votes, RESTRAIN exploits signals from the model's entire answer distribution: penalizing overconfident rollouts and low-consistency examples while preserving promising reasoning chains. The self-penalization mechanism integrates seamlessly into policy optimization methods such as GRPO, enabling continual self-improvement without supervision. On challenging reasoning benchmarks, RESTRAIN delivers large gains using only unlabeled data. With Qwen3-4B-Base and OctoThinker Hybrid-8B-Base, it improves Pass@1 by up to +140.7 percent on AIME25, +36.2 percent on MMLU_STEM, and +19.6 percent on GPQA-Diamond, nearly matching gold-label training while using no gold labels. These results demonstrate that RESTRAIN establishes a scalable path toward stronger reasoning without gold labels.

强化学习自监督推理增强无监督训练

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