A Spatial-Temporal Graph Attention Network for Multi-intersection Traffic Light Control

Liu He,Qing'an Li,Libing Wu,Min Wang,Jianxin Li,Dan Wu
DOI: https://doi.org/10.1109/IJCNN52387.2021.9533335
2021-01-01
Abstract:Traffic light control is an extremely challenging problem in transportation. Recently, an increasing number of studies have employed reinforcement learning approaches to deal with traffic light control problems. However, these studies often observe the target intersection independently and ignore the dynamic effects from the surrounding intersections. Moreover, the dependence of historical information on current traffic conditions is also not fully exploited in multi-intersection traffic light control. In this work, we propose a spatial-temporal graph attention network based multi-agent deep Q-learning model. Our model can capture not only spatial features but also temporal features. Specifically, we first utilize the Graph Attention Network (GAT) to obtain the spatial relationship between the central intersection and surrounding intersections. Then Long Short-Term Memory (LSTM) network and the self-attention mechanism are used to get the time dimension information. Finally, we adopt a deep Q-learning network (DQN) to predict the action-value of each feasible action and choose the signal phase according to the size of the action-value. Experimental results on synthetic and real-world datasets confirm that our method can lead to both the least travel time and the maximum throughput, compared with some existing baselines.
What problem does this paper attempt to address?