Sample-Efficient Reinforcement Learning with Bootstrapped Dual Policy Iteration
This repository contains the complete implementation of the Bootstrapped Dual Policy Iteration algorithm we developed over the past year and a half. The repository also contains scripts to re-run and re-plot our experiments.
What is BDPI?
BDPI is a model-free reinforcement-learning algorithm for discrete action spaces, continuous or discrete state spaces, written in Python with PyTorch and to be used with OpenAI Gym environments. This implementation, by its use of feed-forward neural networks, is tailored to Markov Decision Processes (not POMDPs).
How is it different?
Many reinforcement learning algorithms exist. BDPI is different from them in several key points:
- It uses one actor and several critics. Bootstrapped DQN uses several critics but no actor, A3C uses several actors but no critics.
- The critics are off-policy. Conventional actor-critic algorithms have their critic evaluate the actor, so that the actor can use the critic to get slightly better. BDPI has its critics learn the optimal Q-Function, without paying attention to what the actor does. The actor then imitates the critics.
- BDPI is good at exploration while being extremely simple. Algorithms such as RUDDER are very good at exploring, and Information-Directed Sampling seems quite promising in stochastic environments. However, these algorithms are long, complex and difficult to implement (but also extremely interesting to study!). BDPI combines simple critics with a simple actor, using less than 500 lines of Python code (and those lines include many variants and configuration parameters we use in the paper). Yet, BDPI exhibits exploration superior to any non-exploration-focused algorithm, such as PPO, ACKTR, Bootstrapped DQN, A3C and ACER.
- BDPI is almost trivial to apply to new tasks. We configured BDPI for FrozenLake, an easy-to-explore but highly-stochastic gridworld. The parameters that performed well (
--layers 1 --hidden 32 --er 256 --erfreq 1 --actor-count 16 --loops 16 --qloops 4 --epochs 20 --lr 0.0001) were used as-is on LunarLander and Table, continuous-state environments that are much harder to explore. And these parameters worked flawlessly on the first try, crushing all the other algorithms. We also evaluated BDPI, with these same parameters, on other environments developed in our lab (one of them is LargeGrid, available in this repository). Again, these parameters provided state-of-the-art performance (see poster.pdf, which has results on LargeGrid).
- BDPI is not gradient-based, contrary to any other actor-critic algorithm, and most recent critic-only algorithms. Our equations produce target Q-Values and probabilities, that can simply be stored in a table, regressed with an SVM, put into a Gaussian mixture model, or fitted with a neural network, on the mean-squared-error loss, for as many training epochs as desired.
The poster presents the general architecture of BDPI:
Explaining BDPI can be done by following the loop:
- The actor receives a state and outputs a probability distribution over actions. The agent samples one action and executes it in the environment. Note that BDPI does not assume the actor to be the behavior policy, so if you want to tamper with what the actor outputs (to add safety for instance), you can. BDPI is for practical real-world RL :) .
- The environment produces a next-state and a reward. That, combined with the current state and action, forms an experience added to the experience buffer.
- Every so often (16 times per time-step in our experiment, for maximum sample-efficiency):
- A batch of experiences is sampled from the experience buffer.
- A critic, among 16, is sampled from the set of critics.
- The chosen critic is updated with our Aggressive Bootstrapped Clipped DQN update rule, on the batch of experiences.
- The updated Q-Values are used to compute greedy actions. The agent now has a set of states (from the experiences) mapped to greedy actions.
- The actor is trained to imitate these greedy actions
Because the actor has a learning rate, it learns the expected greedy policy of the critics, that, being trained off-policy, learn the optimal Q-Function. And because the greedy policy of the optimal Q-Function is the optimal policy, the actor learns the optimal policy. The way the actor moves towards that optimal policy closely resembles Thompson sampling, empirically shown to be an extremely good exploration strategy. This may explain why our results look so "BDPI learns immediately while the other algorithms are flat lines", BDPI almost perfectly balances exploration and exploitation, even though it does not have any advanced feature like novelty-based exploration, reward shaping, or reward rewriting.
The following components are available in this repository
- The complete Python 3 source code of our algorithm;
- Two OpenAI Gym environments: Five Rooms and Table (FrozenLake8x8 and LunarLander are available in the OpenAI Gym);
- Scripts to re-run all our experiments in a fully automated way.
- A benchmark that we use to evaluate hardware on a scientific workload.
The files are organized as follows:
gym_envs/: Gym environments.
main.py: A simple RL agent that performs actions in a Gym environment, and learns using BDPI
bdpi.py: The BDPI learning algorithm (actor, critics, and glue between them)
experiments_gym.sh: A script that produces a job description for a given environment. Run an experiment with
./experiments_gym.sh table && cat commands_table.sh | parallel -jCORES
Reproducing our results require a computer with the following components:
- A recent Linux distribution
- Python 3, with
- GNU Parallel