arXiv:2501.18858cs.LGcs.AI2025-01ICML被引 11

用强化学习提升大模型推理能力,无需人工标注。

BRiTE: Bootstrapping Reinforced Thinking Process to Enhance Language Model Reasoning

  • 构建概率图模型,用强化学习优化思考过程。
  • 在数学和编码任务上超越已有方法,接近人工标注效果。
  • 适合想提升模型推理能力但无标注数据的研究者。

大语言模型在复杂推理任务中表现优异,但生成可靠推理过程仍是挑战。本文提出统一的概率框架,通过包含隐式思考过程与评估信号的新型图模型形式化建模。在此框架下,提出自举强化思考过程(BRiTE)算法,分两步:首先通过强化学习近似最优思考路径,采用新颖奖励设计;其次通过最大化推理过程与模型参数的联合概率来增强基座模型。理论上证明其收敛速率为 $1/T$($T$ 为迭代次数)。在数学与编程基准测试中,该方法在不同基座模型上持续提升性能,且无需人类标注的推理过程。相比基于拒绝采样的现有方法,表现更优,甚至可媲美或超过使用人工标注数据的监督微调结果。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have demonstrated remarkable capabilities in complex reasoning tasks, yet generating reliable reasoning processes remains a significant challenge. We present a unified probabilistic framework that formalizes LLM reasoning through a novel graphical model incorporating latent thinking processes and evaluation signals. Within this framework, we introduce the Bootstrapping Reinforced Thinking Process (BRiTE) algorithm, which works in two steps. First, it generates high-quality rationales by approximating the optimal thinking process through reinforcement learning, using a novel reward shaping mechanism. Second, it enhances the base LLM by maximizing the joint probability of rationale generation with respect to the model's parameters. Theoretically, we demonstrate BRiTE's convergence at a rate of $1/T$ with $T$ representing the number of iterations. Empirical evaluations on math and coding benchmarks demonstrate that our approach consistently improves performance across different base models without requiring human-annotated thinking processes. In addition, BRiTE demonstrates superior performance compared to existing algorithms that bootstrap thinking processes use alternative methods such as rejection sampling, and can even match or exceed the results achieved through supervised fine-tuning with human-annotated data.

推理增强强化学习自举训练

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