多智能体协作生成数据,用树搜索动态优化流程,提升推理效率与质量。
Multi-Agent Sampling: Scaling Inference Compute for Data Synthesis with Tree Search-Based Agentic Collaboration
- 用树搜索动态规划多模型协作路径,实时反馈调整生成结构。
- 在WMT上达当前最优,AlpacaEval胜率72.2%,优于单智能体方案。
- 适合需要高质量合成数据的对齐训练与复杂任务建模场景。
多智能体系统中的推理算力扩展规律远未被充分研究,尤其相比单智能体场景。本文聚焦多智能体采样下的数据合成问题,通过从多个不同语言模型中采样生成合成响应。有效模型协同是多智能体协作成功的关键。不同于依赖固定流程的以往方法,我们将其视为多步决策过程,针对每个输入问题动态优化生成结构。提出基于树搜索的智能体协调框架(TOA),在序列采样过程中迭代演化工作流。通过引入蒙特卡洛树搜索(MCTS)并结合奖励模型提供实时反馈,加速探索。在对齐、机器翻译和数学推理任务上的实验表明,随着推理算力增加,多智能体采样显著优于单智能体方案。TOA为最高效的计算方式,在WMT上达到当前最优性能,且在AlpacaEval上实现72.2%的胜率。此外,使用合成对齐数据微调后,在Arena-Hard和AlpacaEval等挑战性基准上超越强偏好学习方法。
原文摘要 · Abstract (English)
Scaling laws for inference compute in multi-agent systems remain under-explored compared to single-agent scenarios. This work aims to bridge this gap by investigating the problem of data synthesis through multi-agent sampling, where synthetic responses are generated by sampling from multiple distinct language models. Effective model coordination is crucial for successful multi-agent collaboration. Unlike previous approaches that rely on fixed workflows, we treat model coordination as a multi-step decision-making process, optimizing generation structures dynamically for each input question. We introduce Tree Search-based Orchestrated Agents~(TOA), where the workflow evolves iteratively during the sequential sampling process. To achieve this, we leverage Monte Carlo Tree Search (MCTS), integrating a reward model to provide real-time feedback and accelerate exploration. Our experiments on alignment, machine translation, and mathematical reasoning demonstrate that multi-agent sampling significantly outperforms single-agent sampling as inference compute scales. TOA is the most compute-efficient approach, achieving SOTA performance on WMT and a 72.2\% LC win rate on AlpacaEval. Moreover, fine-tuning with our synthesized alignment data surpasses strong preference learning methods on challenging benchmarks such as Arena-Hard and AlpacaEval.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。