让多个AI代理协作反思,提升大模型推理准确率
Reinforce LLM Reasoning through Multi-Agent Reflection
- 用强化学习让多个AI代理轮流改进答案,动态优化
- 在MATH 500上,五轮迭代后准确率从58.2%升至63.2%
- 适合需要高精度推理的复杂任务,如数学求解
利用更多测试时计算已被证明是提升大语言模型(LLMs)推理能力的有效方法。在多种方法中,验证与改进范式因其能实现动态解题探索和反馈整合而尤为突出。然而,现有方法常受限于反馈空间狭窄且各参与方缺乏协同训练,导致性能不佳。为此,我们将多轮优化过程建模为马尔可夫决策过程,并提出DPSDP(基于动态规划的直接策略搜索),一种通过自生成数据进行直接偏好学习的强化学习算法,训练一个演员-评论家型LLM系统,以迭代方式改进答案。理论上,DPSDP可达到训练分布内任意策略的性能上限。实证上,我们用多种基础模型实例化DPSDP,展示了其在分布内与分布外基准上的提升效果。例如,在MATH 500基准上,五轮精炼后,基于Ministral模型的首次回答准确率从58.2%提升至63.2%。消融实验进一步验证了多代理协作与分布外泛化的优势。
原文摘要 · Abstract (English)
Leveraging more test-time computation has proven to be an effective way to boost the reasoning capabilities of large language models (LLMs). Among various methods, the verify-and-improve paradigm stands out for enabling dynamic solution exploration and feedback incorporation. However, existing approaches often suffer from restricted feedback spaces and lack of coordinated training of different parties, leading to suboptimal performance. To address this, we model this multi-turn refinement process as a Markov Decision Process and introduce DPSDP (Direct Policy Search by Dynamic Programming), a reinforcement learning algorithm that trains an actor-critic LLM system to iteratively refine answers via direct preference learning on self-generated data. Theoretically, DPSDP can match the performance of any policy within the training distribution. Empirically, we instantiate DPSDP with various base models and show improvements on both in- and out-of-distribution benchmarks. For example, on benchmark MATH 500, majority voting over five refinement steps increases first-turn accuracy from 58.2% to 63.2% with Ministral-based models. An ablation study further confirms the benefits of multi-agent collaboration and out-of-distribution generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。