deeprl_signal_control - GitLite

KO | EN

gitlite — search

#javascript #hacktoberfest #go #golang #rust #deep-learning #linux #java #awesome #awesome-list #mcp #ios #openai #open-source

deeprl_signal_control

(★ 425)

multi-agent deep reinforcement learning for large-scale traffic signal control.

File Explorer

Download Latest Version (.zip)

- result_plot-checkpoint.ipynb
- __init__.py
- models.py
- policies.py
- utils.py
- config_greedy_large.ini
- config_greedy_real.ini
- config_ia2c_large.ini
- config_ia2c_real.ini
- config_iqld_gym.ini
- config_iqld_large.ini
- config_iqld_real.ini
- config_iqll_large.ini
- config_iqll_real.ini
- config_ma2c_large.ini
- config_ma2c_real.ini
- config_test_large.ini
- config_test_real.ini
- config_test_small.ini
- __init__.py
- env.py
- large_grid_env.py
- real_net_env.py
- small_grid_env.py
- test_env.py
- 1500.png
- 2500.png
- 3500.png
- large_grid_train.png
- real_net_train.png
- - __init__.py
  - build_file.py
  - build_file_old.py
  - intersection.pdf
  - network.pdf
  - network1.pdf
  - view.xml
- __init__.py
- - __init__.py
  - build_file.py
  - intersection.pdf
  - most.sumocfg
  - network.pdf
  - view.xml
  - view1.xml
- __init__.py
- - real_net_greedy_control.csv
  - real_net_greedy_traffic.csv
  - real_net_greedy_trip.csv
  - real_net_ia2c_control.csv
  - real_net_ia2c_traffic.csv
  - real_net_ia2c_trip.csv
  - real_net_iqld_control.csv
  - real_net_iqld_traffic.csv
  - real_net_iqld_trip.csv
  - real_net_iqll_control.csv
  - real_net_iqll_traffic.csv
  - real_net_iqll_trip.csv
  - real_net_ma2c_control.csv
  - real_net_ma2c_traffic.csv
  - real_net_ma2c_trip.csv
- - ia2c_train_reward.csv
  - iqld_train_reward.csv
  - iqll_real_train_reward.csv
  - ma2c_real_train_reward.csv
- - __init__.py
  - build_file.py
  - network.pdf
- __init__.py
__init__.py
extract_tensorboard.py
LICENSE
main.py
README.md
result_plot.ipynb
setup_eval.py
setup_mac.sh
setup_ubuntu.sh
utils.py

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Deep RL for traffic signal control

[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)

This repo implements start-of-the-art mutli-agent (decentralized) deep RL algorithms for large-scale traffic signal control in SUMO-simulated environments.

Available cooperation levels:
* Centralized: a global agent that makes global control w/ global observation, reward.
* Decentralized: multiple local agents that make local control independently w/ neighborhood information sharing.

Available NN layers:
Fully-connected, LSTM.

