Title: BOOSTING THE ACTOR WITH DUAL CRITIC

Abstract: This paper proposes a new actor-critic-style algorithm called Dual Actor-Critic or Dual-AC. It is derived in a principled way from the Lagrangian dual form of the Bellman optimality equation, which can be viewed as a two-player game between the actor and a critic-like function, which is named as dual critic. Compared to its actor-critic relatives, Dual-AC has the desired property that the actor and dual critic are updated cooperatively to optimize the same objective function, providing a more transparent way for learning the critic that is directly related to the objective function of the actor. We then provide a concrete algorithm that can effectively solve the minimax optimization problem, using techniques of multistep bootstrapping, path regularization, and stochastic dual ascent algorithm. We demonstrate that the proposed algorithm achieves state-of-the-art performance across several benchmarks.

1. INTRODUCTION:

Reinforcement learning (RL) algorithms aim to learn a policy that maximizes the long-term return by sequentially interacting with an unknown environment. Value-function-based algorithms first approximate the optimal value function, which can then be used to derive a good policy. These methods (Sutton, 1988; Watkins, 1989) often take advantage of the Bellman equation and use bootstrapping to make learning more sample efficient than Monte Carlo estimation (Sutton & Barto, 1998) . However, the relation between the quality of the learned value function and the quality of the derived policy is fairly weak (Bertsekas & Tsitsiklis, 1996) . Policy-search-based algorithms such as REINFORCE (Williams, 1992) and others (Kakade, 2002; Schulman et al., 2015a) , on the other hand, assume a fixed space of parameterized policies and search for the optimal policy parameter based on unbiased Monte Carlo estimates. The parameters are often updated incrementally along stochastic directions that on average are guaranteed to increase the policy quality. Unfortunately, they often have a greater variance that results in a higher sample complexity.

Actor-critic methods combine the benefits of these two classes, and have proved successful in a number of challenging problems such as robotics (Deisenroth et al., 2013) , meta-learning (Bello et al., 2016) , and games (Mnih et al., 2016 ). An actor-critic algorithm has two components: the actor (policy) and the critic (value function). As in policy-search methods, actor is updated towards the direction of policy improvement. However, the update directions are computed with the help of the critic, which can be more efficiently learned as in value-function-based methods (Sutton et al., 1999; Konda & Tsitsiklis, 2003; Peters et al., 2005; Bhatnagar et al., 2009; Schulman et al., 2015b) . Although the use of a critic may introduce bias in learning the actor, its reduces variance and thus the sample complexity as well, compared to pure policy-search algorithms.

While the use of a critic is important for the efficiency of actor-critic algorithms, it is not entirely clear how the critic should be optimized to facilitate improvement of the actor. For some parametric family of policies, it is known that a certain compatibility condition ensures the actor parameter update is an unbiased estimate of the true policy gradient (Sutton et al., 1999) . In practice, temporaldifference methods are perhaps the most popular choice to learn the critic, especially when nonlinear function approximation is used (e.g., Schulman et al. (2015b) ).

In this paper, we propose a new actor-critic-style algorithm where the actor and the critic-like function, which we named as dual critic, are trained cooperatively to optimize the same objective function. The algorithm, called Dual Actor-Critic , is derived in a principled way by solving a dual form of the Bellman equation (Bertsekas & Tsitsiklis, 1996) . The algorithm can be viewed as a two-player game between the actor and the dual critic, and in principle can be solved by standard optimization algorithms like stochastic gradient descent (Section 2). We emphasize the dual critic is not fitting the value function for current policy, but that of the optimal policy. We then show that, when function approximation is used, direct application of standard optimization techniques can result in instability in training, because of the lack of convex-concavity in the objective function (Section 3). Inspired by the augmented Lagrangian method (Luenberger & Ye, 2015; Boyd et al., 2010) , we propose path regularization for enhanced numerical stability. We also generalize the two-player game formulation to the multi-step case to yield a better bias/variance tradeoff. The full algorithm is derived and described in Section 4, and is compared to existing algorithms in Section 5. Finally, our algorithm is evaluated on several locomotion tasks in the MuJoCo benchmark (Todorov et al., 2012) , and compares favorably to state-of-the-art algorithms across the board.

Notation. We denote a discounted MDP by M = (S, A, P, R, γ), where S is the state space, A the action space, P (•|s, a) the transition probability kernel defining the distribution over next-state upon taking action a in state x, R(s, a) the corresponding immediate rewards, and γ ∈ (0, 1) the discount factor. If there is no ambiguity, we will use a f (a) and f (a)da interchangeably.

2. DUALITY OF BELLMAN OPTIMALITY EQUATION:

In this section, we first describe the linear programming formula of the Bellman optimality equation (Bertsekas, 1995; Puterman, 2014) , paving the path for a duality view of reinforcement learning via Lagrangian duality. In the main text, we focus on MDPs with finite state and action spaces for simplicity of exposition. We extend the duality view to continuous state and action spaces in Appendix A.2.

Given an initial state distribution µ(s), the reinforcement learning problem aims to find a policy π(•|s) : S → P(A) that maximizes the total expected discounted reward with P(A) denoting all the probability measures over A, i.e.,

EQUATION

where

EQUATION

which can be formulated as a linear program (Puterman, 2014; Bertsekas, 1995) :

EQUATION

s.t. V (s) R(s, a) + γE s |s,a [V (s )] , ∀(s, a) ∈ S × A.

For completeness, we provide the derivation of the above equivalence in Appendix A. Without loss of generality, we assume there exists an optimal policy for the given MDP, namely, the linear programming is solvable. The optimal policy can be obtained from the solution to the linear program

EQUATION

The dual form of the LP below is often easier to solve and yield more direct relations to the optimal policy.

EQUATION

s.t. a∈A ρ(s , a) = (1 -γ)µ(s ) + γ s,a∈S×A ρ(s, a)P (s |s, a)ds, ∀s ∈ S. Since the primal LP is solvable, the dual LP is also solvable, and P * -D * = 0. The optimal dual variables ρ * (s, a) and optimal policy π * (a|s) are closely related in the following manner:

Theorem 1 (Policy from dual variables) s,a∈S×A ρ * (s, a) = 1, and π * (a|s) = ρ * (s,a)

a∈A ρ * (s,a) .

Since the goal of reinforcement learning is to learn an optimal policy, it is appealing to deal with the Lagrangian dual which optimizes the policy directly, or its equivalent saddle point problem that jointly learns the optimal policy and value function.

Theorem 2 (Competition in one-step setting) The optimal policy π * , actor, and its corresponding value function V * , dual critic, is the solution to the following saddle-point problem

EQUATION

where

∆[V ](s, a) := R(s, a) + γE s |s,a [V (s )] -V (s).

The saddle point optimization (6) provides a game perspective in understanding the reinforcement learning problem (Goodfellow et al., 2014) . The learning procedure can be thought as a game between the dual critic, i.e., value function for optimal policy, and the weighted actor, i.e., α(s)π(a|s): the dual critic V seeks the value function to satisfy the Bellman equation, while the actor π tries to generate state-action pairs that break the satisfaction. Such a competition introduces new roles for the actor and the dual critic, and more importantly, bypasses the unnecessary separation of policy evaluation and policy improvement procedures needed in a traditional actor-critic framework.

3. SOURCES OF INSTABILITY:

To solve the dual problem in (6), a straightforward idea is to apply stochastic mirror prox (Nemirovski et al., 2009) or stochastic primal-dual algorithm (Chen et al., 2014) to address the saddle point problem in (6). Unfortunately, such algorithms have limited use beyond special cases. For example, for an MDP with finite state and action spaces, the one-step saddle-point problem ( 6) with tabular parametrization is convex-concave, and finite-sample convergence rates can be established; see e.g., Chen & Wang (2016) and Wang (2017) . However, when the state/action spaces are large or continuous so that function approximation must be used, such convergence guarantees no longer hold due to lack of convex-concavity. Consequently, directly solving (6) can suffer from severe bias and numerical issues, resulting in poor performance in practice (see, e.g., Figure 1 ):

1. Large bias in one-step Bellman operator: It is well-known that one-step bootstrapping in temporal difference algorithms has lower variance than Monte Carlo methods and often require much fewer samples to learn. But it produces biased estimates, especially when function approximation is used. Such a bias is especially troublesome in our case as it introduces substantial noise in the gradients to update the policy parameters.

2. Absence of local convexity and duality: Using nonlinear parametrization will easily break the local convexity and duality between the original LP and the saddle point problem, which are known as the necessary conditions for the success of applying primal-dual algorithm to constrained problems (Luenberger & Ye, 2015) . Thus none of the existing primal-dual type algorithms will remain stable and convergent when directly optimizing the saddle point problem without local convexity.

3. Biased stochastic gradient estimator with under-fitted value function: In the absence of local convexity, the stochastic gradient w.r.t. the policy π constructed from under-fitted value function will presumably be biased and futile to provide any meaningful improvement of the policy. Hence, naively extending the stochastic primal-dual algorithms in Chen & Wang (2016) ; Wang (2017) for the parametrized Lagrangian dual, will also lead to biased estimators and sample inefficiency.

4. DUAL ACTOR-CRITIC:

In this section, we will introduce several techniques to bypass the three instability issues in the previous section: (1) generalization of the minimax game to the multi-step case to achieve a better bias-variance tradeoff;

(2) use of path regularization in the objective function to promote local convexity and duality; and (3) use of stochastic dual ascent to ensure unbiased gradient estimates.

4.1. COMPETITION IN MULTI-STEP SETTING:

In this subsection, we will extend the minimax game between the actor and critic to the multi-step setting, which has been widely utilized in temporal-difference algorithms for better bias/variance tradeoffs (Sutton & Barto, 1998; Kearns & Singh, 2000) . By the definition of the optimal value function, it is easy to derive the k-step Bellman optimality equation as

V * (s) = T k V * (s) := max π∈P E π k i=0 γ i R(s i , a i ) + γ k+1 E π [V * (s k+1 )] . (7)

Similar to the one-step case, we can reformulate the multi-step Bellman optimality equation into a form similar to the LP formulation, and then we establish the duality, which leads to the following mimimax problem:

Theorem 3 (Competition in multi-step setting) The optimal policy π * and its corresponding value function V * is the solution to the following saddle point problem

EQUATION

where

δ {s i , a i } k i=0 , s k+1 = k i=0 γ i R(s i , a i ) + γ k+1 V (s k+1 ) -V (s) and E π α δ {s i , a i } k i=0 , s k+1 = {si,ai} k i=0 ,s k+1 α(s 0 ) k i=0 π(a i |s i )p(s i+1 |s i , a i )δ {s i , a i } k i=0 , s k+1 .

The saddle-point problem ( 8) is similar to the one-step Lagrangian (6): the dual critic, V , and weighted k-step actor, α(s 0 ) k i=0 π(a i |s i ), are competing for an equilibrium, in which critic and actor become the optimal value function and optimal policy. However, it should be emphasized that due to the existence of max-operator over the space of distributions P(A), rather than A, in the multi-step Bellman optimality equation ( 7), the establishment of the competition in multi-step setting in Theorem 3 is not straightforward: i), its corresponding optimization is no longer a linear programming; ii), the strong duality in ( 8) is not obvious because of the lack of the convex-concave structure. We first generalize the duality to multi-step setting. Due to space limit, detailed analyses for generalizing the competition to multi-step setting are provided in Appendix B.

4.2. PATH REGULARIZATION:

When function approximation is used, the one-step or multi-step saddle-point problems (8) will no longer be convex in the primal parameters. This could lead to instability and even divergence when solved by brute-force stochastic primal-dual algorithms. One then desires to partially convexify the objectives without affecting the optimal solutions. The augmented Lagrangian method (Boyd et al., 2010; Luenberger & Ye, 2015) , also known as the method of multipliers, is designed and widely used for such purposes. However, directly applying this method would require introducing penalty functions of the multi-step Bellman operator, which renders extra complexity and challenges in optimization. Interested readers are referred to Appendix B.2 for further details.

Instead, we propose to use path regularization, as a stepping stone for promoting local convexity and computation efficiency. The regularization term is motivated by the fact that the optimal value function satisfies the constraint

V (s) = E π * ∞ i=0 γ i R(s i , a i )|s .

In the same spirit as augmented Lagrangian, we will introduce to the objective the simple penalty function

E s∼µ(s) E π b ∞ i=0 γ i R(s i , a i ) -V (s)

2 , leading to the following:

L r (V, α, π) := (1 -γ k+1 )E µ [V (s)] + E π α δ {s i , a i } k i=0 , s k+1 (9) + η V E s∼µ(s) E π b ∞ i=0 γ i R(s i , a i ) -V (s) 2 ,

where η V 0 is a hyper-parameter controlling the strength of the regularization.

Note that in the penalty function above we use a behavior policy π b instead of an optimal policy, since the latter is unknown. Adding such a regularization enables local duality in the primal parameters. Indeed, this can be easily verified by showing the positive definiteness of the Hessian at a local solution. We call this approach path regularization, since it exploits the rewards in the sample path to regularize the solution path of value function V in the optimization procedure. As a by-product, the regularization also provides a mechanism to utilize off-policy samples from behavior policy π b .

One can also see that the regularization indeed provides guidance and preference to search for the solution path. Specifically, in each update of V the learning procedure, it tries to move towards the optimal value function while staying close to the value function of the behavior policy π b . Intuitively, such regularization restricts the feasible domain of candidates V to be a ball centered at V π b . Besides enhancing local convexity, such a penalty also avoids unboundedness of V in the learning procedure, and thus more numerical robust. As long as the optimal value function is indeed in such a region, the introduced side-effect can be controlled. Formally, we can show that with appropriate η V , the optimal solution (V * , α * , π * ) is not affected. The main results of this subsection are summarized by the following theorem.

Theorem 4 (Property of path regularization) The local duality holds for L r (V, α, π). Denote (V * , α * , π * ) as the solution to Bellman optimality equation, with some appropriate η V ,

(V * , α * , π * ) = argmax α∈P(S),π∈P(A) argmin V L r (V, α, π).

The proof of the theorem is given in Appendix B.3. We emphasize that the theorem holds when V is given enough capacity, i.e., in the nonparametric limit. With parametrization introduced, definitely approximation error will be introduced, and the valid range of η V , which keeps optimal solution unchanged, will be affected. However, the function approximation error is still an open problem for general class of parametrization, we omit such discussion here which is out of the range of this paper.

4.3. STOCHASTIC DUAL ASCENT UPDATE:

Rather than the primal form, i.e., min V max α∈P(S),π∈P(A) L r (V, α, π), we focus on optimizing the dual form max α∈P(S),π∈P(A) min V L r (V, α, π). The major reason is due to the sample efficiency consideration. In the primal form, to apply the stochastic gradient descent algorithm at V t , one needs to solve max α∈P(S),π∈P(A) L r (V t , α, π) which involves sampling from each π and α during the solution path for the subproblem. We define the regularized dual function r (α, π) := min V L r (V, α, π). We first show the unbiased gradient estimator of r w.r.t. θ ρ = (θ α , θ π ), which are parameters associated with α and π. Then, we incorporate the stochastic update rule to the dual ascent algorithm (Boyd et al., 2010) , resulting in the dual actor-critic (Dual-AC) algorithm.

The gradient estimators of the dual functions can be derived using chain rule and are provided below.

Theorem 5 The regularized dual function r (α, π) has gradients estimators

EQUATION

EQUATION

Therefore, we can apply stochastic mirror descent algorithm with the gradient estimator given in Theorem 5 to the regularized dual function r (α, π). Since the dual variables are probabilistic distributions, it is natural to use KL-divergence as the prox-mapping to characterize the geometry in the family of parameters (Amari & Nagaoka, 1993; Nemirovski et al., 2009) . Specifically, in the t-th iteration,

EQUATION

where ĝt-1

ρ = ∇ θρ r θ t-1 α , θ t-1 π

denotes the stochastic gradients estimated through (10) and (11) via given samples and KL(q(s, a)||p(s, a)) = q(s, a) log q(s,a) p(s,a) dsda. Intuitively, such update rule emphasizes a trade-off between the current policy and possible improvements based on samples. The update of π shares some similarity to the TRPO, which is derived to ensure monotonic improvement of the new policy Schulman et al. (2015a) . We discuss the details in Section 4.4.

Rather than just update V once via the stochastic gradient of ∇ V L r (V, α, π) in each iteration for solving saddle-point problem (Nemirovski et al., 2009) , which is only valid in convexconcave setting, Dual-AC exploits the stochastic dual ascent algorithm which requires

V t = Algorithm 1 Dual Actor-Critic (Dual-AC) 1: Initialize θ 0 V , θ 0 α and θ 0 π randomly, set β ∈ [ 1 2 , 1]. 2: for episode t = 1, . . . , T do 3: Start from s ∼ α t-1 (s), collect samples {τ l } m l=1 follows behavior policy π t-1 . 4: Update θ t V = argmin θ V Lr (V, α t-1 , π t-1

) by SGD based on {τ l } m l=1 .

5:

Update αt (s) according to closed-form ( 14).

6:

Decay the stepsize:

ζ t = C n0+1/t β . 7:

Compute the stochastic gradients for θ π following (11).

8:

Update θ t π according to the exact prox-mapping ( 16) or the approximate closed-form (17). 9: end for argmin V L r (V, α t-1 , π t-1 ) in t-th iteration for estimating ∇ θρ r (θ α , θ π ). As we discussed, such operation will keep the gradient estimator of dual variables unbiased, which provides better direction for convergence.

In Algorithm 1, we update V t by solving optimization min V L r (V, α t-1 , π t-1 ). In fact, the V function in the path-regularized Lagrangian L r (V, α, π) plays two roles: i), inherited from the original Lagrangian, the first two terms in regularized Lagrangian (9) push the V towards the value function of the optimal policy with on-policy samples; ii), on the other hand, the path regularization enforces V to be close to the value function of behavior policy π b with off-policy samples. Therefore, the V function in the Dual-AC algorithm can be understood as an interpolation between these two value functions learned from both on and off policy samples.

4.4. PRACTICAL IMPLEMENTATION:

In above, we have introduced path regularization for recovering local duality property of the parametrized multi-step Lagrangian dual form and tailored stochastic mirror descent algorithm for optimizing the regularized dual function. Here, we present several strategies for practical computation considerations.

Update rule of V t . In each iteration, we need to solve V t = argmin θ V L r (V, α t-1 , π t-1 ), which depends on π b and η V , for estimating the gradient for dual variables. In fact, the closer π b to π * is, the smaller

E s∼µ(s) E π b ∞ i=0 γ i R(s i , a i ) -V * (s)

2 will be. Therefore, we can set η V to be large for better local convexity and faster convergence. Intuitively, the π t-1 is approaching to π * as the algorithm iterates. Therefore, we can exploit the policy obtained in previous iteration, i.e., π t-1 , as the behavior policy. The experience replay can also be used.

Furthermore, notice the L(V, α t-1 , π t-1 ) is a expectation of functions of V , we will use stochastic gradient descent algorithm for the subproblem. Other efficient optimization algorithms can be used too. Specifically, the unbiased gradient estimator for

∇ θ V L(V, α t-1 , π t-1 ) is ∇ θ V L r (V, α t-1 , π t-1 ) = (1 -γ k+1 )E µ [∇ θ V V (s)] + E π α ∇ θ V δ {s i , a i } k i=0 , s k+1 (13) -2η V E π b µ ∞ i=0 γ i R(s i , a i ) -V (s) ∇ θ V V (s) . We can use k-step Monte Carlo approximation for E π b µ ∞ i=0 γ i R(s i , a i ) in the gradient estimator.

As k is large enough, the truncate error is negligible (Sutton & Barto, 1998) . We will iterate via

θ t,i V = θ t,i-1 V + κ i ∇ θ t,i-1 V L r (V, α t-1 , π t-1 ) until the algorithm converges.

It should be emphasized that in our algorithm, V t is not trying to approximate the value or advantage function of π t , in contrast to most actor-critic algorithms. Although V t eventually becomes an approximation of the optimal value function once the solution reaches the global optimum, in each update V t is merely a function that helps the current policy to be improved. From this perspective, the Dual-AC bypasses the policy evaluation step.

Update rule of α t . In practice, we may face with the situation that the initial sampling distribution is fixed, e.g., in MuJoCo tasks. Therefore, we cannot obtain samples from α t (s) at each iteration. We assume that ∃η µ ∈ (0, 1], such that α(s) = (1 -η µ )β(s) + η µ µ(s) with β(s) ∈ P(S). Hence, we have

E π α δ {s i , a i } k i=0 , s k+1 = E π µ (α(s) + η µ ) δ {s i , a i } k i=0 , s k+1 , where α(s) = (1 -η µ ) β(s) µ(s)

. Note that such an assumption is much weaker comparing with the requirement for popular policy gradient algorithms (e.g., Sutton et al. (1999) ; Silver et al. (2014) ) that assumes µ(s) to be a stationary distribution. In fact, we can obtain a closed-form update for α if a square-norm regularization term is introduced into the dual function. Specifically, Theorem 6 In t-th iteration, given V t and π t-1 ,

EQUATION

)

Then, we can update αt through ( 14) with Monte Carlo approximation of

E π t-1 δ {s i , a i } k i=0

, s k+1 , avoiding the parametrization of α. As we can see, the αt (s) reweights the samples based on the temporal differences and this offers a principled justification for the heuristic prioritized reweighting trick used in (Schaul et al., 2015) .

Update rule of θ t π . The parameters for dual function, θ ρ , are updated by the prox-mapping operator (12) following the stochastic mirror descent algorithm for the regularized dual function. Specifically, in t-th iteration, given V t and α t , for θ π , the prox-mapping (12) reduces to

EQUATION

where ĝt π = ∇ θπ r (θ t α , θ t π ). Then, the update rule will become exactly the natural policy gradient (Kakade, 2002) with a principled way to compute the "policy gradient" ĝt π . This can be understood as the penalty version of the trust region policy optimization (Schulman et al., 2015a) , in which the policy parameters conservative update in terms of KL-divergence is achieved by adding explicit constraints.

Exactly solving the prox-mapping for θ π requires another optimization, which may be expensive. To further accelerate the prox-mapping, we approximate the KL-divergence with the second-order Taylor expansion, and obtain an approximate closed-form update given by

EQUATION

where F t := E α t π t-1 ∇ 2 log π θ t-1 π denotes the Fisher information matrix. Empirically, we may normalize the gradient by its norm g t π F -1 t g t π (Rajeswaran et al., 2017) for better performances.

Combining these practical tricks to the stochastic mirror descent update eventually gives rise to the dual actor-critic algorithm outlined in Algorithm 1.

5. RELATED WORK:

