用对抗强化学习让大模型推理更靠谱,错误率明显下降。
Generative Adversarial Reasoner: Enhancing LLM Reasoning with Adversarial Reinforcement Learning
- 让模型和判别器互练,通过对抗学习优化每一步推理。
- 在AIME24上,7B模型准确率从54.0提升至61.3,8B模型提升10.0点。
- 可灵活适配教学蒸馏、偏好对齐等任务,适合改进推理能力。
具备显式推理能力的大语言模型在数学推理中表现优异,但仍存在计算错误、逻辑脆弱、表面合理实则无效的步骤。本文提出生成式对抗推理器(Generative Adversarial Reasoner),一种基于策略的联合训练框架,通过对抗强化学习共同演化一个语言模型推理器与一个基于LLM的判别器。采用高效的审查机制将推理链划分为长度相近的逻辑完整片段,判别器以简洁结构化理由评估每个片段的合理性。学习过程耦合互补信号:推理器因逻辑一致且答案正确获得奖励,判别器因准确识别错误或区分推理轨迹而获益。该机制生成密集、校准良好的在线策略级步骤奖励,补充稀疏的精确匹配信号,改善信用分配,提升样本效率,增强整体推理质量。在多个数学基准测试中,该方法持续优于强基线模型。具体地,在AIME24上,DeepSeek-R1-Distill-Qwen-7B准确率由54.0提升至61.3(+7.3),DeepSeek-R1-Distill-Llama-8B由43.7提升至53.7(+10.0)。模块化判别器还支持灵活的奖励塑造,适用于教师蒸馏、偏好对齐及基于数学证明的推理等目标。
原文摘要 · Abstract (English)
Large language models (LLMs) with explicit reasoning capabilities excel at mathematical reasoning yet still commit process errors, such as incorrect calculations, brittle logic, and superficially plausible but invalid steps. In this paper, we introduce Generative Adversarial Reasoner, an on-policy joint training framework designed to enhance reasoning by co-evolving an LLM reasoner and an LLM-based discriminator through adversarial reinforcement learning. A compute-efficient review schedule partitions each reasoning chain into logically complete slices of comparable length, and the discriminator evaluates each slice's soundness with concise, structured justifications. Learning couples complementary signals: the LLM reasoner is rewarded for logically consistent steps that yield correct answers, while the discriminator earns rewards for correctly detecting errors or distinguishing traces in the reasoning process. This produces dense, well-calibrated, on-policy step-level rewards that supplement sparse exact-match signals, improving credit assignment, increasing sample efficiency, and enhancing overall reasoning quality of LLMs. Across various mathematical benchmarks, the method delivers consistent gains over strong baselines with standard RL post-training. Specifically, on AIME24, we improve DeepSeek-R1-Distill-Qwen-7B from 54.0 to 61.3 (+7.3) and DeepSeek-R1-Distill-Llama-8B from 43.7 to 53.7 (+10.0). The modular discriminator also enables flexible reward shaping for objectives such as teacher distillation, preference alignment, and mathematical proof-based reasoning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。