Multi-Agent Reinforcement Learning Based on Representational Communication for Large-Scale Traffic Signal Control

TL;DR

This paper introduces a novel multi-agent reinforcement learning framework for large-scale traffic signal control that enables agents to selectively communicate with relevant neighbors, improving scalability and reducing noise.

Contribution

It proposes a flexible, selective communication mechanism allowing agents to choose message recipients and variable message lengths, enhancing coordination in large-scale TSC.

Findings

Achieved lowest network congestion compared to related methods.

Agents utilized approximately 47-65% of the communication channel.

Ablation studies confirmed the effectiveness of learned communication policies.

Abstract

Traffic signal control (TSC) is a challenging problem within intelligent transportation systems and has been tackled using multi-agent reinforcement learning (MARL). While centralized approaches are often infeasible for large-scale TSC problems, decentralized approaches provide scalability but introduce new challenges, such as partial observability. Communication plays a critical role in decentralized MARL, as agents must learn to exchange information using messages to better understand the system and achieve effective coordination. Deep MARL has been used to enable inter-agent communication by learning communication protocols in a differentiable manner. However, many deep MARL communication frameworks proposed for TSC allow agents to communicate with all other agents at all times, which can add to the existing noise in the system and degrade overall performance. In this study, we propose a communication-based MARL framework for large-scale TSC. Our framework allows each agent to learn a communication policy that dictates "which" part of the message is sent "to whom". In essence, our framework enables agents to selectively choose the recipients of their messages and exchange variable length messages with them. This results in a decentralized and flexible communication mechanism in which agents can effectively use the communication channel only when necessary. We designed two networks, a synthetic $4 \times 4$ grid network and a real-world network based on the Pasubio neighborhood in Bologna. Our framework achieved the lowest network congestion compared to related methods, with agents utilizing $\sim 47-65 \%$ of the communication channel. Ablation studies further demonstrated the effectiveness of the communication policies learned within our framework.