多智能体协作提升大模型推理时计算扩展能力
TMAS: Scaling Test-Time Compute via Multi-Agent Synergy

- 用专业分工的多智能体协同推理,结构化信息流动
- 在多个基准测试中实现更强的迭代扩展效果
- 适合需要高效推理优化的研究者与工程实践者
测试时扩展已成为通过推理阶段增加计算量来提升大语言模型推理能力的有效范式。近期的结构化方法进一步推进了这一范式,通过组织多条推理轨迹、多轮修正和基于验证的反馈来实现。然而,现有方法或对并行推理轨迹协调不足,或依赖有噪声的历史信息而未明确决定保留与复用的内容,限制了探索与利用之间的平衡。本文提出TMAS,一种基于多智能体协同的测试时计算扩展框架。TMAS将推理组织为专业化智能体间的协作过程,支持跨智能体、轨迹和修正迭代的结构化信息流。为促进跨轨迹协作,引入分层记忆机制:经验库复用低层次可靠中间结论和局部反馈,指南库记录此前探索的高层次策略以引导后续推演避开重复推理模式。此外,设计了针对TMAS的混合奖励强化学习方案,兼顾基础推理能力、经验利用率,并鼓励探索未尝试过的解题策略。大量实验表明,TMAS在挑战性推理基准上优于现有测试时扩展基线,混合奖励训练进一步提升了迭代中的扩展效率与稳定性。代码与数据见https://github.com/IQuestLab/tmas。
原文摘要 · Abstract (English)
Test-time scaling has become an effective paradigm for improving the reasoning ability of large language models by allocating additional computation during inference. Recent structured approaches have further advanced this paradigm by organizing inference across multiple trajectories, refinement rounds, and verification-based feedback. However, existing structured test-time scaling methods either weakly coordinate parallel reasoning trajectories or rely on noisy historical information without explicitly deciding what should be retained and reused, limiting their ability to balance exploration and exploitation. In this work, we propose TMAS, a framework for scaling test-time compute via multi-agent synergy. TMAS organizes inference as a collaborative process among specialized agents, enabling structured information flow across agents, trajectories, and refinement iterations. To support effective cross-trajectory collaboration, TMAS introduces hierarchical memories: the experience bank reuses low-level reliable intermediate conclusions and local feedback, while the guideline bank records previously explored high-level strategies to steer subsequent rollouts away from redundant reasoning patterns. Furthermore, we design a hybrid reward reinforcement learning scheme tailored to TMAS, which jointly preserves basic reasoning capability, enhances experience utilization, and encourages exploration beyond previously attempted solution strategies. Extensive experiments on challenging reasoning benchmarks show that TMAS achieves stronger iterative scaling than existing test-time scaling baselines, with hybrid reward training further improving scaling effectiveness and stability across iterations. Code and data are available at https://github.com/IQuestLab/tmas.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。