让小模型通过辩论式强化学习提升推理能力,仅用极少参数就显著提高准确率。
MADA-RL: Multi-Agent Debate-Aware Reinforcement Learning for Parameter-Efficient Reasoning in Compact Models
- 将小模型分为生成与批判角色,用动态反事实优势信号训练,只微调少量参数。
- 1.5B模型在数学推理任务上准确率从39.9%提至41.9%,参数量仅为全微调的1/16。
- 适合资源有限但需高效推理的小模型部署,尤其关注误差纠正而非答案模仿。
大语言模型虽具强推理能力,但训练成本高昂,对参数量≤40亿、预算受限的小模型尤为严峻。本文提出MADA-RL,一种后训练框架,将紧凑模型分别赋予生成器与批评者角色,并通过具辩论意识的学习信号进行训练,仅使用LoRA适配器微调极小部分参数。核心贡献是反事实批评者优势:一种动态、角色依赖的基准,将批评者的奖励定义为自身奖励减去生成器集合的实例级准确率。这促使批评者超越生成器共识,而非简单复现正确答案,实现更精准的信用分配。部署时,专用代理以轻量多轮协议组合。在五个数学推理基准上,该方法使DeepSeek-R1-Distill-Qwen-1.5B模型准确率从39.9%提升至41.9%(+2.0点,p<0.001),训练参数仅为全微调基线的1/16,处于准确率-可训练参数帕累托前沿。其性能接近但未超越更强基线(DeepScaleR、STILL-3),后者基于更大数据集训练;文中分析了这一差距及相应的推理开销。控制实验表明,反事实优势带来所有模型中最高的批评者改进率,证明训练后的批评者学会纠正生成错误,而非模仿答案。
原文摘要 · Abstract (English)
Large language models achieve strong reasoning performance, but often at prohibitive training cost - a challenge that is especially acute for compact models ($\leq 4 \, \mathrm{B}$ parameters) trained under limited budgets. We introduce MADA-RL, a post-training framework that specializes compact models into generator and critic roles and trains them with a debate-aware learning signal, fine-tuning only a small subset of parameters via LoRA adapters. Our central contribution is a counterfactual critic advantage: a dynamic, role-conditioned baseline that redefines the critic's advantage as its reward minus the generator ensemble's per-instance accuracy. This explicitly optimizes critics to improve over generator consensus rather than to merely reproduce a correct answer, yielding more targeted credit assignment than static mean-reward normalization. At deployment, the specialized agents are composed in a lightweight multi-round protocol. Across five mathematical reasoning benchmarks, MADA-RL raises the accuracy of the DeepSeek-R1-Distill-Qwen-1.5B model from $39.9 \, \%$ to $41.9 \, \%$ ($+2.0$ points, $p < 0.001$) using $16$ times fewer trainable parameters than fully fine-tuned baselines, placing it on the accuracy-trainable-parameter Pareto front. It approaches, but does not surpass, the strongest baselines (DeepScaleR, STILL-3), which are trained on substantially larger datasets; we analyse this gap and the associated inference-time cost directly. A controlled study isolates the source of MADA-RL's gains: the counterfactual advantage produces the highest critic improvement rate of any model evaluated, indicating that trained critics learn to correct generator errors rather than to imitate them.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。