通过中间推理建模提升强化学习在可验证结果中的表现
Masked-and-Reordered Self-Supervision for Reinforcement Learning from Verifiable Rewards
- 用掩码填充和步骤重排构建过程级自监督信号
- 在固定预算下相对原RLVR提升9.86%的Pass@1
- 适合数学推理与定理证明等需长链思考任务
测试时扩展已被证明能显著提升大语言模型的数学推理能力。然而,对于大部分数学语料,尤其是定理证明任务,RLVR的可扩展性受限:中间推理至关重要,但最终答案难以直接且可靠地验证。同时,令牌级监督微调常退化为机械记忆而非生成长链思维。受BERT自监督任务启发,我们提出MR-RLVR(掩码与重排强化学习从可验证奖励),通过“掩码后填空”和“步骤重排”构建过程级自监督奖励,从中间推理中提取可学习信号。训练分两阶段:先在采样的数学计算与证明数据上进行自监督训练;再在仅结果可验证的数学计算数据集上进行RLVR微调。我们在Qwen2.5-3B和DeepSeek-R1-Distill-Qwen-1.5B上实现MR-RLVR,评估在AIME24、AIME25、AMC23和MATH500上。在固定采样与解码预算下,相对于原始RLVR,MR-RLVR在Pass@1上平均提升9.86%,Pass@5提升5.27%,Pass@8提升4.00%。结果表明,引入过程感知的自监督信号可有效增强RLVR在仅结果可验证场景下的可扩展性与性能。
原文摘要 · Abstract (English)
Test-time scaling has been shown to substantially improve large language models' (LLMs) mathematical reasoning. However, for a large portion of mathematical corpora, especially theorem proving, RLVR's scalability is limited: intermediate reasoning is crucial, while final answers are difficult to directly and reliably verify. Meanwhile, token-level SFT often degenerates into rote memorization rather than inducing longer chains of thought. Inspired by BERT's self-supervised tasks, we propose MR-RLVR (Masked-and-Reordered RLVR), which constructs process-level self-supervised rewards via "masked-then-fill" and "step reordering" to extract learnable signals from intermediate reasoning. Our training pipeline comprises two stages: we first perform self-supervised training on sampled mathematical calculation and proof data; we then conduct RLVR fine-tuning on mathematical calculation datasets where only outcomes are verifiable. We implement MR-RLVR on Qwen2.5-3B and DeepSeek-R1-Distill-Qwen-1.5B, and evaluate on AIME24, AIME25, AMC23, and MATH500. Under a fixed sampling and decoding budget, MR-RLVR achieves average relative gains over the original RLVR of +9.86% Pass@1, +5.27% Pass@5, and +4.00% Pass@8. These results indicate that incorporating process-aware self-supervised signals can effectively enhance RLVR's scalability and performance in only outcome-verifiable settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。