让大模型学会预判他人策略,提升合作决策能力
ToMPO: Training LLM Strategic Decision Making from a Multi-Agent Perspective
- 基于多智能体视角构建决策框架,动态感知对手策略
- 相比GRPO算法,合作成功率提升35%,小模型表现超越百倍大模型18%
- 适合研究智能体协作、博弈策略优化的学者与工程师
大型语言模型在复杂场景中需具备深度思考、逻辑推理和明智决策能力。现有研究多聚焦于社交任务中的多轮对话或模拟环境,忽视了多种决策类型及其时间依赖关系。当前强化学习方法难以在训练中考虑其他个体的策略。为此,本文首先定义包含两类决策及其时序依赖的战略决策问题;提出理论之眼策略优化(ToMPO)算法,用于优化对其他个体策略及局势趋势的感知。相较于群体相对策略优化(GRPO),ToMPO通过三方面改进:1)基于对他人策略的推理生成轨迹;2)在图级与样本级同时估计优势;3)平衡全局与局部奖励。实验表明,ToMPO在模型输出合规性与合作结果上比GRPO提升35%;且相较参数量大100倍的模型,仍实现18%的性能提升,验证了其在增强模型战略决策能力上的有效性。
原文摘要 · Abstract (English)
Large Language Models (LLMs) have been used to make decisions in complex scenarios, where they need models to think deeply, reason logically, and decide wisely. Many existing studies focus solely on multi-round conversations in social tasks or simulated environments, neglecting the various types of decisions and their interdependence. Current reinforcement learning methods struggle to consider the strategies of others during training. To address these issues, we first define a strategic decision-making problem that includes two types of decisions and their temporal dependencies. Furthermore, we propose **T**heory **o**f **M**ind **P**olicy **O**ptimization **(ToMPO)** algorithm to optimize the perception of other individual strategies and the game situation trends. Compared to the Group Relative Policy Optimization (GRPO) algorithm, ToMPO enhances the LLM's strategic decision-making mainly by: 1) generating rollouts based on reasoning the strategies of other individuals, 2) estimating advantages at both the graph-level and sample-level, and 3) balancing global and partial rewards. The ToMPO algorithm outperforms the GRPO method by 35% in terms of model output compliance and cooperative outcomes. Additionally, when compared to models with parameter sizes 100 times larger, it shows an 18% improvement. This demonstrates the effectiveness of the ToMPO algorithm in enhancing the model's strategic decision-making capabilities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。