ADLight: A Universal Approach of Traffic Signal Control with Augmented Data Using Reinforcement Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wang, Maonan, Xu, Yutong, Xiong, Xi, Kan, Yuheng, Xu, Chengcheng, Pun, Man-On
Format: Preprint
Published: 2022
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913388056018944
author Wang, Maonan
Xu, Yutong
Xiong, Xi
Kan, Yuheng
Xu, Chengcheng
Pun, Man-On
author_facet Wang, Maonan
Xu, Yutong
Xiong, Xi
Kan, Yuheng
Xu, Chengcheng
Pun, Man-On
contents Traffic signal control has the potential to reduce congestion in dynamic networks. Recent studies show that traffic signal control with reinforcement learning (RL) methods can significantly reduce the average waiting time. However, a shortcoming of existing methods is that they require model retraining for new intersections with different structures. In this paper, we propose a novel reinforcement learning approach with augmented data (ADLight) to train a universal model for intersections with different structures. We propose a new agent design incorporating features on movements and actions with set current phase duration to allow the generalized model to have the same structure for different intersections. A new data augmentation method named \textit{movement shuffle} is developed to improve the generalization performance. We also test the universal model with new intersections in Simulation of Urban MObility (SUMO). The results show that the performance of our approach is close to the models trained in a single environment directly (only a 5% loss of average waiting time), and we can reduce more than 80% of training time, which saves a lot of computational resources in scalable operations of traffic lights.
format Preprint
id arxiv_https___arxiv_org_abs_2210_13378
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle ADLight: A Universal Approach of Traffic Signal Control with Augmented Data Using Reinforcement Learning
Wang, Maonan
Xu, Yutong
Xiong, Xi
Kan, Yuheng
Xu, Chengcheng
Pun, Man-On
Systems and Control
Artificial Intelligence
Machine Learning
Traffic signal control has the potential to reduce congestion in dynamic networks. Recent studies show that traffic signal control with reinforcement learning (RL) methods can significantly reduce the average waiting time. However, a shortcoming of existing methods is that they require model retraining for new intersections with different structures. In this paper, we propose a novel reinforcement learning approach with augmented data (ADLight) to train a universal model for intersections with different structures. We propose a new agent design incorporating features on movements and actions with set current phase duration to allow the generalized model to have the same structure for different intersections. A new data augmentation method named \textit{movement shuffle} is developed to improve the generalization performance. We also test the universal model with new intersections in Simulation of Urban MObility (SUMO). The results show that the performance of our approach is close to the models trained in a single environment directly (only a 5% loss of average waiting time), and we can reduce more than 80% of training time, which saves a lot of computational resources in scalable operations of traffic lights.
title ADLight: A Universal Approach of Traffic Signal Control with Augmented Data Using Reinforcement Learning
topic Systems and Control
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2210.13378