用对抗自博弈训练模型,自动发现大模型推理中的错误。
SPC: Evolving Self-Play Critic via Adversarial Games for LLM Reasoning
- 让两个模型互为对手,一个造假步骤,一个找漏洞,互相提升判断力。
- 在三个基准上错误检测准确率从70.8%提升至77.7%,优于现有方法。
- 无需人工标注,适合用于改进各类大模型的数学推理能力。
评估大语言模型(LLM)链式思考等逐步推理过程的可靠性仍具挑战,因高质量步骤级监督难以获取。本文提出自博弈评论器(SPC),通过对抗自博弈方式,使评论模型在无须人工步骤标注的情况下持续进化其推理步骤评估能力。SPC对基础模型进行微调,生成两个角色:‘狡猾生成器’刻意制造难以察觉的错误步骤,以及‘评论者’分析推理步骤的正确性。两者在对抗游戏中交替攻防,生成器试图误导评论者,评论者则力求识别错误。基于游戏结果,利用强化学习迭代优化:胜者获正奖励,败者获负奖励,推动模型持续自我演化。在ProcessBench、PRM800K和DeltaBench三个推理基准上的实验表明,SPC的错误检测能力持续增强(如ProcessBench上准确率由70.8%提升至77.7%),超越强基线,包括蒸馏版R1模型。此外,SPC能有效指导多种大模型在测试时的搜索策略,在MATH500和AIME2024上显著提升数学推理表现,优于当前最优过程奖励模型。
原文摘要 · Abstract (English)
Evaluating the step-by-step reliability of large language model (LLM) reasoning, such as Chain-of-Thought, remains challenging due to the difficulty and cost of obtaining high-quality step-level supervision. In this paper, we introduce Self-Play Critic (SPC), a novel approach where a critic model evolves its ability to assess reasoning steps through adversarial self-play games, eliminating the need for manual step-level annotation. SPC involves fine-tuning two copies of a base model to play two roles, namely a "sneaky generator" that deliberately produces erroneous steps designed to be difficult to detect, and a "critic" that analyzes the correctness of reasoning steps. These two models engage in an adversarial game in which the generator aims to fool the critic, while the critic model seeks to identify the generator's errors. Using reinforcement learning based on the game outcomes, the models iteratively improve; the winner of each confrontation receives a positive reward and the loser receives a negative reward, driving continuous self-evolution. Experiments on three reasoning process benchmarks (ProcessBench, PRM800K, DeltaBench) demonstrate that our SPC progressively enhances its error detection capabilities (e.g., accuracy increases from 70.8% to 77.7% on ProcessBench) and surpasses strong baselines, including distilled R1 model. Furthermore, SPC can guide the test-time search of diverse LLMs and significantly improve their mathematical reasoning performance on MATH500 and AIME2024, surpassing those guided by state-of-the-art process reward models.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。