Continuous control with deep reinforcement learning

Timothy P. Lillicrap, Jonathan J. Hunt, Alexander Pritzel, Nicolas Heess, Tom Erez, Yuval Tassa, David Silver, Daan Wierstra

2015 · ICLR

Continuous control with deep reinforcement learning

Problem

Framing

Deep RL had not made off-policy value learning work in continuous action spaces with large neural function approximators. The paper closes this gap by combining deterministic policy gradients with replay, target networks, and batch normalization, then learning directly from pixels on many MuJoCo tasks and Torcs.

Currently Used Methods

Foundational

Proposed Method

Architecture

DDPG uses an actor μ(sθμ)\mu(\mathbf{s}\mid \theta^{\mu}) and critic Q(s,aθQ)Q(\mathbf{s},\mathbf{a}\mid \theta^Q), each with target copies. For low-dimensional states, both networks use two hidden layers with 400 and 300 units; the critic injects the action at the second hidden layer, and batch normalization is applied to state inputs and intermediate layers.

Verified results page: environment screenshots for the continuous-control suite, including swing-up, reaching, locomotion, and Torcs.

Loss / Objective

The critic minimizes the Bellman residual with a target-network bootstrap:

L(θQ)=1Ni(yiQ(si,aiθQ))2,L(\theta^Q) = \frac{1}{N}\sum_i \Big(y_i - Q(\mathbf{s}_i, \mathbf{a}_i \mid \theta^Q)\Big)^2, yi=ri+γQ(si+1,μ(si+1θμ)θQ).y_i = r_i + \gamma \, Q'\big(\mathbf{s}_{i+1}, \mu'(\mathbf{s}_{i+1} \mid \theta^{\mu'}) \mid \theta^{Q'}\big).

Sampling Rule / Algorithm

The actor follows the deterministic policy gradient estimated from replay-buffer states:

θμJ1NiaQ(s,aθQ)s=si,a=μ(si)θμμ(sθμ)s=si.\nabla_{\theta^{\mu}} J \approx \frac{1}{N}\sum_i \left. \nabla_{\mathbf{a}} Q(\mathbf{s}, \mathbf{a} \mid \theta^Q) \right|_{\mathbf{s}=\mathbf{s}_i,\, \mathbf{a}=\mu(\mathbf{s}_i)} \nabla_{\theta^{\mu}} \mu(\mathbf{s} \mid \theta^{\mu}) \big|_{\mathbf{s}=\mathbf{s}_i}.

Target networks are updated by soft averaging:

θQτθQ+(1τ)θQ,θμτθμ+(1τ)θμ.\theta^{Q'} \leftarrow \tau \theta^Q + (1-\tau)\theta^{Q'}, \qquad \theta^{\mu'} \leftarrow \tau \theta^{\mu} + (1-\tau)\theta^{\mu'}.

Training Procedure

Evaluation

Datasets

Metrics

Headline results

Table 1: Performance after training for at most 2.5 million steps; random policy is 0 and planner is 1 (except Torcs).

Rav,lowdRbest,lowdRav,pixRbest,pixRav,cntrl
1.1561.5110.4661.299-0.080
0.3400.7050.8892.225-0.139
0.3031.7350.1760.6880.125
0.4000.978-0.2850.119-0.045
0.9381.3361.0961.2580.343
0.8441.1150.4821.1380.244
0.9511.0000.3350.996-0.468
0.5490.9000.1880.3230.197
0.2720.7190.1950.6420.143
0.7360.9460.4120.4270.583
0.9031.2060.4570.792-0.008
0.8491.0210.6930.9810.259
0.9240.9960.8720.9430.290
0.9541.0000.8270.9950.620
0.6550.9720.4060.7900.461
0.6180.9370.0820.7910.557
1.3111.9901.2041.431-0.031
0.6760.9360.1120.9240.078
0.4160.7220.2340.6720.198
0.4740.9360.4800.6440.416
0.9461.0210.6631.0550.099
0.7200.9870.1940.8780.231
0.5850.9430.4530.9220.204
0.4670.7390.3740.735-0.046
0.9811.1021.0001.0831.010
0.7051.5730.9441.4760.393

Ablations

Verified results figure: density plots of estimated Q versus realized return for Pendulum, Cartpole, and Cheetah.

Method Strengths and Weaknesses

Strengths

Weaknesses

Suggestions from the authors

Links

Prior Papers

Further Papers