GBC: Gradient-Based Connections for Optimizing Multi-Agent Systems
Abstract
Gradient-Based Connections enables fine-grained attribution and optimization in multi-agent systems by modeling agent interactions as a computational graph and using gradient-based weights to identify error sources at the token level.
Multi-agent systems (MAS) built on large language models (LLMs) provide a promising framework for solving complex tasks through role specialization and structured interaction. However, their performance is often limited by miscoordination and, more fundamentally, the lack of fine-grained credit assignment across agents. Existing approaches typically rely on coarse-grained feedback, making it difficult to identify which agents or interaction steps are responsible for errors. We propose Gradient-Based Connections (GBC), an approach for fine-grained attribution and optimization of multi-agent systems. GBC models a MAS as a computational graph and introduces gradient-based connection weights to quantify the influence of each agent's output on downstream agents at the token level. By constructing an attribution graph and propagating task-specific loss signals backward, our method enables precise identification of error sources and targeted prompt optimization. We further develop AgentChord, an efficient implementation that leverages prefix-based gradient computation. Experiments on MultiWOZ and τ-bench show that GBC improves multi-agent performance and outperforms strong single-agent and multi-agent baselines, and higher attribution quality is associated with greater optimization effectiveness. Code is available at: https://github.com/yxc-cyber/AgentChord.
Community
Excited to share our work: GBC: Gradient-Based Connections for Optimizing Multi-Agent Systems. Multi-agent LLM systems are promising for complex tasks, but they often suffer from miscoordination and weak credit assignment. When the final output is wrong, how do we know which agent caused the failure?
We propose Gradient-Based Connections (GBC). GBC models a multi-agent system as a computational graph and uses gradient-based signals to quantify how each agent output influences downstream agents at the token level. This allows us to construct an attribution graph and propagate task-specific loss signals backward to identify error sources. We evaluate GBC on multi-agent task-oriented dialogue and tool-use settings, showing the effectiveness of this mechanism and that better attribution can lead to more effective system-level optimization.
Happy to hear any thoughts, feedback, or suggestions!
Get this paper in your agent:
hf papers read 2606.28187 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper