Deep Q-Learning
Contents
Motivation
What is deep Q-learning, and why do we want to use it?
Brief History
Deep Q-learning was introduced in 2015 by Google's DeepMind in a Nature article called Human-level control through deep reinforcement learning. DeepMind used Q-learning backed by a deep neural network to train an agent to play Atari games from raw pixel data - often outperforming humans. Previous attempts to combine reinforcement learning with neural networks had largely failed due to unstable learning. To address these instabilities, DeepMind introduced a mechanism by which the algorithm stores all of the agent's experiences and then randomly samples and replays these experiences to provide diverse and decorrelated training data.
Advantages
Continuous state space. Instead of building a Q-table, we use a neural network as part of deep Q-learning to approximate a Q-function. There is no need to discretize our data into arbitrary, independent buckets. Unlike the rows of a table, a function is continuous - so we predict Q-values for states that we have never seen before.
Can handle high-dimensional data. Deep networks can use convolutional layers and other trickery to extract features from high-dimensional data. Table-based Q-learning would fail miserably at this task, as the curse of dimensionality would leave us with gigantic state space to explore.
Stock prediction
How might we use deep Q-learning to predict stocks? Our actions would be BUY, HOLD, or SELL, and our state space might just be one feature, RETURNS, a continuous value. Deep Q-learning is a much more natural fit to the trading problem than the Q-table implementation we did in class, where we had to discretize our technical indicator values. In theory, technical indicators derived from price are superfluous if we provide our network with raw price data - deep learning should extract these features, and perhaps better ones, on its own.
Intro
To understand deep Q-learning, it is imperative you first have an understanding of normal, table-based Q-learning. And even if you do, read over this recipe because it will be used as reference later.
Table-based Q-learning
A Q-value is the expected future reward of taking an action in a certain state. Q-learning is model free, and so we do not try to learn the transition or immediate reward functions; we are just trying to construct a model of future reward. In this case, that model is a Q-table, which is a simple matrix.
Initialization step (happens only once)
- Define a table
Q
with a row for each state and a column for each action. We will be able to index intoQ
withQ(s,a)
to get the expected utility of being in states
and taking actiona
. Alternatively, you can get an entire row containing Q-values for each action withQ(s) = [q1, q2, ..., qn]
. Initialize this table with small, random values. - Define exploration probability
0 <= p <= 1
Training step
- Be in state
s
- With probability
p
, choose a random actiona
from the action space (explore). Otherwise, choose actiona = argmaxa(Q(s,a))
(exploit). - Take action
a
and witness the rewardr
and the new state you are ins'
. - This unit of training has yielded an experience tuple -
<s, a, s', r>
. Using this experience tuple, perform value iteration with Bellman's equation.
Legend: gamma = discount rate, alpha = learning rate.
Q(s,a) = (1 - alpha)Q(s,a) + (alpha)(r + gamma*max(Q(s')))
-
Multiply
p
by some discount factor (perhaps .999).
Loop the above training steps until your epoch completes. Keep doing more epochs (reset to initial state) until the reward achieved by your agent converges and no longer improves. Note that the Q-table persists between epochs.
Deep Q-Learning Algorithm
Now that we have established the vanilla, table-based way of doing things, here is the shiny new deep Q-learning algorithm. Differences from the table-based algorithm are highlighted in blue.
The main differences are the presence of a neural network backing the Q-function, and the experience replay bucket. Experience replay works very much like dyna-q learning. We store experience tuples and later randomly sample from our bucket and train off those examples. This enables us to train off of I.I.D. experience tuples rather than just training off of sequential data, which is highly correlated.
Initialization step (happens only once)
- Define a neural network
Q
with an input node for each feature and an output node for each action. The hidden layers are defined however you like. Ultimately, providing this network with current states = [feature1, feature2, ..., featuren]
as input should generate the Q-values of each action[q1, q2, ..., qn]
as output - in other words,Q(s) = [q1, q2, ..., qn]
. Just as before,Q(s,a)
is the Q-value of taking actiona
in states
. Initialize this table with small, random values. Picking the right activation functions is critical to success - the output should be a linear activation function (outputs need to be continuous and unbounded), and the hidden layer activations should perhaps be ReLu. The loss function is also an important hyperparameter, try mean-squared error to start. - Define exploration probability
0 <= p <= 1
- Define a deque of some size. This will henceforth be referred to as
the bucket
. The bucket will hold our historical data for experience replay.
Training step
- Be in state
s
- With probability
p
, choose a random actiona
from the action space (explore). Otherwise, choose actiona = argmaxa(Q(s,a))
(exploit). - Take action
a
and witness the rewardr
and the new state you are ins'
. - This unit of training has yielded an experience tuple -
<s, a, s', r>
. Add this experience tuple to the bucket. If your bucket overflows its size, remove the oldest entry. - Experience replay step. If you have more than
BATCH_SIZE
(usually 32) entries in the bucket, sampleBATCH_SIZE
experience tuples from the bucket without replacement. For each sampled tuple<s, a, s', r>
, generate a new training instance with x-values
. The y-value of each instance is where things get a little complicated, so pay attention. Let the variableP = Q(s)
, all current Q-value predictions for all actions. If thea
from our experience tuple is the ith action in the neural net, then we update only the Q-value corresponding toa
withP[i] = r + gamma*max(Q(s'))
. The y-value of each sample is set to its respectiveP
, and the model is updated with a training session on just these samples.
Loop the above training steps until your epoch completes. Keep doing more epochs (reset to initial state) until the reward achieved by your agent converges and no longer improves. Note that the neural network weights persist between epochs.
Advanced Deep Q-Learning
DQNs tend not to be stable. Because the loss is calculated from the same network that weights are applied to, you tend to see oscillations in policies. Think of this like a cat chasing its own tail. There are a few tricks you can use to mitigate these problems.
- Target network. Instead of calculating targets with
P[i] = r + gamma*max(Q(s'))
, we have a separate target networkQ~
that we will use inP[i] = r + gamma*max(Q~(s'))
.Q~
has a structure identical toQ
. Every 1000 or so training iterations, we updateQ~
's weight to be the same asQ
's weights. In this way, there will be reduced correlation betweenQ~
andQ
, mitigating the aforementioned cat-chasing-tail scenario. - Double Q-learning. One problem in the DQN algorithm is that the agent tends to overestimate the Q-function value, due to the max in the formula used to set targets:
P[i] = r + gamma*max(Q(s'))
. The solution is to use two Q-functions, Q1 and Q2, which are independently learned. One function is then used to determine the maximizing action and second to estimate its value. For our purposes, the target networkQ~
is relatively independent ofQ
, so we can useQ
andQ~
as our networks for double Q-learning. So our update now becomesP[i] = r + gamma*Q~(s',argmaxa(Q(s',a))
.
Notes
Hyperparameters:
- Network architecture
- Weight initilizations (glorot, he...)
- Network architecture
- Optimizer
- Learning rate
- Bucket size
- Sample training batch size
- Gamma
- Explore starting probability
- Explore ending probability
- Explore probability decay rate
- Target network update interval
Resources
Cartpole DQN implementation
https://gist.github.com/tsu-nera/edd306ddeefebe4afb1efceefbc3f953
Another explanation + algorithm for deep Q-learning
http://neuro.cs.ut.ee/demystifying-deep-reinforcement-learning/
A full tutorial on how to build a DQN from scratch with keras + gym:
https://jaromiru.com/2016/09/27/lets-make-a-dqn-theory/