Authors: Dong-Kyum Kim¹ and Hawoong Jeong^1,2

¹ _{Department of Physics, KAIST} ² _{Center for Complex Systems, KAIST}

This repo contains source code for the runs in Deep reinforcement learning for feedback control in a collective flashing ratchet.

git clone https://github.com/kdkyum/RatchetDRL
cd RatchetDRL
conda create -y --name ratchet python=3.7
conda activate ratchet
pip install -r requirements.txt
python -m ipykernel install --name ratchet
export PYTHONPATH='.'

See option details by running the following command

python main.py --help

The training process is logged in data/runs directory. You can inspect the training process by tensorboard (run tensorboard --logdir data/runs).

# N=2, Smooth potential.
python main.py --env A --N 2

# N=2, Sawtooth potential.
python main.py --env B --N 2 

# N=4, PPO algorithm with DeepSets (ds) architecture, 10 time-steps are delayed. 
python main.py -a ds --env A_delay --N 4 --delay 10 

# N=4, PPO algorithm with RNN architecture, 10 time-steps are delayed.
python main_rnn.py --env A_delay --N 4 --delay 10 

• data/results contains the all results (csv files) of the runs in the paper.
• data/runs contains the training logs and trained policy and value networks.

• notebook files for plotting the figures in the paper.
  • data/trained_policies.ipynb: Figs. 1(a) and 2(b) (Decision boundaries for N=1 and N=2 cases).
  • data/ppo_results.ipynb: Fig. 2(c) (Results of PPO algorithm).
  • data/time-delay_results.ipynb: Fig. 3(b) (Results of time-delayed feedback controls).

This repository is built off the publicly released repository openai/spinningup.

Cite the following Bibtex.

@article{PhysRevResearch.3.L022002,
  title = {Deep reinforcement learning for feedback control in a collective flashing ratchet},
  author = {Kim, Dong-Kyum and Jeong, Hawoong},
  journal = {Phys. Rev. Research},
  volume = {3},
  issue = {2},
  pages = {L022002},
  numpages = {6},
  year = {2021},
  month = {Apr},
  publisher = {American Physical Society},
  doi = {10.1103/PhysRevResearch.3.L022002},
  url = {https://link.aps.org/doi/10.1103/PhysRevResearch.3.L022002}
}

This project following the MIT License.