Available algorithms:
IQL, IA2C, IA2C with stabilization (called MA2C in this paper). For more advanced algorithms, please check [deeprl_network](https://github.com/cts198859/deeprl_network).

Available environments:
* A 6-intersection benchmark traffic network. [Ye, Bao-Lin, et al. "A hierarchical model predictive control approach for signal splits optimization in large-scale urban road networks." IEEE Transactions on Intelligent Transportation Systems 17.8 (2016): 2182-2192.](https://ieeexplore.ieee.org/abstract/document/7406703/)
* A 5X5 traffic grid. [Chu, Tianshu, Shuhui Qu, and Jie Wang. "Large-scale traffic grid signal control with regional reinforcement learning." American Control Conference (ACC), 2016. IEEE, 2016.](https://ieeexplore.ieee.org/abstract/document/7525014/)
* A modified Monaco traffic network with 30 signalized intersections. [L. Codeca, J. Härri, "Monaco SUMO Traffic (MoST) Scenario: A 3D Mobility Scenario for Cooperative ITS" SUMO 2018, SUMO User Conference, Simulating Autonomous and Intermodal Transport Systems May 14-16, 2018, Berlin, Germany.](http://www.eurecom.fr/en/publication/5527/download/comsys-publi-5527.pdf) ([code](https://github.com/lcodeca/MoSTScenario))

## Requirements
* Python3==3.5
* [Tensorflow](http://www.tensorflow.org/install)==1.12.0
* [SUMO](http://sumo.dlr.de/wiki/Installing)>=1.1.0

Required packages can be installed by running `setup_mac.sh` or `setup_ubuntu.sh`.

Attention: the code on master branch is for SUMO version >= 1.1.0. Please go to branch [sumo-0.32.0](https://github.com/cts198859/deeprl_signal_control/tree/sumo-0.32.0) if you are using the old SUMO version.

## Usages
First define all hyperparameters in a config file under `[config_dir]`, and create the base directory of experiements `[base_dir]`. Before training, please call `build_file.py` under `[environment_dir]/data/` to generate SUMO network files for `small_grid` and `large_grid` environments.

1. To train a new agent, run
~~~
python3 main.py --base-dir [base_dir]/[agent] train --config-dir [config_dir] --test-mode no_test
~~~
`[agent]` is from `{ia2c, ma2c, iqll, iqld}`. `no_test` is suggested, since tests will significantly slow down the training speed.

2. To access tensorboard during training, run
~~~
tensorboard --logdir=[base_dir]/log
~~~

3. To evaluate and compare trained agents, run
~~~
python3 main.py --base-dir [base_dir] evaluate --agents [agents] --evaluation-seeds [seeds]
~~~
Evaluation data will be output to `[base_dir]/eva_data`, and make sure evaluation seeds are different from those used in training. Under default evaluation setting, the inference policy of A2C is stochastic whereas that of Q-learning is greedy (deterministic). To explicitly specifiy the inference policy type, pass argument `--evaluation-policy-type [default/stochastic/deterministic]`. Please note running a determinisitc inference policy for A2C may cause the performance loss, due to the violation of "on-policy" learning.

4. To visualize the agent behavior, run
~~~
python3 main.py --base-dir [base_dir] evaluate --agents [agent] --evaluation-seeds [seed] --demo
~~~
It is recommended to have only one agent and one evaluation seed for the demo run. This will launch the SUMO GUI, and `./large_grid/data/view.xml` can be applied to visualize queue length and intersectin delay in edge color and thickness. Below are a few example screenshots.

|   t=1500s          |      t=2500s         |     t=3500s              
:-------------------:|:--------------------:|:--------------------:
![](https://raw.githubusercontent.com/cts198859/deeprl_signal_control/master/figs/1500.png) | ![](https://raw.githubusercontent.com/cts198859/deeprl_signal_control/master/figs/2500.png) | ![](https://raw.githubusercontent.com/cts198859/deeprl_signal_control/master/figs/3500.png)

## Reproducibility
Due to SUMO version change and a few corresponding code modifications (e.g. `tau="0.5"` has to be removed from `vType` to prevent extensive vehicle collisions in simulation), it becomes difficult to reproduce paper results, which are based on SUMO 0.32.0. So we have re-run the experiments using SUMO 1.1.0 and provided the following training plots as reference. The conclusion still remains the same, that is, MA2C ~ IQL-LR > IA2C in large grid and MA2C > IA2C > IQL-LR in Monaco net. Note rather than reproducing exactly the same results, an evaluation is always valid as far as the comparison is fair, that is, fixing env config and seed across agents.

|   large grid                   |      Monaco net                   
:-------------------------------:|:------------------------------:
![](https://raw.githubusercontent.com/cts198859/deeprl_signal_control/master/figs/large_grid_train.png) | ![](https://raw.githubusercontent.com/cts198859/deeprl_signal_control/master/figs/real_net_train.png)

## Citation
If you find this useful in your research, please cite our paper "Multi-Agent Deep Reinforcement Learning for Large-Scale Traffic Signal Control" ([early access version](https://ieeexplore.ieee.org/document/8667868), [preprint version](https://arxiv.org/pdf/1903.04527.pdf)):
~~~
@article{chu2019multi,
  title={Multi-Agent Deep Reinforcement Learning for Large-Scale Traffic Signal Control},
  author={Chu, Tianshu and Wang, Jie and Codec{\`a}, Lara and Li, Zhaojian},
  journal={IEEE Transactions on Intelligent Transportation Systems},
  year={2019},
  publisher={IEEE}
}
~~~