The dual actor-critic algorithm includes both the learning of optimal value function and optimal policy in a unified framework based on the duality of the linear programming (LP) representation of Bellman optimality equation. The linear programming formulation of Bellman optimality equation and its duality have been used for (approximate) planning problem (Schweitzer & Seidmann, 1985; de Farias & Van Roy, 2003; Wang et al., 2007; O'Donoghue et al., 2011; Cogill, 2015) , in which the transition probability of the MDP is known and the value function or policy are in tabular form. Chen & Wang (2016) ; Wang (2017) apply stochastic first-order algorithms (Nemirovski et al., 2009) for the one-step Lagrangian of the LP problem in reinforcement learning setting. However, as we discussed in Section 3, their algorithm is restricted to tabular parametrization and are not applicable to MDPs with large or continuous state/action spaces. Other authors have also 2017), but they either do not focus on concrete algorithms for solving the optimization problem, or require certain knowledge of the transition probability function that may be hard to obtain in practice.

The duality view has also been exploited in Neu et al. (2017) . Their algorithm is based on the duality of entropy-regularized Bellman equation (Todorov, 2007; Rubin et al., 2012; Fox et al., Published as a conference paper at ICLR 2018 2016; Haarnoja et al., 2017; Asadi & Littman, 2017; Nachum et al., 2017) , rather than the exact Bellman optimality equation we try to solve in this work.

Our dual actor-critic algorithm can be understood as a nontrivial extension of the (approximate) dual gradient method (Bertsekas, 1999, Chapter 6 .3) using stochastic gradient and Bregman divergence, which essentially parallels the view of (approximate) stochastic mirror descent algorithm (Nemirovski et al., 2009) in the primal space. As a result, the algorithm converges with diminishing stepsizes and decaying errors from solving subproblems.

Particularly, the update rules of α and π in the dual actor-critic are related to several existing algorithms. As we see in the update of α, the algorithm reweighs the samples which are not fitted well. This is related to the heuristic prioritized experience replay (Schaul et al., 2015) . For the update in π, the proposed algorithm bears some similarities with trust region poicy gradient (TRPO) (Schulman et al., 2015a) and natural policy gradient (Kakade, 2002; Rajeswaran et al., 2017) . Indeed, TRPO and NPR solve the same prox-mapping but are derived from different perspectives. We emphasize that although the updating rules share some resemblance to several reinforcement learning algorithms in the literature, they are purely originated from a stochastic dual ascent algorithm for solving the two-play game derived from Bellman optimality equation.

6. EXPERIMENTS:

We evaluated the dual actor-critic (Dual-AC) algorithm on several continuous control environments from the OpenAI Gym (Brockman et al., 2016) with MuJoCo physics simulator (Todorov et al., 2012) . We compared Dual-AC with several representative actor-critic algorithms, including trust region policy optimization (TRPO) (Schulman et al., 2015a) and proximal policy optimization (PPO) (Schulman et al., 2017) 1 . We ran the algorithms with 5 random seeds and reported the average rewards with 50% confidence interval. Details of the tasks and setups of these experiments including the policy/value function architectures and the hyperparameters values, are provided in Appendix C.

6.1. ABLATION STUDY:

To justify our analysis in identifying the sources of instability in directly optimizing the parametrized one-step Lagrangian duality and the effect of the corresponding components in the dual actor-critic algorithm, we perform comprehensive Ablation study in InvertedDoublePendulum-v1, Swimmer-v1, and Hopper-v1 environments. We also considered the effect of k = {10, 50} besides the one-step result in the study to demonstrate the benefits of multi-step.

We step can be alleviated via multi-step trajectories. The performances of Dual-AC become better with the length of step k increasing on these three tasks. We conjecture that the main reason may be that in these three MuJoCo environments, the bias dominates the variance. Therefore, with the k increasing, the proposed Dual-AC obtains more accumulate rewards.

6.2. COMPARISON IN CONTINUOUS CONTROL TASKS:

In this section, we evaluated the Dual-AC against TRPO and PPO across multiple tasks, including the InvertedDoublePendulum-v1, Hopper-v1, HalfCheetah-v1, Swimmer-v1 and Walker-v1. These tasks have different dynamic properties, ranging from unstable to stable, Therefore, they provide sufficient benchmarks for testing the algorithms. In Figure 2 , we reported the average rewards across 5 runs of each algorithm with 50% confidence interval during the training stage. We also reported the average final rewards in Table 1 . A notable case is the InvertedDoublePendulum, where Dual-AC substantially outperforms TRPO and PPO in terms of the learning speed and sample efficiency, implying that Dual-AC is preferable to unstable dynamics. We conjecture this advantage might come from the different meaning of V in our algorithm. For unstable system, the failure will happen frequently, resulting the collected data are far away from the optimal trajectories. Therefore, the policy improvement through the value function corresponding to current policy is slower, while our algorithm learns the optimal value function and enhances the sample efficiency.

7. CONCLUSION:

In this paper, we revisited the linear program formulation of the Bellman optimality equation, whose Lagrangian dual form yields a game-theoretic view for the roles of the actor and the dual critic. Although such a framework for actor and dual critic allows them to be optimized for the same objective function, parametering the actor and dual critic unfortunately induces instablity in optimization. We analyze the sources of instability, which is corroborated by numerical experiments. We then propose Dual Actor-Critic , which exploits stochastic dual ascent algorithm for the path regularized, multi-step bootstrapping two-player game, to bypass these issues. The algorithm achieves the stateof-the-art performances on several MuJoCo benchmarks.

Specifically, without loss of generality, we consider the solvable MDP, i.e., the optimal policy, π * (a|s), exists.

If R(s, a) ∞ C R , V * ∞ C R 1-γ . Moreover, V * 2 2,µ = (V * (s)) 2 µ(s)ds = R(s, a) + γE s |s,a [V * (s )] 2 π * (a|s)µ(s)d(s, a) 2 (R(s, a)) 2 π * (a|s)µ(s)ds + 2γ 2 E s |s,a [V * (s )] 2 π * (a|s)µ(s)ds 2 max a∈A R(s, a) 2 µ + 2γ 2 P * (s |s)µ(s)ds (V * (s )) 2 ds 2 max a∈A R(s, a) 2 µ + 2γ 2 V * (s ) 2 ∞ P * (s |s)µ(s)dsds 2 max a∈A R(s, a) 2 µ + 2γ 2 V * (s ) 2 ∞ ,

where the first inequality comes from

EQUATION

∞ , for some π b ∈ P that π b (a|s) > 0 for ∀ (s, a) ∈ S × A. Therefore, with the assumption that R(s, a)

2 µ C µ R , ∀a ∈ A, we have R(s, a) ∈ L 2 µπ b (S × A) and V * (s ) ∈ L 2 µ (S).

The constraints in the primal form of linear programming can be written as

(I -γP) V -R L 2 µπ b 0,

where I -γP : L 2 µ (S) → L 2 µπ b (S × A) without any effect on the optimality. For simplicity, we denote as L 2 µπ b and f, g = f (s, a)g(s, a)µ(s)π b (a|s)dsda. Apply the Lagrangian multiplier for constraints in ordered Banach space in Burger (2003) , we have

EQUATION

The solution (V * , * ) also satisfies the KKT conditions,

EQUATION

EQUATION

(I -γP) V * -R 0, (22) * , (I -γP) V * -R = 0.

(23) where denotes the conjugate operation. By the KKT condition, we have

EQUATION

The strongly duality also holds, i.e.,

EQUATION

EQUATION

Proof We compute the duality gap

(1 -γ) 1, V * -R, * = * , (I -γP) V * -R, * = * , (I -γP) V * -R = 0,

which shows the strongly duality holds.

B DETAILS OF THE PROOFS FOR SECTION 4 B.1 COMPETITION IN MULTI-STEP SETTING:

Once we establish the k-step Bellman optimality equation ( 7), it is easy to derive the λ-Bellman optimality equation, i.e.,

EQUATION

Proof Denote the optimal policy as π * (a|s), we have

V * (s) = E π * {st} i=0 |s k i=0 γ i R(s i , a i ) + γ k+1 E π * s k+1 |s [V * (s k+1 )] ,

holds for arbitrary ∀k ∈ N. Then, we conduct k ∼ Geo(λ) and take expectation over the countable infinite many equation, resulting

V * (s) = (1 -λ) ∞ k=0 λ k E π * k i=0 γ i R(s i , a i ) + γ k+1 V * (s k+1 ) = max π∈P (1 -λ) ∞ k=0 λ k E π k i=0 γ i R(s i , a i ) + γ k+1 V * (s k+1 )

Next, we investigate the equivalent optimization form of the k-step and λ-Bellman optimality equation, which requires the following monotonic property of T k and T λ .

Lemma 7 Both T k and T λ are monotonic.

Proof Assume U and V are the value functions corresponding to π 1 and π 2 , and U V , i.e., U (s) V (s), ∀s ∈ S, apply the operator T k on U and V , we have

(T k U ) (s) = max π∈P E π {si} k i=1 |s k i=0 γ i R(s i , a i ) + γ k+1 E π s k+1 |s [U (s k+1 )] , (T k V ) (s) = max π∈P E π {si} k i=1 |s k i=0 γ i R(s i , a i ) + γ k+1 E π s k+1 |s [V (s k+1 )] . Due to U V , we have E π s k+1 |s [U (s k+1 )] E π s k+1 |s [V (s k+1 )], ∀π ∈ P, which leads to the first conclusion, T k U T k V . Since T λ = (1 -λ) ∞ k=1 T k = E k∼Geo(λ) [T k ], therefore, T λ is also monotonic.

With the monotonicity of T k and T λ , we can rewrite the V * as the solution to an optimization, Theorem 8 The optimal value function V * is the solution to the optimization

EQUATION

where µ(s) is an arbitrary distribution over S.

Proof Recall the T k is monotonic, i.e., V

T k V ⇒ T k V T 2 k V and V * = T ∞ k V for arbitrary V , we have for ∀V , V T k V T 2 k V . . . T ∞ k V = V *

, where the last equality comes from the Banach fixed point theorem (Puterman, 2014) . Similarly, we can also show that ∀V , V T ∞ λ V = V * . By combining these two inequalities, we achieve the optimization.

We rewrite the optimization as min

V (1 -γ k+1 )E s∼µ(s) [V (s)] (29) s.t. V (s) R(s, a) + max π∈P E π {si} k+1 i=1 |s k i=1 γ i R(s i , a i ) + γ k+1 V (s k+1 ) ,

(s, a) ∈ S × A, We emphasize that this optimization is no longer linear programming since the existence of maxoperator over distribution space in the constraints. However, Theorem 1 still holds for the dual variables in (32).

Proof Denote the optimal policy as π *

V = argmax π∈P E π {si} k+1 i=1 |s k i=1 γ i R(s i , a i ) + γ k+1 V (s k+1

) , the KKT condition of the optimization (29) can be written as 

1 -γ k+1 µ(s ) + γ k+1 {si,ai} k i=0 p(s |s k , a k ) k-1 i=0 p(s i+1 |s i , a i ) k i=1 π * V (a i |s i )ρ * (s 0 , a 0 ) = a0,{si,ai} k i=1 k i=0 p(s i+1 |s i , a i )ρ * (s , a) k i=1 π * V (a i |s i ). Denote P π k (s k+1 |s, a) = {si,ai} k i=1 p(s k+1 |s k , a k ) k-1 i=0 p(s i+1 |s i , a i ) k i=1 π(a i |s i ),

EQUATION

)

