Reinforcement learning (rl)-based traffic signal control (tsc) method and apparatus, device, medium, and product

Abstract

Provided are a reinforcement learning (RL)-based traffic signal control (TSC) method and apparatus, a device, a medium, and a product. The TSC method includes: obtaining traffic state data of a target intersection at a current time point and a road network graph, where the traffic state data includes a quantity of lanes at the target intersection and a traffic flow of each of the lanes; inputting the traffic state data and the road network graph into a preset traffic signal prediction model, and obtaining a target phase action output by the traffic signal prediction model, where the traffic signal prediction model includes a spatiotemporal encoder and a return-based action decoder, and the traffic signal prediction model is obtained through training based on return-based contrastive learning; and controlling, based on the target phase action, a traffic light at the target intersection to execute the target phase action.

Dependencies are inferred from the claim text.