arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2411.16579cs.CLcs.AIcs.LG

通过测试时与训练时监督的批判模型增强大语言模型推理

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

Fudan University(复旦大学);Meituan(美团)
Zhiheng Xi, Dingwen Yang, Jixuan Huang, Jiafu Tang, Guanyu Li, Yiwen Ding, Wei He, Boyang Hong, Shihan Do, Wenyu Zhan, Xiao Wang, Rui Zheng, Tao Ji, Xiaowei Shi… 展开作者

Zhiheng Xi, Dingwen Yang, Jixuan Huang, Jiafu Tang, Guanyu Li, Yiwen Ding, Wei He, Boyang Hong, Shihan Do, Wenyu Zhan, Xiao Wang, Rui Zheng, Tao Ji, Xiaowei Shi, Yitao Zhai, Rongxiang Weng, Jingang Wang, Xunliang Cai, Tao Gui, Zuxuan Wu, Qi Zhang, Xipeng Qiu, Xuanjing Huang, Yu-Gang Jiang

AI总结:

本文提出将推理模型与批判模型分离的双角色范式,通过AutoMathCritique框架构建步骤级反馈数据集,在测试时和训练时用批判监督提升大语言模型的数学推理能力与自我改进效果。

摘要:

训练大语言模型(LLMs)在回答前花更多时间思考和反思,对于有效解决科学、编程和数学等领域的复杂推理任务至关重要。然而,自我反思和自我纠正等机制的有效性取决于模型准确评估自身表现的能力,而这种能力可能受到初始准确率、问题难度以及缺乏外部反馈等因素的限制。在本文中,我们深入探讨了一种双角色范式,将推理模型与批判模型的角色分离,其中批判模型在测试时和训练时均提供步骤级反馈,以监督推理(行动者)模型。我们首先提出了AutoMathCritique,一个自动化且可扩展的批判数据收集框架,由此构建了一个包含76,321条响应及其步骤级反馈的数据集。使用该数据集微调语言模型,使其能够为数学推理生成自然语言反馈。我们证明,批判模型在测试时能够持续提升行动者模型在困难查询上的表现,尤其是在扩大推理时计算规模的情况下。受这些发现的启发,我们将基于批判的监督引入行动者模型的自我训练过程,并提出了一种批判在环的自我改进方法。实验表明,该方法提高了行动者模型的探索效率和解决方案多样性,尤其是在具有挑战性的查询上,从而得到更强的推理模型。最后,我们初步探索了通过批判监督训练自对话推理模型,并展示了其潜力。我们的代码和数据集见https://mathcritique.github.io/。

英文摘要:

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/}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