Attn-CommNet: Coordinated Traffic Lights Control On Large-Scale Network Level

Published in IEEE 33rd International Conference on Tools with Artificial Intelligence (ICTAI), 2021

Recommended citation: Jiashi Gao, Xinming Shi and James Jian Qiao Yu, "Attn-CommNet: Coordinated Traffic Lights Control On Large-Scale Network Level," 2021 IEEE 33rd International Conference on Tools with Artificial Intelligence (ICTAI), Washington, DC, USA, 2021, pp. 289-293, doi: 10.1109/ICTAI52525.2021.00048.

Attn-CommNet
Traffic lights control could be regarded as a multiagent coordinated problem. A model-free reinforcement learning (RL) approach is a powerful framework for solving such coordinated policy-making problems without prior environmental knowledge. In order to approach a global policy, communication among agents needs to be built. To enable dynamic and scalable communication, we propose a new RL model, CommNet based on Local Attention Mechanism (Attn-CommNet), which uses local selection and attention mechanism between hidden layers to facilitate cooperation. We evaluated the proposed method using synthetic and real word traffic flows under multi-scale road networks. The results demonstrate that the proposed method can get better performance in multi-scale problems, especially large-scale problems compared to the state-of-the-art methods.

Download paper here