Recall V * denotes the value function of the optimal policy, then, based on the definition, π * V * = π * which denotes the optimal policy. Then, the condition (30) implies ρ(s, a) = 0 if and only if a = a * , therefore, we can decompose ρ * (s, a) = α * (s)π * (a|s).

The corresponding Lagrangian of optimization ( 29) is

EQUATION

where

∆ π k [V ](s, a) = R(s, a) + E π {st} k+1 i=1 |s k i=1 γ i R(s i , a i ) + γ k+1 V (s k+1 ) -V (s).

We further simplify the optimization. Since the dual variables are positive, we have

EQUATION

After clarifying these properties of the optimization corresponding to the multi-step Bellman optimality equation, we are ready to prove the Theorem 3.

Theorem 3 The optimal policy π * and its corresponding value function V * is the solution to the following saddle point problem max

α∈P(S),π∈P(A) min V L k (V, α, π) := (1 -γ k+1 )E µ [V (s)] (8) + {si,ai} k i=0 ,s k+1 α(s 0 ) k i=0 π(a i |s i )p(s i+1 |s i , a i )δ[V ] {s i , a i } k i=0 , s k+1 where δ[V ] {s i , a i } k i=0 , s k+1 = k i=0 γ i R(s i , a i ) + γ k+1 V (s k+1 ) -V (s).

Proof By Theorem 1 in multi-step setting, we can decompose ρ(s, a) = α(s)π(a|s) without any loss. Plugging such decomposition into the Lagrangian 32 and realizing the equivalence among the optimal policies, we arrive the optimization as min V max α∈P(S),π∈P(A) L k (V, α, π). Then, because of the strong duality as we proved in Lemma 9, we can switch min and max operators in optimization 8 without any loss.

Lemma 9 The strong duality holds in optimization (8).

Proof Specifically, for every α ∈ P(S), π ∈ P(A),

(α, π) = min V L k (V, α, π) min V L k (V, α, π); δ[V ] {s i , a i } k i=0 , s k+1 0 min V (1 -γ k+1 )E s∼µ(s) [V (s)] , s.t. δ[V ] {s i , a i } k i=0 , s k+1 0 = (1 -γ k+1 )E s∼µ(s) [V * (s)] .

On the other hand, since

