arXiv:2503.09501cs.AIcs.CL2025-03NeurIPS被引 74

用多智能体强化学习让大模型学会自我反思,提升解题能力

ReMA: Learning to Meta-think for LLMs with Multi-Agent Reinforcement Learning

  • 分层设计两个智能体:高层负责策略规划,低层执行推理
  • 在数学和评判类任务上超越单智能体基线,性能显著提升
  • 适合研究大模型推理机制或想改进AI决策能力的开发者

近期关于大语言模型(LLMs)推理的研究致力于通过引入元思考——使模型能够监控、评估并控制自身推理过程——以实现更自适应、高效的解题。然而,现有单智能体方法缺乏专门设计,导致元思考效果不佳。为此,我们提出强化元思考智能体(ReMA),采用多智能体强化学习(MARL)激发元思考行为,促使大模型‘思考如何思考’。ReMA将推理过程解耦为两个层级:高层元思考智能体负责生成战略监督与计划,低层推理智能体负责具体执行。通过目标对齐的迭代强化学习,两智能体探索并学习协作,从而提升泛化性与鲁棒性。单轮实验结果表明,ReMA在复杂推理任务中优于单智能体强化学习基线,涵盖竞赛级数学基准与LLM-as-a-Judge基准。此外,我们还将ReMA扩展至多轮交互场景,利用轮次级比率与参数共享提升效率。全面消融实验揭示了各智能体的动态演变,为理解元思考如何增强大模型推理能力提供了洞见。代码已公开于 https://github.com/ziyuwan/ReMA-public。

原文摘要 · Abstract (English)

Recent research on Reasoning of Large Language Models (LLMs) has sought to further enhance their performance by integrating meta-thinking -- enabling models to monitor, evaluate, and control their reasoning processes for more adaptive and effective problem-solving. However, current single-agent work lacks a specialized design for acquiring meta-thinking, resulting in low efficacy. To address this challenge, we introduce Reinforced Meta-thinking Agents (ReMA), a novel framework that leverages Multi-Agent Reinforcement Learning (MARL) to elicit meta-thinking behaviors, encouraging LLMs to think about thinking. ReMA decouples the reasoning process into two hierarchical agents: a high-level meta-thinking agent responsible for generating strategic oversight and plans, and a low-level reasoning agent for detailed executions. Through iterative reinforcement learning with aligned objectives, these agents explore and learn collaboration, leading to improved generalization and robustness. Empirical results from single-turn experiments demonstrate that ReMA outperforms single-agent RL baselines on complex reasoning tasks, including competitive-level mathematical benchmarks and LLM-as-a-Judge benchmarks. Additionally, we further extend ReMA to multi-turn interaction settings, leveraging turn-level ratio and parameter sharing to improve efficiency. Comprehensive ablation studies further illustrate the evolving dynamics of each distinct agent, providing valuable insights into how the meta-thinking reasoning process enhances the reasoning capabilities of LLMs. Our code can be found in https://github.com/ziyuwan/ReMA-public

大模型推理元思考多智能体强化学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。