arXiv:2512.13106cs.LGcs.AI2025-12被引 5

用少量标注数据提升大模型推理能力,训练更高效更稳定。

TraPO: A Semi-Supervised Reinforcement Learning Framework for Boosting LLM Reasoning

  • 通过匹配学习轨迹相似性筛选可靠无标签样本
  • 仅用1000个标注样本达42.6%平均准确率,优于用4.5万无标签样本的无监督方法
  • 适合资源有限但需强推理能力的场景,如小样本数学推理

强化学习结合可验证奖励(RLVR)已被证明能有效训练大模型推理能力,但依赖高成本标注。现有无监督方法仅靠模型内部一致性(如熵或多数投票)生成奖励,易在训练后期出现模型坍塌,因缺乏外部监督而强化错误推理模式。本文提出半监督RLVR框架TraPO,利用少量标注数据引导无标签样本的训练。核心思想是:标注数据提供的监督信号对稳定基于一致性的训练至关重要,确保仅在标注样本上被验证的推理模式才进入强化学习。技术上,提出TraPO算法,通过比较无标签样本与标注样本的学习轨迹相似性识别可信样本。实验显示,仅用1000个标注和3000个无标签样本,即可在六项主流数学推理基准(AIME24/25、AMC、MATH-500、Minerva、Olympiad)及三项分布外任务(ARC-c、GPQA-diamond、MMLU-pro)上达到42.6%平均准确率,超越使用4.5万无标签样本的最优无监督方法(38.3%)。当使用4000个标注和1.2万无标签样本时,甚至优于全监督模型(使用4.5万标注样本),且仅需10%标注数据。代码已开源。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) has proven effective in training large reasoning models (LRMs) by leveraging answer-verifiable signals to guide policy optimization, which, however, suffers from high annotation costs. To alleviate this problem, recent work has explored unsupervised RLVR methods that derive rewards solely from the model's internal consistency, such as through entropy and majority voting. While seemingly promising, these methods often suffer from model collapse in the later stages of training, which may arise from the reinforcement of incorrect reasoning patterns in the absence of external supervision. In this work, we investigate a novel semi-supervised RLVR paradigm that utilizes a small labeled set to guide RLVR training on unlabeled samples. Our key insight is that supervised rewards are essential for stabilizing consistency-based training on unlabeled samples, ensuring that only reasoning patterns verified on labeled instances are incorporated into RL training. Technically, we propose an effective policy optimization algorithm, TraPO, that identifies reliable unlabeled samples by matching their learning trajectory similarity to labeled ones. Building on this, TraPO achieves remarkable data efficiency and strong generalization on six widely used mathematical reasoning benchmarks (AIME24/25, AMC, MATH-500, Minerva, and Olympiad) and three out-of-distribution tasks (ARC-c, GPQA-diamond, and MMLU-pro). With only 1K labeled and 3K unlabeled samples, TraPO reaches 42.6% average accuracy, surpassing the best unsupervised method trained on 45K unlabeled samples (38.3%). Notably, when using 4K labeled and 12K unlabeled samples, TraPO even outperforms the fully supervised model trained on the full 45K labeled samples on all benchmarks, while using only 10% of the labeled data. The code is available via https://github.com/ShenzhiYang2000/TRAPO.

强化学习半监督大模型推理数学题

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