L k (V, α * , π * ) is convex w.r.t. V , we have V * ∈ argmin V L k (V, α * , π * ),

by checking the first-order optimality. Therefore, we have max

α∈P(S),π∈P(A) (α, π) = max α∈P(S),π∈P(A),V ∈argmin V L k (V,α,π) L k (V, α, π) L(V * , α * , π * ) = (1 -γ k+1 )E s∼µ(s) [V * (s)]

. Combine these two conditions, we achieve the strong duality even without convex-concave property

(1 -γ k+1 )E s∼µ(s) [V * (s)] max α∈P(S),π∈P(A) (α, π) (1 -γ k+1 )E s∼µ(s) [V * (s)] .

B.2 THE COMPOSITION IN APPLYING AUGMENTED LAGRANGIAN METHOD:

We consider the one-step Lagrangian duality first. Following the vanilla augmented Lagrangian method, one can achieve the dual function as

(α, π) = min V (1 -γ) E s∼µ(s) [V (s)] + (s,a)∈S×A P c (∆[V ](s, a), α(s)π(a|s)) ,

where

P c (∆[V ](s, a), α(s)π(a|s)) = 1 2c [max (0, α(s)π(a|s) + c∆[V ](s, a))] 2 -α 2 (s)π 2 (a|s) .

The computation of P c is in general intractable due to the composition of max and the condition expectation in ∆[V ](s, a), which makes the optimization for augmented Lagrangian method difficult.

For the multi-step Lagrangian duality, the objective will become even more difficult due to constraints are on distribution family P(S) and P(A), rather than S × A.

B.3 PATH REGULARIZATION:



Theorem 4:

The local duality holds for L r (V, α, π). Denote (V * , α * , π * ) as the solution to Bellman optimality equation, with some appropriate η V , (V * , α * , π * ) = argmax α∈P(S),π∈P(A) argmin V L r (V, α, π).

Proof The local duality can be verified by checking the Hessian of L r (θ V * ). We apply the local duality theorem (Luenberger & Ye, 2015) [Chapter 14]. Suppose ( Ṽ * , α * , π * ) is a local solution to min V max α∈P(S),π∈P(A) L r (V, α, π), then, max α∈P(S),π∈P(A) min V L r (V, α, π) has a local solution Ṽ * with corresponding α * , π * .

Next, we show that with some appropriate η V , the path regularization does not change the optimum. Let U π (s) = E π ∞ i=0 γ i R(s i , a i )|s , and thus, U π * = V * . We first show that for ∀π b ∈ P(A), we have

E E π b ∞ i=0 γ i R(s i , a i ) -V * (s) 2 = E U π b (s) -U π * (s) + U π * (s) -V * (s) 2 = E U π b (s) -U π * (s) 2 E   ∞ i=0 π b (a i |s i ) - ∞ i=0 π * (a i |s i ) ∞ i=0 p(s i+1 |s i , a i ) ∞ i=1 γ i R(s i , a i ) d {s i , a i } ∞ i=0 2   E   ∞ i=1 γ i R(s i , a i ) 2 ∞ ∞ i=0 π b (a i |s i ) - ∞ i=0 π * (a i |s i ) ∞ i=0 p(s i+1 |s i , a i ) 2 1   4 ∞ i=1 γ i R(s i , a i ) 2 ∞ 4 (1-γ) 2 R(s, a) 2 ∞

where the last second inequality comes from the fact that π b (a i |s i )p(s i+1 |s i , a i ) is distribution.

We then rewrite the optimization min V max α∈P(S),π∈P(A) L r (V, α, π) as min V max α∈P(S),π∈P(A)

L k (V, α, π)

s.t. V ∈ Ω ,π b := V : E s∼µ(s) E π b ∞ i=0 γ i R(s i , a i ) -V (s) 2 ,

due to the well-known one-to-one correspondence between regularization η V and Nesterov (2005) .

If we set η V with appropriate value so that its corresponding (η V ) 2 1-γ R(s, a) ∞ , we will have V * ∈ Ω (η V ) , which means adding such constraint, or equivalently, adding the path regularization, does not affect the optimality. Combine with the local duality, we achieve the conclusion.

In fact, based on the proof, the closer π b to π * is, the smaller Proof We mainly focus on deriving ∇ θπ r (θ α , θ π ). The derivation of ∇ θα r (θ α , θ π ) is similar.

