TSC: Topology-Conditioned Stackelberg Coordination for Multi-Agent Reinforcement Learning in Interactive Driving
Authors: Xiaotong Zhang, Gang Xiong, Yuanjing Wang, Siyu Teng, Alois Knoll, Long Chen
Organizations: State Key Laboratory of Multimodal Artificial Intelligence Systems, Institute of Automation, Chinese Academy of Sciences, Beijing 100190, China · School of Artificial Intelligence, University of Chinese Academy of Sciences, Beijing 100049, China · Department of Natural Sciences, University of Durham, Durham DH1 3LE, United Kingdom · College of Civil and Transportation Engineering, Shenzhen University, Guangdong, 518060, China · Chair of Robotics, Artificial Intelligence and Realtime Systems, Technical University of Munich, 80333 Munich, Germany · State Key Laboratory of Management and Control for Complex Systems, Institute of Automation, Chinese Academy of Sciences, Beijing, 100190, China · Waytous Inc., Qingdao 266109, China
Safe and efficient autonomous driving in dense traffic is fundamentally a decentralized multi-agent coordination problem, where interactions at conflict points such as merging and weaving must be resolved reliably under partial observability. With only local and incomplete cues, interaction patterns can change rapidly, often causing unstable behaviors such as oscillatory yielding or unsafe commitments. Existing multi-agent reinforcement learning (MARL) approaches either adopt synchronous decision-making, which exacerbate non-stationarity, or depend on centralized sequencing mechanisms that scale poorly as traffic density increases. To address these limitations, we propose Topology-conditioned Stackelberg Coordination (TSC), a learning framework for decentralized interactive driving under communication-free execution, which extracts a time-varying directed priority graph from braid-inspired weaving relations between trajectories, thereby defining local leader-follower dependencies without constructing a global order of play. Conditioned on this graph, TSC endogenously factorizes dense interactions into graph-local Stackelberg subgames and, under centralized training and decentralized execution (CTDE), learns a sequential coordination policy that anticipates leaders via action prediction and trains followers through action-conditioned value learning to approximate local best responses, improving training stability and safety in dense traffic. Experiments across four dense traffic scenarios show that TSC achieves superior performance over representative MARL baselines across key metrics, most notably reducing collisions while maintaining competitive traffic efficiency and control smoothness.
Figures & tables
Fig. 1 : Topology-conditioned Stackelberg coordination from trajectory weaving in multi-vehicle traffic.
Fig. 2 : The TSC-Net architecture. Ego and neighbor encoders construct an interaction representation Hit , from which the TopoDecoder predicts pairwise priorities p^i←jt and node scores s^it to form a local priority graph. Guided by this graph, a TopK filter and TopoAttention aggregate the most influential neighbors into a compact decision state uit , which the PolicyHead maps to the continuous action ait . The same priority structure defines the local leader set Lit . The PredictHead estimates leader actions {a^j∣it}j∈Lit , and the ValueHead conditions on these predictions to approximate the local Stackelberg value, yielding TD advantages A^it for actor optimization under CTDE.
Fig. 3 : Evaluation scenarios for dense multi-vehicle interaction.
MFPO
SigmaRL
XP-MARL
TSC
Clover
CRAA↓
1.71
1.67
0.19
0.04
CRAM↓
0.00
0.00
0.03
0.03
CR↓
1.71
1.67
0.22
0.07
AS↑
86.6
87.1
84.8
88.2
SMLO↓
2.63
2.58
2.27
1.95
SMLA↓
11.2
10.4
8.95
8.89
TABLE I : Testing results on four scenarios from different MARL methods. Collision rate (CR [%] ) measures safety and is decomposed into agent-agent collisions CRAA and agent-map collisions CRAM . Average speed (AS [%] ) measures traffic efficiency. Smoothness (SM [%] ) measures control stability. Best is bold and second best is underlined within each scenario.
Fig. 4 : Comparison of safety, efficiency, and smoothness across four scenarios.
Fig. 5 : Collision rate of different methods under varying numbers of vehicles at test time.
Fig. 6 : Visualization of vehicle trajectories in one test episode. Black dots indicate collision events.
Fig. 7 : Qualitative trajectory visualization of TSC across four interaction scenarios, highlighting emergent behaviors including yielding, queuing, cutting in, bypassing, and reversing. For each vehicle, we plot a 25-step trajectory segment with distinct colors, where the luminosity fades from light to deep to indicate earlier to later positions.
Method
Clover
Weave
Merge
Bypass
TSC w/ random priority
0.30
1.62
1.11
1.79
TSC w/o Stackelberg
0.27
1.08
0.78
1.91
TSC w/o Top- K Filter
1.00
2.92
0.97
1.87
TSC
0.07
0.99
0.03
0.10
TABLE II : Ablation study on total collision rate (CR [%] ) across different scenarios.
Fig. 8 : Comparison of TSC with different TopoNbr-K settings.
Department of Mechanical Engineering, College of Design and Engineering, National University of Singapore, 21 Lower Kent Ridge Rd, 119077, Singapore · Department of Electrical and Electronic Engineering, The Hong Kong Polytechnic University, Hong Kong, 100872, China · School of Vehicle and Mobility, Tsinghua University, Beijing, 100084, China
Department of Mechanical Engineering, National University of Singapore · Chandra Department of Electrical and Computer Engineering, The University of Texas at Austin · Department of Electrical Engineering and Computer Science, Massachusetts Institute of Technology