Actions space: Continuous

Algorithm Description

Choosing an action

Pass the current states through the actor network, and get an action mean vector $\mu$. While in training phase, use a continuous exploration policy, such as the Ornstein-Uhlenbeck process, to add exploration noise to the action. When testing, use the mean vector $\mu$ as-is.

Training the network

Start by sampling a batch of transitions from the experience replay.

• To train the critic network, use the following targets:

First run the actor target network, using the next states as the inputs, and get $\mu (s_{t+1} )$. Next, run the critic target network using the next states and $\mu (s_{t+1} )$, and use the output to calculate $y_t$ according to the equation above. To train the network, use the current states and actions as the inputs, and $y_t$ as the targets.

• To train the actor network, use the following equation:

Use the actor's online network to get the action mean values using the current states as the inputs. Then, use the critic online network in order to get the gradients of the critic output with respect to the action mean values $\nabla _a Q(s,a)|_{s=s_t,a=\mu(s_t ) }$. Using the chain rule, calculate the gradients of the actor's output, with respect to the actor weights, given $\nabla_a Q(s,a)$. Finally, apply those gradients to the actor network.

After every training step, do a soft update of the critic and actor target networks' weights from the online networks.