CityFlow: A Multi-Agent Reinforcement Learning Environment for Large Scale City Traffic Scenario
Not on the map. Why?
Abstract (arXiv)
Traffic signal control is an emerging application scenario for reinforcement learning. Besides being as an important problem that affects people's daily life in commuting, traffic signal control poses its unique challenges for reinforcement learning in terms of adapting to dynamic traffic environment and coordinating thousands of agents including vehicles and pedestrians. A key factor in the success of modern reinforcement learning relies on a good simulator to generate a large number of data samples for learning. The most commonly used open-source traffic simulator SUMO is, however, not scalable to large road network and large traffic flow, which hinders the study of reinforcement learning on traffic scenarios. This motivates us to create a new traffic simulator CityFlow with fundamentally optimized data structures and efficient algorithms. CityFlow can support flexible definitions for road network and traffic flow based on synthetic and real-world data. It also provides user-friendly interface for reinforcement learning. Most importantly, CityFlow is more than twenty times faster than SUMO and is capable of supporting city-wide traffic simulation with an interactive render for monitoring. Besides traffic signal control, CityFlow could serve as the base for other transportation studies and can create new possibilities to test machine learning methods in the intelligent transportation domain.
Key facts. Cited by 114 papers in the map (data through 7 Oct 2026). First posted on arXiv 13 May 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 17; 2022: 17; 2023: 23; 2024: 20; 2025: 26; 2026: 11.
114 papers.
Sort:
By area:
- 2021MetaDrive: Composing Diverse Driving Scenarios for Generalizable Reinforcement Learningcited by 281
- 2025AgentSociety: Large-Scale Simulation of LLM-Driven Generative Agents Advances Understanding of Human Behaviors and Societycited by 180
- 2024Embodied LLM Agents Learn to Cooperate in Organized Teamscited by 81
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2023Open-TI: Open Traffic Intelligence with Augmented Language Modelcited by 38
- 2023Prompt to Transfer: Sim-to-Real Transfer for Traffic Signal Control with Prompt Learningcited by 36
- 2025A Survey of Sim-to-Real Methods in RL: Progress, Prospects and Challenges with Foundation Modelscited by 36
- 2021Learning to Simulate Self-Driven Particles System with Coordinated Policy Optimizationcited by 33
- 2024Multi-Agent Reinforcement Learning for Autonomous Driving: A Surveycited by 30
- 2021Expression might be enough: representing pressure and demand for reinforcement learning based traffic signal controlcited by 28
- 2023Choose Your Simulator Wisely: A Review on Open-source Simulators for Autonomous Drivingcited by 26
- 2024CityBench: Evaluating the Capabilities of Large Language Models for Urban Taskscited by 26
38 more
- 2021Efficient Pressure: Improving efficiency for signalized intersectionscited by 24
- 2022An Introduction to Multi-Agent Reinforcement Learning and Review of its Application to Autonomous Mobilitycited by 16
- 2022LibSignal: An Open Library for Traffic Signal Controlcited by 16
- 2025CoLLMLight: Cooperative Large Language Model Agents for Network-Wide Traffic Signal Controlcited by 15
- 2025Multi-Agent Reinforcement Learning for Resources Allocation Optimization: A Surveycited by 14
- 2023Reinforcement Learning Approaches for Traffic Signal Control under Missing Datacited by 13
- 2024FightLadder: A Benchmark for Competitive Multi-Agent Reinforcement Learningcited by 13
- 2022Modeling Network-level Traffic Flow Transitions on Sparse Datacited by 12
- 2021DynSTGAT: Dynamic Spatial-Temporal Graph Attention Network for Traffic Signal Controlcited by 11
- 2022Multi-Agent Reinforcement Learning for Traffic Signal Control through Universal Communication Methodcited by 10
- 2023TransWorldNG: Traffic Simulation via Foundation Modelcited by 9
- 2023IMP-MARL: a Suite of Environments for Large-scale Infrastructure Management Planning via MARLcited by 9
- 2021Objective-aware Traffic Simulation via Inverse Reinforcement Learningcited by 8
- 2022Consensus Learning for Cooperative Multi-Agent Reinforcement Learningcited by 8
- 2022The Real Deal: A Review of Challenges and Opportunities in Moving Reinforcement Learning-Based Traffic Signal Control Systems Towards Realitycited by 8
- 2021Space Meets Time: Local Spacetime Neural Network For Traffic Flow Forecastingcited by 7
- 2021Leveraging Queue Length and Attention Mechanisms for Enhanced Traffic Signal Control Optimizationcited by 7
- 2023Off-the-Grid MARL: Datasets with Baselines for Offline Multi-Agent Reinforcement Learningcited by 7
- 2023DenseLight: Efficient Control for Large-scale Traffic Signals with Dense Feedbackcited by 7
- 2023FusionPlanner: A Multi-task Motion Planner for Mining Trucks via Multi-sensor Fusioncited by 7
- 2021MetaVIM: Meta Variationally Intrinsic Motivated Reinforcement Learning for Decentralized Traffic Signal Controlcited by 6
- 2021Learning to Simulate on Sparse Trajectory Datacited by 6
- 2022CBLab: Supporting the Training of Large-scale Traffic Control Policies with Scalable Traffic Simulationcited by 6
- 2023Large-Scale Traffic Signal Control Using Constrained Network Partition and Adaptive Deep Reinforcement Learningcited by 6
- 2023From Model-Based to Data-Driven Simulation: Challenges and Trends in Autonomous Drivingcited by 6
- 2024Heterogeneous Multi-Agent Reinforcement Learning for Zero-Shot Scalable Collaborationcited by 6
- 2021Independent Reinforcement Learning for Weakly Cooperative Multiagent Traffic Control Problemcited by 5
- 2022Offline Reinforcement Learning for Road Traffic Controlcited by 5
- 2024Towards Multi-agent Reinforcement Learning based Traffic Signal Control through Spatio-temporal Hypergraphscited by 5
- 2024MalLight: Influence-Aware Coordinated Traffic Signal Control for Traffic Signal Malfunctionscited by 5
- 2025Fox in the Henhouse: Supply-Chain Backdoor Attacks Against Reinforcement Learningcited by 5
- 2021Two-stage training algorithm for AI robot soccercited by 4
- 2021Learning to Share in Multi-Agent Reinforcement Learningcited by 4
- 2024CityFlowER: An Efficient and Realistic Traffic Simulator with Embedded Machine Learning Modelscited by 4
- 2024SynTraC: A Synthetic Dataset for Traffic Signal Control from Traffic Monitoring Camerascited by 4
- 2024Putting Data at the Centre of Offline Multi-Agent Reinforcement Learningcited by 4
- 2024DiffLight: A Partial Rewards Conditioned Diffusion Model for Traffic Signal Control with Missing Datacited by 4
- 2024Goal-Conditioned Data Augmentation for Offline Reinforcement Learningcited by 4
50 most cited of 114 All 114 (JSON)
- 2026ProcessLight: Process Supervision for Large Language Model Based Traffic Signal Control19 Sep 2026
- 2026SIGMA: Symmetry-aware, Intelligent, Geometric, Multi-objective Adaptive Control for Robust, Dependable Traffic Management18 Aug 2026
- 2026Large Multimodal Agents for Intelligent Transportation Systems: Architectures, Evidence, and Deployment Challenges8 Aug 2026
- 2026OverFlowLight: Real-Time Gridlock Prevention and Traffic Signal Optimization for Urban Intersections24 May 2026
- 2026DeCoR: Design and Control Co-Optimization for Urban Streets Using Reinforcement Learning20 May 2026
- 2026OracleTSC: Oracle-Informed Reward Hurdle and Uncertainty Regularization for Traffic Signal Control8 May 2026
- 2026C2T: Captioning-Structure and LLM-Aligned Common-Sense Reward Learning for Traffic--Vehicle Coordination10 Apr 2026
- 2026A Robust and Efficient Multi-Agent Reinforcement Learning Framework for Traffic Signal Control12 Mar 2026
- 2026UCATSC: Uncertainty-Aware Constrained Traffic Signal Control Under Vision-Based Partial Observability8 Feb 2026
- 2026Spatiotemporal Decision Transformer for Traffic Coordination2 Feb 2026
- 2026Dimensional Peeking for Low-Variance Gradients in Zeroth-Order Discrete Optimization via Simulation21 Jan 2026
- 2025TrafficSimAgent: A Hierarchical Agent Framework for Autonomous Traffic Simulation with MCP Control24 Dec 2025
38 more
- 2025StarCraft+: Benchmarking Multi-agent Algorithms in Adversary Paradigm18 Dec 2025
- 2025Evolutionary Discovery of Heuristic Policies for Traffic Signal Control28 Nov 2025
- 2025MAESTRO: Multi-Agent Environment Shaping through Task and Reward Optimization24 Nov 2025
- 2025A Dual Large Language Models Architecture with Herald Guided Prompts for Parallel Fine Grained Traffic Signal Control31 Oct 2025
- 2025TranSimHub:A Unified Air-Ground Simulation Platform for Multi-Modal Perception and Decision-Making17 Oct 2025
- 2025The Distribution Shift Problem in Transportation Networks using Reinforcement Learning and AI18 Sep 2025
- 2025A Hierarchical Deep Reinforcement Learning Framework for Traffic Signal Control with Predictable Cycle Planning3 Sep 2025
- 2025cMALC-D: Contextual Multi-Agent LLM-Guided Curriculum Learning with Diversity-Based Context Blending28 Aug 2025
- 2025GPLight+: A Genetic Programming Method for Learning Symmetric Traffic Signal Control Policy22 Aug 2025
- 2025Joint-Local Grounded Action Transformation for Sim-to-Real Transfer in Multi-Agent Traffic Control21 Jul 2025
- 2025Toward Adaptable Multi-Agent Reinforcement Learning: An Assumption-Aware Review14 Jul 2025
- 2025Towards Physics-informed Diffusion for Anomaly Detection in Trajectories8 Jun 2025
- 2025Fox in the Henhouse: Supply-Chain Backdoor Attacks Against Reinforcement Learning26 May 2025
- 2025SAINT: Attention-Based Policies for Discrete Combinatorial Action Spaces17 May 2025
- 2025DARLR: Dual-Agent Offline Reinforcement Learning for Recommender Systems with Dynamic Reward12 May 2025
- 2025Multi-Agent Reinforcement Learning for Resources Allocation Optimization: A Survey29 Apr 2025
- 2025Sky-Drive: A Distributed Multi-Agent Simulation Platform for Human-AI Collaborative and Socially-Aware Future Transportation25 Apr 2025
- 2025A Constrained Multi-Agent Reinforcement Learning Approach to Autonomous Traffic Signal Control30 Mar 2025
- 2025CoLLMLight: Cooperative Large Language Model Agents for Network-Wide Traffic Signal Control14 Mar 2025
- 2025RouteRL: Multi-agent reinforcement learning framework for urban route choice with autonomous vehicles27 Feb 2025
- 2025Communication Strategy on Macro-and-Micro Traffic State in Cooperative Deep Reinforcement Learning for Regional Traffic Signal Control18 Feb 2025
- 2025A Survey of Sim-to-Real Methods in RL: Progress, Prospects and Challenges with Foundation Models18 Feb 2025
- 2025FitLight: Federated Imitation Learning for Plug-and-Play Autonomous Traffic Signal Control17 Feb 2025
- 2025AgentSociety: Large-Scale Simulation of LLM-Driven Generative Agents Advances Understanding of Human Behaviors and Society12 Feb 2025
- 2025Planning Under Observation Mismatch for Traffic Signal Control via Adaptive Modular World Models5 Jan 2025
- 2024Goal-Conditioned Data Augmentation for Offline Reinforcement Learning29 Dec 2024
- 2024OffLight: An Offline Multi-Agent Reinforcement Learning Framework for Traffic Signal Control10 Nov 2024
- 2024DiffLight: A Partial Rewards Conditioned Diffusion Model for Traffic Signal Control with Missing Data30 Oct 2024
- 2024PyTSC: A Unified Platform for Multi-Agent Reinforcement Learning in Traffic Signal Control23 Oct 2024
- 2024Putting Data at the Centre of Offline Multi-Agent Reinforcement Learning18 Sep 2024
- 2024TrafficGamer: Reliable and Flexible Traffic Simulation for Safety-Critical Scenarios with Game-Theoretic Oracles28 Aug 2024
- 2024MalLight: Influence-Aware Coordinated Traffic Signal Control for Traffic Signal Malfunctions19 Aug 2024
- 2024Multi-Agent Reinforcement Learning for Autonomous Driving: A Survey19 Aug 2024
- 2024SynTraC: A Synthetic Dataset for Traffic Signal Control from Traffic Monitoring Cameras18 Aug 2024
- 2024CityBench: Evaluating the Capabilities of Large Language Models for Urban Tasks20 Jun 2024
- 2024A GPU-accelerated Large-scale Simulator for Transportation System Optimization Benchmarking15 Jun 2024
- 2024FightLadder: A Benchmark for Competitive Multi-Agent Reinforcement Learning4 Jun 2024
- 2024Towards Multi-agent Reinforcement Learning based Traffic Signal Control through Spatio-temporal Hypergraphs17 Apr 2024
50 newest of 114 All 114 (JSON)
- 2021Objective-aware Traffic Simulation via Inverse Reinforcement Learningcited by 8
- 2023Off-the-Grid MARL: Datasets with Baselines for Offline Multi-Agent Reinforcement Learningcited by 7
- 2022Offline Reinforcement Learning for Road Traffic Controlcited by 5
- 2024Putting Data at the Centre of Offline Multi-Agent Reinforcement Learningcited by 4
- 2024Goal-Conditioned Data Augmentation for Offline Reinforcement Learningcited by 4
- 2024OffLight: An Offline Multi-Agent Reinforcement Learning Framework for Traffic Signal Controlcited by 3
- 2023A Fully Data-Driven Approach for Realistic Traffic Signal Control Using Offline Reinforcement Learningcited by 1
- 2025FitLight: Federated Imitation Learning for Plug-and-Play Autonomous Traffic Signal Controlcited by 0
- 2025FitLight: Federated Imitation Learning for Plug-and-Play Autonomous Traffic Signal Control17 Feb 2025
- 2024Goal-Conditioned Data Augmentation for Offline Reinforcement Learning29 Dec 2024
- 2024OffLight: An Offline Multi-Agent Reinforcement Learning Framework for Traffic Signal Control10 Nov 2024
- 2024Putting Data at the Centre of Offline Multi-Agent Reinforcement Learning18 Sep 2024
- 2023A Fully Data-Driven Approach for Realistic Traffic Signal Control Using Offline Reinforcement Learning27 Nov 2023
- 2023Off-the-Grid MARL: Datasets with Baselines for Offline Multi-Agent Reinforcement Learning1 Feb 2023
- 2022Offline Reinforcement Learning for Road Traffic Control7 Jan 2022
- 2021Objective-aware Traffic Simulation via Inverse Reinforcement Learning20 May 2021
- 2021MetaDrive: Composing Diverse Driving Scenarios for Generalizable Reinforcement Learningcited by 281
- 2025AgentSociety: Large-Scale Simulation of LLM-Driven Generative Agents Advances Understanding of Human Behaviors and Societycited by 180
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2024CityBench: Evaluating the Capabilities of Large Language Models for Urban Taskscited by 26
- 2025DARLR: Dual-Agent Offline Reinforcement Learning for Recommender Systems with Dynamic Rewardcited by 2
- 2025Planning Under Observation Mismatch for Traffic Signal Control via Adaptive Modular World Modelscited by 0
- 2025DARLR: Dual-Agent Offline Reinforcement Learning for Recommender Systems with Dynamic Reward12 May 2025
- 2025AgentSociety: Large-Scale Simulation of LLM-Driven Generative Agents Advances Understanding of Human Behaviors and Society12 Feb 2025
- 2025Planning Under Observation Mismatch for Traffic Signal Control via Adaptive Modular World Models5 Jan 2025
- 2024CityBench: Evaluating the Capabilities of Large Language Models for Urban Tasks20 Jun 2024
- 2022A Survey on Model-based Reinforcement Learning19 Jun 2022
- 2021MetaDrive: Composing Diverse Driving Scenarios for Generalizable Reinforcement Learning26 Sep 2021
- 2024Embodied LLM Agents Learn to Cooperate in Organized Teamscited by 81
- 2025CoLLMLight: Cooperative Large Language Model Agents for Network-Wide Traffic Signal Controlcited by 15
- 2025TrafficSimAgent: A Hierarchical Agent Framework for Autonomous Traffic Simulation with MCP Controlcited by 3
- 2025cMALC-D: Contextual Multi-Agent LLM-Guided Curriculum Learning with Diversity-Based Context Blendingcited by 2
- 2025MAESTRO: Multi-Agent Environment Shaping through Task and Reward Optimizationcited by 1
- 2025TrafficSimAgent: A Hierarchical Agent Framework for Autonomous Traffic Simulation with MCP Control24 Dec 2025
- 2025MAESTRO: Multi-Agent Environment Shaping through Task and Reward Optimization24 Nov 2025
- 2025cMALC-D: Contextual Multi-Agent LLM-Guided Curriculum Learning with Diversity-Based Context Blending28 Aug 2025
- 2025CoLLMLight: Cooperative Large Language Model Agents for Network-Wide Traffic Signal Control14 Mar 2025
- 2024Embodied LLM Agents Learn to Cooperate in Organized Teams19 Mar 2024
- 2024Multi-Agent Reinforcement Learning for Autonomous Driving: A Surveycited by 30
- 2025RouteRL: Multi-agent reinforcement learning framework for urban route choice with autonomous vehiclescited by 2
- 2024TrafficGamer: Reliable and Flexible Traffic Simulation for Safety-Critical Scenarios with Game-Theoretic Oraclescited by 1
- 2025Sky-Drive: A Distributed Multi-Agent Simulation Platform for Human-AI Collaborative and Socially-Aware Future Transportationcited by 0
- 2025Sky-Drive: A Distributed Multi-Agent Simulation Platform for Human-AI Collaborative and Socially-Aware Future Transportation25 Apr 2025
- 2025RouteRL: Multi-agent reinforcement learning framework for urban route choice with autonomous vehicles27 Feb 2025
- 2024TrafficGamer: Reliable and Flexible Traffic Simulation for Safety-Critical Scenarios with Game-Theoretic Oracles28 Aug 2024
- 2024Multi-Agent Reinforcement Learning for Autonomous Driving: A Survey19 Aug 2024
- 2023Deploying Offline Reinforcement Learning with Human Feedback13 Mar 2023
- 2023Open-TI: Open Traffic Intelligence with Augmented Language Modelcited by 38
- 2023Prompt to Transfer: Sim-to-Real Transfer for Traffic Signal Control with Prompt Learningcited by 36
- 2025A Survey of Sim-to-Real Methods in RL: Progress, Prospects and Challenges with Foundation Modelscited by 36
- 2021Learning to Simulate Self-Driven Particles System with Coordinated Policy Optimizationcited by 33
- 2021Expression might be enough: representing pressure and demand for reinforcement learning based traffic signal controlcited by 28
- 2023Choose Your Simulator Wisely: A Review on Open-source Simulators for Autonomous Drivingcited by 26
- 2021Efficient Pressure: Improving efficiency for signalized intersectionscited by 24
- 2022An Introduction to Multi-Agent Reinforcement Learning and Review of its Application to Autonomous Mobilitycited by 16
- 2022LibSignal: An Open Library for Traffic Signal Controlcited by 16
- 2025Multi-Agent Reinforcement Learning for Resources Allocation Optimization: A Surveycited by 14
- 2023Reinforcement Learning Approaches for Traffic Signal Control under Missing Datacited by 13
- 2024FightLadder: A Benchmark for Competitive Multi-Agent Reinforcement Learningcited by 13
12 most cited of 88 All 88 (JSON)
- 2026SIGMA: Symmetry-aware, Intelligent, Geometric, Multi-objective Adaptive Control for Robust, Dependable Traffic Management18 Aug 2026
- 2026Large Multimodal Agents for Intelligent Transportation Systems: Architectures, Evidence, and Deployment Challenges8 Aug 2026
- 2026OverFlowLight: Real-Time Gridlock Prevention and Traffic Signal Optimization for Urban Intersections24 May 2026
- 2026DeCoR: Design and Control Co-Optimization for Urban Streets Using Reinforcement Learning20 May 2026
- 2026OracleTSC: Oracle-Informed Reward Hurdle and Uncertainty Regularization for Traffic Signal Control8 May 2026
- 2026C2T: Captioning-Structure and LLM-Aligned Common-Sense Reward Learning for Traffic--Vehicle Coordination10 Apr 2026
- 2026A Robust and Efficient Multi-Agent Reinforcement Learning Framework for Traffic Signal Control12 Mar 2026
- 2026UCATSC: Uncertainty-Aware Constrained Traffic Signal Control Under Vision-Based Partial Observability8 Feb 2026
- 2026Spatiotemporal Decision Transformer for Traffic Coordination2 Feb 2026
- 2026Dimensional Peeking for Low-Variance Gradients in Zeroth-Order Discrete Optimization via Simulation21 Jan 2026
- 2025StarCraft+: Benchmarking Multi-agent Algorithms in Adversary Paradigm18 Dec 2025
- 2025Evolutionary Discovery of Heuristic Policies for Traffic Signal Control28 Nov 2025
12 newest of 88 All 88 (JSON)