arXiv:2411.16579cs.CLcs.AI2024-11被引 54

用批判模型在训练和推理时双重监督,提升大模型的数学推理能力。

Enhancing LLM Reasoning via Critique Models with Test-Time and Training-Time Supervision

  • 分离推理与批判模块,批判模型提供逐步反馈
  • 构建7.6万条带步骤反馈的数据集,支持自然语言批评生成
  • 提升复杂问题求解效率与多样性,适合需要深度思考的任务

让大语言模型在回答复杂推理问题前进行更深入的思考,对科学、编程和数学等领域至关重要。然而,自我反思与纠错的有效性依赖于模型对自身表现的准确评估,而初始准确率、题目难度及缺乏外部反馈等因素会限制其能力。本文提出一种双角色范式:将推理模型(演员)与批判模型分离,由批判模型在训练和推理阶段提供逐步反馈。我们构建了AutoMathCritique框架,自动收集76,321条响应及其对应的步骤级反馈数据。基于该数据微调的语言模型可生成自然语言形式的数学推理批评。实验表明,批判模型能显著提升演员在困难问题上的表现,尤其在增加推理计算量时效果更明显。受此启发,我们将批判监督引入演员的自训练过程,提出‘批判循环’自改进方法。结果表明,该方法提升了演员在复杂问题上的探索效率与解法多样性,进而训练出更强的推理模型。最后,我们初步探索了通过批判监督训练自言自语式推理模型的潜力。代码与数据集见:https://mathcritique.github.io/

原文摘要 · Abstract (English)

Training large language models (LLMs) to spend more time thinking and reflection before responding is crucial for effectively solving complex reasoning tasks in fields such as science, coding, and mathematics. However, the effectiveness of mechanisms like self-reflection and self-correction depends on the model's capacity to accurately assess its own performance, which can be limited by factors such as initial accuracy, question difficulty, and the lack of external feedback. In this paper, we delve into a two-player paradigm that separates the roles of reasoning and critique models, where the critique model provides step-level feedback to supervise the reasoning (actor) model during both test-time and train-time. We first propose AutoMathCritique, an automated and scalable framework for collecting critique data, resulting in a dataset of $76,321$ responses paired with step-level feedback. Fine-tuning language models with this dataset enables them to generate natural language feedback for mathematical reasoning. We demonstrate that the critique models consistently improve the actor's performance on difficult queries at test-time, especially when scaling up inference-time computation. Motivated by these findings, we introduce the critique-based supervision to the actor's self-training process, and propose a critique-in-the-loop self-improvement method. Experiments show that the method improves the actor's exploration efficiency and solution diversity, especially on challenging queries, leading to a stronger reasoning model. Lastly, we take the preliminary step to explore training self-talk reasoning models via critique supervision and showcase its potential. Our code and datasets are at \href{https://mathcritique.github.io/}{https://mathcritique.github.io/}.

大模型推理批判机制自训练数学推理

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