arXiv:2603.02630cs.LGcs.AI2026-03被引 4

用强化学习优化多智能体系统的提示词,高效且精准。

MASPOB: Bandit-Based Prompt Optimization for Multi-Agent Systems with Graph Neural Networks

  • 基于贝叶斯上置信界,平衡探索与利用,节省评估成本。
  • 结合图神经网络捕捉提示词间结构关联,提升优化精度。
  • 适合高成本、复杂依赖场景的智能体系统提示词调优。

大语言模型在多智能体系统(MAS)中扮演认知核心角色,但其性能高度依赖输入提示词,而实际部署中难以修改工作流,因此提示词优化成为更自然的改进方式。然而,真实场景下的提示词优化面临三大挑战:(1)评估成本高昂,需高样本效率;(2)提示词间存在由系统拓扑引发的耦合关系;(3)搜索空间呈组合爆炸。为此,本文提出MASPOB(基于强化学习的多智能体系统提示词优化),采用贝叶斯上置信界(UCB)机制,在严格预算下实现探索与利用的平衡。通过引入图神经网络(GNN)建模拓扑先验,学习提示语义的拓扑感知表示,缓解耦合问题。同时,采用坐标上升法将优化分解为单变量子问题,使搜索复杂度从指数级降至线性。在多个基准测试中,MASPOB表现优异,持续优于现有基线方法。

原文摘要 · Abstract (English)

Large Language Models (LLMs) have achieved great success in many real-world applications, especially the one serving as the cognitive backbone of Multi-Agent Systems (MAS) to orchestrate complex workflows in practice. Since many deployment scenarios preclude MAS workflow modifications and its performance is highly sensitive to the input prompts, prompt optimization emerges as a more natural approach to improve its performance. However, real-world prompt optimization for MAS is impeded by three key challenges: (1) the need of sample efficiency due to prohibitive evaluation costs, (2) topology-induced coupling among prompts, and (3) the combinatorial explosion of the search space. To address these challenges, we introduce MASPOB (Multi-Agent System Prompt Optimization via Bandits), a novel sample-efficient framework based on bandits. By leveraging Upper Confidence Bound (UCB) to quantify uncertainty, the bandit framework balances exploration and exploitation, maximizing gains within a strictly limited budget. To handle topology-induced coupling, MASPOB integrates Graph Neural Networks (GNNs) to capture structural priors, learning topology-aware representations of prompt semantics. Furthermore, it employs coordinate ascent to decompose the optimization into univariate sub-problems, reducing search complexity from exponential to linear. Extensive experiments across diverse benchmarks demonstrate that MASPOB achieves state-of-the-art performance, consistently outperforming existing baselines.

提示优化多智能体图神经网络强化学习

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