E s∼µ(s) E π b ∞ i=0 γ i R(s i , a i ) -V * (

By chain rule, we have The first term in RHS equals to zero due to the first-order optimality condition for V (α, π) = argmin V L r (V, α, π).

∇ θπ r (θ α , θ π ) = ∇ V L k (V (α, θ), α, θ) -2η V E π b ∞ i=0 γ i R(s i , a i ) -V * (s) 0 ∇ θπ V (α,

B.5 PRACTICAL ALGORITHM:

Theorem 6 In t-th iteration, given V t and π Therefore, in t-th iteration, αt (s) = 1 ηα max 0, E π δ {s i , a i } k i=0 , s k+1 .

C EXPERIMENT DETAILS:

Policy and value function parametrization. For fairness, we use the same parametrization across all the algorithms. The parametrization of policy and value functions are largely based on the recent paper by Rajeswaran et al. (2017) , which shows the natural policy gradient with the RBF neural network achieves the state-of-the-art performances of TRPO on MuJoCo. For the policy distribution, we parametrize it as π θπ (a|s) = N (µ θπ (s), Σ θπ ), where µ θπ (s) is a two-layer neural nets with the random features of RBF kernel as the hidden layer and the Σ θπ is a diagonal matrix. The RBF kernel bandwidth is chosen via median trick (Dai et al., 2014; Rajeswaran et al., 2017) . The same as Rajeswaran et al. (2017) , we use 100 hidden nodes in Pendulum, InvertedDoublePendulum, Swimmer, Hopper, and use 500 hidden nodes in HalfCheetah and Walker. Since the TRPO and PPO uses GAE (Schulman et al., 2015b) with linear baseline as V , we also use the parametrization for V in our algorithm. However, the Dual-AC can adopt arbitrary function approximator without any change.

Training details. We report the hyperparameters for each algorithms here. We use the γ = 0.995 for all the algorithms. We keep constant stepsize and tuned for TRPO, PPO and Dual-AC in {0.001, 0.01, 0.1}. The batchsize are set to be 52 trajectories for comparison to the competitors in Section 6.2. For the Ablation study, we set batchsize to be 24 trajectories for faster runtime. The CG damping parameter for TRPO is set to be 10 -4 . We iterate 20 steps for the Fisher information matrix computation. For the η V , η µ , 1 ηα in Dual-AC from {0.001, 0.01, 0.1, 1}.

:

As discussed inHenderson et al. (2018), different implementations of TRPO and PPO can provide different performances. For a fair comparison, we use the codes from https://github.com/joschu/modular rl reported to have achieved the best scores inHenderson et al. (2018).

Figures and Tables:

considered approximate LP formulations to solve large-scale problems Pazis & Parr (2011); Abbasi-Yadkori et al. (2014); Lakshminarayanan et al. (

Figure 1: Comparison between the Dual-AC and its variants for justifying the analysis of the source of instability.

Figure2: The results of Dual-AC against TRPO and PPO baselines. Each plot shows average reward during training across 5 random seeded runs, with 50% confidence interval. The x-axis is the number of training iterations. The achieves comparable performances comparing with TRPO and PPO in some tasks, but outperforms on more challenging tasks.

we simplify the condition, i.e., 1 -γ k+1 µ(s ) + γ k+1 s|s, a)ρ * (s, a) = a ρ * (s , a). Due to the P π * V k (s |s, a) is a conditional probability for ∀V , with similar argument in Theorem 1, we have s,a ρ * (s, a) = 1. By the KKT complementary condition, the primal and dual solutions, i.e., V * and ρ * , satisfy ρ * (s, a) R(s, a) + E

s) 2 will be. Therefore, we can set η V bigger for better local convexity, which resulting faster convergence.B.4 STOCHASTIC DUAL ASCENT UPDATECorollary 5 The regularized dual function r (α, π) has gradients estimators∇ θα r (θ α , θ π ) = E π α δ {s i , a i } k i=0 , s k+1 ∇ θα log α(s) , ∇ θπ r (θ α , θ π ) = E π α δ {s i , a i } k i=0 , s k+1 k i=0∇ θπ log π(a|s) .

θ) +E π α δ {s i , a i } k i=0 , s k+1 k i=0 ∇ θπ log π(a|s) = E π α δ {s i , a i } k i=0 , s k+1 k i=0∇ θπ log π(a|s) .

(s) τ (s) + E π δ {s i , a i } δ {s i , a i } k i=0 , s k+1 E π δ {s i , a i } k i=0 , s k+1 < 0 0 E π δ {s i , a i } k i=0 , s k+10.

The average final performances of the policies learned from Dual-AC and the competitors.

Table content: <table><tr><td>Environment</td><td>Dual-AC</td><td>PPO</td><td>TRPO</td></tr><tr><td>Pendulum</td><td colspan="3">-155.45 -266.98 -245.11</td></tr><tr><td colspan="4">InvertedDoublePendulum 8599.47 1776.26 3070.96</td></tr><tr><td>Swimmer</td><td>234.56</td><td>223.13</td><td>232.89</td></tr><tr><td>Hopper</td><td colspan="3">2983.79 2376.15 2483.57</td></tr><tr><td>HalfCheetah</td><td colspan="3">3041.47 2249.10 2347.19</td></tr><tr><td>Walker</td><td colspan="3">4103.60 3315.45 2838.99</td></tr><tr><td colspan="4">The proposed Dual-AC achieves the best performance in almost all environments, including Pen-</td></tr><tr><td colspan="4">dulum, InvertedDoublePendulum, Hopper, HalfCheetah and Walker. These results demonstrate that</td></tr><tr><td colspan="4">Dual-AC is a viable and competitive RL algorithm for a wide spectrum of RL tasks with different</td></tr><tr><td>dynamic properties.</td><td/><td/><td/></tr></table>

t-1 , E µ(s)π t-1 (s) (α(s) + η µ ) δ {s i , a i } Proof Recall the optimization w.r.t. α is max α 0 E µ α(s)E π δ {s i , a i } E π δ {s i , a i }

Table content: <table><tr><td/><td/><td colspan="2">argmax α 0</td><td/><td/><td>k i=0 , s k+1 -η α α</td><td>2 µ</td></tr><tr><td/><td>=</td><td>1 η α</td><td>max 0, E π t-1</td><td>δ {s i , a i }</td><td>k i=0 , s k+1</td><td>.</td></tr><tr><td/><td/><td/><td/><td/><td/><td>k i=0 , s k+1 -η α</td><td>α2 (s) ,</td></tr><tr><td colspan="7">denote τ (s) as the dual variables of the optimization, we have the KKT condition as</td></tr><tr><td/><td/><td/><td colspan="4">  η α α    τ (s)α(s) = 0, = τ + k i=0 , s k+1 ,</td></tr><tr><td/><td/><td/><td>    τ α</td><td>0, 0,</td><td/></tr><tr><td>⇒</td><td>     </td><td>α</td><td/><td/><td/><td>=</td><td>τ +E π [δ({si,ai} k i=0 ,s k+1)] ηα</td></tr><tr><td/><td>    </td><td/><td/><td/><td/></tr></table>

Peer review (iclr2018):
 This paper studies a new architecture DualAC. The author give strong and convincing justifications based on the Lagrangian dual of the Bellman equation (although not new, introducing this as the justification for the architecture design is plausible). There are several drawbacks of the current format of the paper: 1. The algorithm is vague. Alg 1 line 5: *closed form*: there is no closed form in Eq(14). It is just an MC approximation. line 6: Decay O(1/t^.beta). This is indeed vague albeit easy to understand. The algorithm requires that every step is crystal clear. 2. Also, there are several format error which may be due to compiling, e.g., line 2 of Abstract,*Dual-AC * (an extra space). There are many format errors like this throughout the paper. The author is suggested to do a careful format check. 3. The author is suggested to explain more about the necessity of introducing path regularization and SDA. The current justification is reasonable but too brief. 4. The experimental part is ok to me, but not very impressive. Overall, this seems to be a nice paper to me.