Organizations: 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
Adaptive traffic signal control (ATSC) is crucial in alleviating congestion, maximizing throughput and promoting sustainable mobility in ever-expanding cities. Multi-Agent Reinforcement Learning (MARL) has recently shown significant potential in addressing complex traffic dynamics, but the intricacies of partial observability and coordination in decentralized environments still remain key challenges in formulating scalable and efficient control strategies. To address these challenges, we present CoordLight, a MARL-based framework designed to improve intra-neighborhood traffic by enhancing decision-making at individual junctions (agents), as well as coordination with neighboring agents, thereby scaling up to network-level traffic optimization. Specifically, we introduce the Queue Dynamic State Encoding (QDSE), a novel state representation based on vehicle queuing models, which strengthens the agents' capability to analyze, predict, and respond to local traffic dynamics. We further propose an advanced MARL algorithm, named Neighbor-aware Policy Optimization (NAPO). It integrates an attention mechanism that discerns the state and action dependencies among adjacent agents, aiming to facilitate more coordinated decision-making, and to improve policy learning updates through robust advantage calculation. This enables agents to identify and prioritize crucial interactions with influential neighbors, thus enhancing the targeted coordination and collaboration among agents. Through comprehensive evaluations against state-of-the-art traffic signal control methods over three real-world traffic datasets composed of up to 196 intersections, we empirically show that CoordLight consistently exhibits superior performance across diverse traffic networks with varying traffic flows. The code is available at https://github.com/marmotlab/CoordLight
Figures & tables
Fig. 1: Overall learning framework of CoordLight, which introduces a novel state definition, Queueing Dynamic State Encoding (QDSE) for fine-grained traffic representation, and a Neighbor-aware Policy Optimization (NAPO) algorithm to learn efficient coordination strategies among neighboring agents.
Fig. 2: An illustration of a single intersection with eight traffic signal phases, where the phase NS-Left is currently activated, allowing the vehicles along the specified traffic movements (green dotted lines) to traverse the intersection.
Fig. 3: An illustration of the proposed state representation QDSE for an incoming lane l of a single intersection, where Ql(t)=3 , Ninl(t)=1 , Noutl(t)=0 , Nfrl(t)=3 , Nrl(t)=6 , and Dfrl(t)=15 in this case.
Fig. 4: Detailed structure of our neighbor-aware actor-critic network in CoordLight: Fig. (a) shows the overall structure of our attention-based, spatio-temporal actor network (STN). This network consists of a spatial aggregation unit and a temporal aggregation unit to process information from the agent’s neighborhood for policy output. Fig.(b) illustrates the structure of the privileged local critic network, which includes a state encoder and a state-action decoder that captures crucial interactions and incorporates neighbors’ state-action dependencies into the value estimation.
Fig. 5: Real-world traffic road networks of the CityFlow simulation environments used in our experimental results: Jinan, China map (left, 3×4 intersections), Hangzhou, China map (center, 4×4 intersections), and New York, USA map (right, 7×28 intersections).
Network
Type
Volume (veh)
Arrival Rate(veh/min)
Mean
Std.
Max
Min
Jinan (3x4)
DJN(1)
6295
104.92
19.79
136.00
50.00
DJN(2)
4365
72.75
15.15
101.00
43.00
DJN(3)
5494
91.57
9.51
111.00
69.00
Hangzhou (4x4)
DHZ(1)
2983
49.72
8.24
67.00
40.00
DHZ(2)
6984
116.40
63.72
230.00
39.00
TABLE I: Traffic dataset for three cities - Jinan, Hangzhou, and New York, each with different traffic demands.
Method
DJN(1)
DJN(2)
DJN(3)
DHZ(1)
DHZ(2)
DNY(1)
DNY(2)
Fixed-Time
346.36
289.74
316.70
432.32
359.44
1507.12
1660.29
MaxPressure
273.96
245.38
245.81
288.54
348.98
1179.55
1535.77
Advanced-MP
253.61
238.62
235.21
279.47
318.67
1060.41 ∗
-
CoLight
276.33 ± 3.98
237.14 ± 0.73
278.16 ± 1.13
271.07 ± 2.11
297.26 ± 9.47
1221.77 ± 19.83
1476.18 ± 29.22
MPLight
300.93 ± 3.57
259.10 ± 8.90
261.45 ± 1.74
343.47 ± 2.89
282.14 ± 7.20
1168.49 ± 23.99
1597.24 ± 18.57
Advanced-CoLight
253.95 ± 0.77
234.03 ± 0.61
244.51 ± 0.66
269.62 ± 0.51
308.62 ± 4.58
1025.47 ± 34.33
-
TABLE II: Average travel times (sec) comparing CoordLight with baselines in each CityFlow environment/dataset ( ∗ indicate results that had to be reported from the original baseline paper).
Traffic Dataset
DJN(1)
DJN(2)
DJN(3)
DHZ(1)
DHZ(2)
DNY(1)
DNY(2)
P-value
2.23e-16
9.99e-20
1.24e-21
8.63e-20
1.08e-18
5.45e-09
9.79e-09
Threshold (Bonferroni Correction)
1.57e-04
TABLE III: Unpaired T-test Results for CoordLight and SocialLight (baseline method).
Fig. 6: Comparative evaluation of the mean and standard deviation of the average travel time at intersections (lower is better) against three MARL baselines over high-demand traffic datasets DJN(1) and DHZ(2) .
Fig. 7: Training results of the ablation variants of CoordLight in terms of average queue length ( ↓ ), average speed ( ↑ ), average travel time ( ↓ ), and standard deviation of queue length ( ↓ ) with respect to the DHZ(2) dataset for 10000 episodes. Here, ↓ indicates lower is better, while ↑ indicates higher is better.
Fig. 8: Evaluation of the proposed state representation QDSE on Jinan and Hangzhou datasets, comparing average travel time with and without sensor noise (Gaussian noise) across various traffic demands. All variants are trained with noise-free QDSE and tested with noisy-QDSE.
Traffic signal control studies how to set intersection phases and green times to reduce congestion while balancing safety, emissions, and priority needs.
Adaptive Traffic Signal Control (ATSC) aims to optimize traffic flow and minimize delays by adjusting traffic lights in real time. Recent advances in Multi-agent Reinforcement Learning (MARL) have shown promise for ATSC, yet existing approaches still suffer from limited representational capacity, often leading to suboptimal performance and poor generalization in complex and dynamic traffic environments. On the other hand, Large Language Models (LLMs) excel at semantic representation, reasoning, and analysis, yet their propensity for hallucination and slow inference speeds often hinder their direct application to decision-making tasks. To address these challenges, we propose a novel learning paradigm named LATS that integrates LLMs and MARL, leveraging the former's strong prior knowledge and inductive abilities to enhance the latter's decision-making process. Specifically, we introduce a plug-and-play teacher-student learning module, where a trained embedding LLM serves as a teacher to generate rich semantic features that capture each intersection's topology structures and traffic dynamics. A much simpler (student) neural network then learns to emulate these features through knowledge distillation in the latent space, enabling the final model to operate independently from the LLM for downstream use in the RL decision-making process. This integration significantly enhances the overall model's representational capacity across diverse traffic scenarios, thus leading to more efficient and generalizable control strategies. Extensive experiments across diverse traffic datasets empirically demonstrate that our method enhances the representation learning capability of RL models, thereby leading to improved overall performance and generalization over both traditional RL and LLM-only approaches. [...]
Yifeng Zhang, Peizhuo Li, Tingguang Zhou +2
Department of Mechanical Engineering, National University of Singapore
Reinforcement Learning (RL) in Traffic Signal Control (TSC) faces significant hurdles in real-world deployment due to limited generalization to dynamic traffic flow variations. Existing approaches often overfit static patterns and use action spaces incompatible with driver expectations. This paper proposes a robust Multi-Agent Reinforcement Learning (MARL) framework validated in the Vissim traffic simulator. The framework integrates three mechanisms: (1) Turning Ratio Randomization, a training strategy that exposes agents to dynamic turning probabilities to enhance robustness against unseen scenarios; (2) a stability-oriented Exponential Phase Duration Adjustment action space, which balances responsiveness and precision through cyclical, exponential phase adjustments; and (3) a Neighbor-Based Observation scheme utilizing the MAPPO algorithm with Centralized Training with Decentralized Execution (CTDE). By leveraging centralized updates, this approach approximates the efficacy of global observations while maintaining scalable local communication. Experimental results demonstrate that our framework outperforms standard RL baselines, reducing average waiting time by over 10%. The proposed model exhibits superior generalization in unseen traffic scenarios and maintains high control stability, offering a practical solution for adaptive signal control.
State-of-the-art (SOTA) urban traffic control increasingly employs Multi-Agent Reinforcement Learning (MARL) to coordinate Traffic Light Controllers (TLCs) and Connected Autonomous Vehicles (CAVs). However, the performance of these systems is fundamentally capped by their hand-crafted, myopic rewards (e.g., intersection pressure), which fail to capture high-level, human-centric goals like safety, flow stability, and comfort. To overcome this limitation, we introduce C2T, a novel framework that learns a common-sense coordination model from traffic-vehicle dynamics. C2T distills "common-sense" knowledge from a Large Language Model (LLM) into a learned intrinsic reward function. This new reward is then used to guide the coordination policy of a cooperative multi-intersection TLC MARL system on CityFlow-based multi-intersection benchmarks. Our framework significantly outperforms strong MARL baselines in traffic efficiency, safety, and an energy-related proxy. We further highlight C2T's flexibility in principle, allowing distinct "efficiency-focused" versus "safety-focused" policies by modifying the LLM prompt.
Yuyang Chen, Kaiyan Zhao, Yiming Wang +3
The State Key Laboratory of Internet of Things for Smart City, University of Macau · Wuhan University · Computer and Information Science, University of Macau +1