|
# FINITE-TIME CONVERGENCE AND SAMPLE COM## PLEXITY OF MULTI-AGENT ACTOR-CRITIC REIN- FORCEMENT LEARNING WITH AVERAGE REWARD |
|
|
|
|
|
Hairi and Jia Liu |
|
Department of Electrical and Computer Engineering |
|
The Ohio State University |
|
Columbus, OH 43210, USA |
|
hairi.1@osu.edu, liu@ece.osu.edu |
|
|
|
ABSTRACT |
|
|
|
|
|
Songtao Lu |
|
IBM Research AI |
|
IBM Thomas J. Watson Research Center |
|
Yorktown Heights, NY 10598 |
|
songtao@ibm.com |
|
|
|
|
|
In this paper, we establish the first finite-time convergence result of the actor-critic |
|
algorithm for fully decentralized multi-agent reinforcement learning (MARL) |
|
problems with average reward. In this problem, a set of N agents work cooperatively to maximize the global average reward through interacting with their |
|
neighbors over a communication network. We consider a practical MARL setting, where the rewards and actions of each agent are only known to itself, and the |
|
knowledge of joint actions of the agents is not assumed. Toward this end, we propose a mini-batch Markovian sampled fully decentralized actor-critic algorithm |
|
and analyze its finite-time convergence and sample complexity. We show that the |
|
sample complexity of this algorithm is O(N [2]/ǫ[2] log(N/ǫ)). Interestingly, this |
|
sample complexity bound matches that of the state-of-the-art single-agent actorcritic algorithms for reinforcement learning. |
|
|
|
1 INTRODUCTION |
|
|
|
1) Background and Motivations: In recent years, multi-agent reinforcement learning (MARL) |
|
has found a wide range of applications in networked large-scale systems, such as power grid systems (Riedmiller et al., 2000), autonomous driving (Yu et al., 2019; Shalev-Shwartz et al., 2016) |
|
and strategic games (Silver et al., 2018; Foerster et al., 2018), to name just a few. Although empirical successes of MARL applications have been widely observed, the fundamental theoretical |
|
understanding of how to develop fast-converging and low sample-complexity MARL algorithms, |
|
two of the most important performance metrics for MARL, remains in its infancy so far (see, e.g., |
|
(Zhang et al., 2021a) for an excellent survey). In particular, two important aspects of cooperative |
|
MARL algorithm designs deserve special attention: |
|
|
|
- First, in the multi-agent collaborative setting, the information structure (i.e., the assumptions of |
|
who have the knowledge of what) is far more complex than its single-agent counterpart and care |
|
must be taken in MARL problem formulations. In the cooperative MARL literature so far, many |
|
existing works assume full knowledge of joint states and joint actions, which often do not hold |
|
true in practice. For example, in autonomous driving (Yu et al., 2019), each vehicle can only |
|
observe/detect the actions of the surrounding vehicles that are within its communication range. |
|
As another example, in power grid networks (Riedmiller et al., 2000), each power distributor |
|
generally does not know the resistor values set by other distributors. |
|
|
|
- Second, most MARL theoretical studies in the MARL literature are focused on the discounted |
|
total reward setting, where a hyperparameter γ ∈ (0, 1) is introduced as the discount factor in the |
|
objective function. Although the discounted total reward setting captures the important aspect of |
|
diminishing return in the future, it may not be appropriate for many other applications where the |
|
long-term average reward is of interest. For example, in the optimization of distributed communication networks with MARL, the typical and natural performance metrics are long-term average |
|
throughput or latency in the steady-state. |
|
The lack of a fundamental understanding on how to develop efficient cooperative MARL algorithms |
|
that consider the above two aspects in terms of information structure, scalability, and communication |
|
|
|
|
|
----- |
|
|
|
and sample complexities motivates us to fill this gap by developing a fully decentralized cooperative |
|
MARL algorithm in the average reward setting, without assuming joint action knowledge. |
|
|
|
2) Technical Challenges: Developing a fully decentralized cooperative MARL algorithm for the |
|
average reward setting without full joint action knowledge is highly non-trivial and several major |
|
technical challenges naturally arise. First, it is well-known that, even in the single-agent reinforcement learning (RL) setting, the average reward setting is more challenging to analyze compared to |
|
the discounted reward setting, which necessitates different proof techniques (Tsitsiklis & Van Roy, |
|
1999). In MARL, the decentralized nature and the lack of joint action-state information further |
|
complicate the algorithm design and analysis in the average reward setting. Second, due to the lack |
|
of joint action knowledge, the communication costs among agents in MARL will be significantly |
|
increased to achieve a satisfying performance, which implies that low sample and communication |
|
complexities are even more challenging and critical for MARL without joint action information. |
|
|
|
3) Main Results and Contributions: The main contribution of this paper is that we overcome the |
|
aforementioned challenges and develop a consensus-based Markovian sampled decentralized actorcritic algorithm for MARL. Our key results in this paper are summarized as follows: |
|
|
|
- We propose a batch-sampled actor-critic algorithm that uses consensus updates in TD-sharing |
|
among agents. This batch sampling approach enables more efficient communication compared |
|
to the classical fully decentralized MARL. Specifically, in order to converge to an ǫ neighborhood of the stationary, we require O(ǫ[−][1] log(ǫ[−][1])) rounds of communication while only needing |
|
O(ǫ[−][2] log(ǫ[−][1])) samples. By contrast, the state-of-the-art MARL requires a communication |
|
round per sampling. Also, our algorithm allows the use of constant step-sizes in both the actor |
|
and the critic steps. |
|
|
|
- We provide the first-ever sample complexity analysis in the MARL average reward problem setting without joint action information. Our obtained complexity is O(ǫ[−][2] log(ǫ[−][1])), where ǫ is the |
|
closeness to the neighborhood of stationary point (treating network size N as a fixed constant). |
|
It is worth noting that the order-wise sample complexity of our algorithm matches that of the |
|
state-of-the-art single-agent RL algorithms. |
|
|
|
2 RELATED WORK |
|
|
|
In this section, we provide a quick overview on the closely related work on MARL algorithms and |
|
their theoretical results, along with several notable related counterparts in single-agent RL. |
|
|
|
1) MARL Theoretical Analysis and Algorithm Design: For recent advances in MARL algorithms |
|
and their theoretical results, Zhang et al. (2021a) provided a comprehensive survey. Also, Lee et al. |
|
(2020) highlighted the evolution from single-agent to multi-agent RL from a distributed optimization |
|
perspective. In the broader area of MARL, a line of research has been focused on the MARL policy |
|
evaluation problem. These works analyzed the convergence Doan et al. (2019) and proposed various variance reduction of policy evaluation in decentralized MARL algorithms Zhang et al. (2021b). |
|
Doan et al. (2019) used i.i.d. sampling and has shown the sample complexity of O[˜](ǫ[−][1]) for their |
|
TD(0) learning algorithm to reach a mean-square error convergence. However, these algorithms do |
|
not involve policy improvement and solely focus on the performance evaluation of given policies. In |
|
the areas of joint policy evaluation and improvement, Foerster et al. (2018) considered multi-agent |
|
actor-critic algorithm that has a centralized critic and decentralized actors, which is different from |
|
our fully decentralized actor-critic algorithm. In contrast, Zhang et al. (2018) established asymptotic convergence results for fully decentralized MARL actor-critic algorithms. Concurrent with |
|
our work, Chen et al. (2021) has recently studied the mini-batch Markovian sampling actor-critic |
|
algorithm for a class of discounted reward MARL problems, where the finite-time convergence result is obtained. They have applied batch sampling for both actor and critic steps and achieved a |
|
sample complexity of O(ǫ[−][2] log(ǫ[−][1])), which is the same as ours. We note that, together with our |
|
work, these are the first finite-time convergence results for MARL. However, there are several key |
|
differences between our work and (Chen et al., 2021). First, we focus on the average reward problem, while Chen et al. (2021) studied the discounted reward setting. Second, in (Chen et al., 2021), |
|
agents share a noisy version of the rewards with the neighbors, which requires a re-sampling process |
|
from every sampled reward instance. In contrast, we allow agents to share local TD-errors with their |
|
neighbors and no re-sampling is required. |
|
|
|
|
|
----- |
|
|
|
Table 1: Comparison of sample complexity of single-agent (SA) and multi-agent (MA) AC algorithms and TD(0) algorithms at Average Reward (AR) and Discounted Reward (DR) settings. |
|
|
|
|Paper|Problem|Sampling|Col4|Sample Complexity| |
|
|---|---|---|---|---| |
|
|||actor step|critic step|| |
|
|Qiu et al. (2021)|SAAR|i.i.d.|Markovian|O(ǫ−3 log2(ǫ−1))| |
|
|Xu et al. (2020)|SADR|Markovian|Markovian|O(ǫ−2 log(ǫ−1))| |
|
|Zhang et al. (2018)|MAAR|Markovian|Markovian|Asymptotic| |
|
|Doan et al. (2019)|MADR|N/A|i.i.d|O˜(ǫ−1)| |
|
|Chen et al. (2021)|MADR|Markovian|Markovian|O(ǫ−2 log(ǫ−1))| |
|
|This paper|MAAR|Markovian|Markovian|O(N 2/ǫ2 log(N/ǫ))| |
|
|
|
|
|
|
|
2) Related Literature in Single-Agent RL: We note that single-agent RL can be viewed as a |
|
centralized approach, where a central controller collects joint actions, rewards and even designs |
|
policies for agents. For the single-agent average reward setting, Tsitsiklis & Van Roy (1999) first |
|
analyzed the asymptotic convergence of TD(λ) algorithm with function approximations in the policy |
|
evaluation problem. Also, Tsitsiklis & Van Roy (2002) provided insights in terms of differences and |
|
connections between average reward and discounted reward of TD-based learning algorithms with |
|
function approximations. Recently, Qiu et al. (2021) analyzed the sample complexity for an actorcritic algorithm for the average reward problem. In their actor-critic algorithm, they used batch |
|
sampling for the critic learning and i.i.d. sampling for the actor step, with sample complexity being |
|
O(ǫ[−][3] log[2](ǫ[−][1])). By applying mini-batch sampling update, we are able to improve the sample |
|
complexity by a factor of O(ǫ[−][1]). Another closely related work on single-agent RL is (Xu et al., |
|
2020), where the authors studied the discounted reward problem. They used batch sampling for both |
|
actor and critic steps in their actor-critic algorithm and developed a new technique to handle bias |
|
error in the critic step, which we also adopted for the average approximation parameter analysis in |
|
our critic step. This achieved the state-of-the-art sample complexity of O(ǫ[−][2] log(ǫ[−][1])) for singleagent RL. In addition, the global convergence of actor-critic algorithm to the optimal policy has |
|
been studied in the case of discounted setting with single time scale in (Fu et al., 2020) and linear |
|
quadratic regulator in (Yang et al., 2019). However, we note that these settings are fundamentally |
|
different from the average reward setting and it will be an interesting future direction to consider |
|
global convergence possibility in the average reward setting. |
|
|
|
To conclude this section, we summarize the aforementioned related actor-critic and TD algorithms |
|
and their sample complexity results in Table 1. |
|
|
|
3 MULTI-AGENT REINFORCEMENT LEARNING WITH AVERAGE REWARD |
|
|
|
3.1 SYSTEM MODEL |
|
|
|
Consider a multi-agent system with N agents, denoted by N = {1, · · ·, N }, operating in a networked environment. Let E be the edge set for a given network G = (N, E). To formulate our |
|
MARL problem and facilitate our subsequent discussions, we first define the notion of networked |
|
multi-agent MDP as follows. |
|
Definition 1 (Networked Multi-Agent MDP). Let G = (N, E) be a communication network that connects N agents. A networked multi-agent MDP is defined by following tuple |
|
(S, {A[i]}i∈N, P, {R[i]}i∈N, G), where S is the global state space observed by all agents, A[i] is the |
|
action set for agent i, and P : S ×A×S → [0, 1] is a global state transition function, and R[i] : S ×A |
|
is the local reward function for agent i. Let A = i∈N [A][i][ be the joint action set of all agents.] |
|
|
|
In this paper, we assume that the global state space S and action space for agent A[i] are finite. As |
|
|
|
[Q] |
|
|
|
a result, the joint action space A is also finite for finite N . We also note that at time t ≥ 0, all |
|
agents can observe the current global state st. However, agent i can only observe its own action |
|
a[i]t [∈A][i][, which is the key difference between our model and that in (Zhang et al., 2018), where it is] |
|
assumed that the joint actions are observable to all agents. Moreover, each agent can only observe |
|
its own reward rt[i][, i.e., agents do not observe or share rewards with other agents at time][ t][. The] |
|
|
|
|
|
----- |
|
|
|
reward function R[i](s, a) is an expectation given s and a, and the instantaneous reward is denoted |
|
by r[i](s, a), i.e., R[i](s, a) = E[r[i](s, a)]. |
|
|
|
We consider policies that are stationary. In our MARL system, each agent chooses its action following its local policy π[i] that is conditioned on the current global state s, i.e., π[i](a[i]|s) is the probability |
|
for agent i to choose an action a[i] ∈A[i]. Then, the joint policy π : S × A → [0, 1] can be written as |
|
π(a|s) = i∈N [π][i][(][a][i][|][s][)][.] |
|
|
|
Moreover, the policies at the agents are parameterized. Specifically, each agent i’s local policy can |
|
be written as[Q] πθ[i] [i][, where][ θ][i][ ∈] [R][m][i][ denotes the parameter. We let][ θ][ ≜] [[(][θ][1][)][T][,][ · · ·][,][ (][θ][N] [)][T][ ]][T][ ∈] |
|
N |
|
RPi=1 [m][i] . Then, we can write the joint policy as follows: πθ(a|s) = i∈N [π][θ][i] [(][a][i][|][s][)][.] |
|
|
|
3.2 TECHNICAL ASSUMPTIONS [Q] |
|
|
|
We now state the following assumptions on the positivity and continuity of πθ[i] [i] [(][a][i][|][s][)][, which guar-] |
|
antee the stationary distribution of {st} under any given policy. |
|
Assumption 1. For any i ∈N, s ∈S, a[i] ∈A[i] and θ[i] ∈ R[m][i], the policy function πθ[i] [i][(][a][i][|][s][)][ ≥] [0][.] |
|
Also, πθ[i] [i][(][s, a][)][ is a continuously differentiable with respect to the parameter][ θ][i][. In addition, for] |
|
any θ, we assume the induced Markov chain {st}t≥0 is irreducible and aperiodic, and its transition |
|
matrix P [θ] is P [θ](s[′]|s) = a∈A [π][θ][(][a][|][s][)][ ·][ P] [(][s][′][|][s, a][)][,] ∀s, s[′] ∈S. |
|
|
|
Assumption 1 guarantees that the states have a stationary distribution dθ(s) over S given any policy |
|
|
|
[P] |
|
|
|
πθ. As a result, the Markov chain of state action pair {(st, at)} also has a stationary distribution |
|
dθ(s) · πθ(a|s). |
|
Assumption 2. The instantaneous reward rt[i] [is uniformly bounded by a constant][ r][max] [>][ 0][ for any] |
|
i ∈N and t ≥ 0. |
|
|
|
Assumption 2 is common in the literature (see, e.g., (Zhang et al., 2018; Xu et al., 2020; Doan et al., |
|
2019)) and easy to be satisfied in many practical MDP models with finite state and action spaces. |
|
Assumption 3. Let A be a consensus weight matrix for a given communication network G. There |
|
exists a positive constant η > 0 such that A ∈ R[N] [×][N] is doubly stochastic and Aii ≥ η, ∀i ∈N . |
|
Moreover, Aij ≥ η if i, j are connected, otherwise Aij = 0 for all i, j. |
|
|
|
Assumption 3 is standard in the distributed multi-agent optimization literature Nedic & Ozdaglar |
|
(2009). We remark that for a practical choice of A, one can use the following form A = |
|
1 |
|
|
|
deg(G) [(][deg][(][G][)][ ·][ I][ −] [L][)][,][ where deg][(][G][)][ is the degree of the graph][ G][ (i.e. the maximal vertex de-] |
|
gree), I is the identity matrix of conforming dimensionality, and L is the Laplacian matrix of the |
|
graph. It is easy to verify that this matrix is symmetric, doubly stochastic and η ≥ deg1(G) [≥] N[1] [.] |
|
|
|
Assumption 4. Each agent i’s value function is parameterized by the class of linear functions, i.e., |
|
Vθ(s; w) = φ(s)[T] w where φ(s) ≜ [φ1(s), · · ·, φK (s)][T] ∈ R[K] is the feature associated with the |
|
state s ∈S and K < |S|. The feature vectors φ(s) are uniformly bounded for any s ∈S. Without |
|
loss of generality, we assume that ∥φ(s)∥≤ 1. Furthermore, the feature matrix Φ ∈ R[|S|×][K] has |
|
full column rank. Also, for any u ∈ R[K], Φu ̸= 1, where 1 is an all-one vector. |
|
|
|
This assumption on features is standard and has been widely adopted in the literature, e.g., |
|
(Tsitsiklis & Van Roy, 1999; Zhang et al., 2018; Qiu et al., 2021). This assumption implies the following property: for any policy πθ, the inequality w[T] Aπθ w < 0 holds for any w ̸= 0, where Aπθ is |
|
defined as |
|
|
|
Aπθ := Es∼dθ(s),s′∼P (·|s)[(φ(s[′]) − φ(s))φ[T] (s)]. (1) |
|
|
|
This property further implies that for all θ, Aπθ is invertible and λmax(Aπθ + A[T]πθ [)][ <][ 0][ (Qiu et al.,] |
|
2021), where λmax(·) is the largest eigenvalues of the matrix. |
|
Assumption 5. There exists a constant λA > 0 such that λmax(Aπθ + A[T]πθ [)][ ≤−][λ][A] [holds for all] |
|
θ ∈ RPi∈N [m][i] . |
|
|
|
Assumption 5 ensures the optimal approximation wθ[∗] [for any given policy][ π][θ][ is uniformly bounded] |
|
(see discussion before Theorem 2 and Lemma 4). |
|
|
|
|
|
----- |
|
|
|
Assumption 6. Let ψθ(s, a) = ∇θ log πθ(a|s) be the score function for any state-action pair (s, a). |
|
For any two policy parameters θ, θ[′] ∈ RPi∈N [m][i], and any state-action pair (s, a) ∈S×A, there exist |
|
positive constants such that the following hold: 1): ∥ψθ(s, a)∥≤ Cψ; 2): ∥∇θJ(θ) −∇θJ(θ[′])∥≤ |
|
LJ ∥θ − θ[′]∥; where J(θ) is defined in (2) and ∥· ∥ denotes the ℓ2-norm. |
|
|
|
Assumption 6 says that the score function is uniformly bounded for any policy and the gradient of |
|
the objective function has a Lipschitz property with respect to the policy parameter. This assumption |
|
has also been adopted in the analysis of the single-agent actor-critic algorithm in (Qiu et al., 2021). |
|
We note that for the discounted reward problem, this gradient Lipschitz property can be guaranteed |
|
through (Xu et al., 2020, Assumption 2). We note that Assumption 6 can be satisfied by the class of |
|
soft-max policy under the Assumption 1, as in (Guo et al., 2021). |
|
|
|
3.3 THE OBJECTIVE FUNCTION |
|
|
|
The goal of the agents is to find a joint policy πθ to maximize the global average long-term reward. |
|
Mathematically, this can be written as: |
|
|
|
|
|
T −1 |
|
|
|
t=0 |
|
|
|
X |
|
|
|
|
|
πθ(a|s) · R[¯](s, a), (2) |
|
a∈A |
|
|
|
X |
|
|
|
|
|
rt[i]+1 |
|
i∈N |
|
|
|
X |
|
|
|
|
|
maximizeθ J(θ) = lim |
|
T →∞ |
|
|
|
|
|
T [E] |
|
|
|
|
|
dθ(s) |
|
s∈S |
|
|
|
X |
|
|
|
|
|
where R[¯](s, a) = N1 i∈N [R][i][(][s, a][)][ is the global average reward function. Let][ ¯]rt = N1 i∈N [r]t[i][,] |
|
|
|
then we have R[¯](s, a) = E[¯rt+1|st = s, at = a]. Next, we define the state-action value |
|
|
|
P P |
|
|
|
function: Qθ(s, a) = E[[P][∞]t=0 r[¯]t+1 − J(θ)|s0 = s, a0 = a, πθ], and the state value function |
|
Vθ(s) = a∈A [Q][θ][(][s, a][)][ ·][ π][θ][(][a][|][s][)][.][ The advantage function is defined as follows:] |
|
|
|
Advθ(s, a) = Qθ(s, a) − Vθ(s). (3) |
|
|
|
[P] |
|
|
|
3.4 POLICY GRADIENT THEOREM |
|
|
|
The gradient of a policy πθ for decentralized policy gradient is stated in the following theorem. |
|
Theorem 1 (Policy Gradient Theorem for MARL (Zhang et al., 2018)). For any θ, let πθ : S ×A → |
|
|
|
[0, 1] be a policy and let J(θ) be the global average long-term average return defined in (2). Then, |
|
the gradient of J(θ) with respect to parameter θ[i] can be computed as: |
|
|
|
∇θi J(θ) = Es∼dθ,a∼πθ [∇θi log πθ[i] [i] [(][a][i][|][s][)][ ·][ Adv][θ][(][s, a][)]][.] (4) |
|
|
|
4 A CONSENSUS-BASED ACTOR-CRITIC ALGORITHM |
|
|
|
In this section, we propose a consensus-based actor-critic algorithm that includes two key steps: |
|
actor and critic. In the critic step, the algorithm evaluates the value functions for the policy πθt at |
|
time t. After the critic step, the algorithm enters the actor step, which improves the policy parameter |
|
θt according to the direction from policy gradient as shown in Theorem 1. In both steps, we use |
|
constant step-sizes and adopt batch sampling. |
|
|
|
In this paper, we use linear function approximations for the value functions. Specifically, each agent |
|
i has a parameter w[i] ∈ R[K] to approximate the global value functions Vθ(s; w[i]) for each state s ∈S. |
|
For linear approximation, we have Vθ(s; w[i]) = φ(s)[T] w[i], where φ(s) ∈ R[K] denotes the feature for |
|
state s. As a result, the gradient of value function at state s with respect to approximation parameter |
|
w[i] is φ(s), i.e. ∇wVθ(s; w[i]) = φ(s). |
|
|
|
1) The Critic Step: The critic step is achieved through its own oracle, which is summarized in |
|
Algorithm 1. In the critic step, we allow the agents to communicate the approximation parameters |
|
w(s) with their neighbors via the communication network with consensus weight matrix A. For |
|
agent i ∈N, the parameter is locally updated by following rules: |
|
|
|
µ[i]k,τ +1 [= (1][ −] [β][)][µ]k,τ[i] [+][ βr]k,τ[i] +1 (5) |
|
|
|
δk,τ[i] [=][ r]k,τ[i] +1 [−] [µ]k,τ[i] [+][ φ][T][ (][s][k,τ] [+1][)][w]k[i] [−] [φ][T][ (][s][k,τ] [)][w]k[i] (6) |
|
|
|
|
|
----- |
|
|
|
Algorithm 1: Mini-batch TD learning for Critic |
|
|
|
Input : s0, πθt, φ, step-size β, critic step iteration number K, critic batch size M, the |
|
communication network A |
|
|
|
1 for k = 0, · · ·, Kc − 1 do |
|
|
|
2 sk,0 = sk−1,M ( when k = 0, sk,0 = s0); |
|
|
|
3 for all i ∈N do |
|
|
|
4 for τ = 0, · · ·, M − 1 do |
|
|
|
5 Execute action a[i]k,τ [∼] [π]θ[i] t[i] [(][·|][s][k,τ] [)][;] |
|
|
|
6 Observe the state sk,τ +1 and reward rk,τ[i] +1[;] |
|
|
|
7 Update µ[i]k,τ +1 [←] [(1][ −] [β][)][ ·][ µ]k,τ[i] [+][ β][ ·][ r]k,τ[i] +1[;] |
|
|
|
8 Update δk,τ[i] [←] [r]k,τ[i] +1 [−] [µ]k,τ[i] [+][ φ][T][ (][s][k,τ] [+1][)][w]k[i] [−] [φ][T][ (][s][k,τ] [)][w]k[i] [;] |
|
|
|
9 end |
|
|
|
|
|
M −1 |
|
|
|
10 Critic Step: ˜wk[i] [←] [w]k[i] [+][ β]M τ =0 [δ]k,τ[i] [·][ φ][(][s][k,τ] [)][;] |
|
|
|
11 Consensus Update wk[i] +1 [←] P[P]j∈Ni [A][(][i, j][)][ ·][ ˜]wk[j] [;] |
|
|
|
12 end |
|
|
|
13 end |
|
Output: sKc−1,M, wKc |
|
|
|
|
|
M −1 |
|
|
|
δk,τ[i] [·][ φ][(][s][k,τ] [)][,] (7) |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
w˜k[i] [=][ w]k[i] [+][ β] |
|
|
|
M |
|
|
|
|
|
where β > 0 is the step-size of the critic step, µ[i]k,τ [is the estimate of the long-term return of] |
|
agent i, and δk,τ[i] [is the local TD-error for agent][ i][ at iteration][ k][ using sample][ τ] [. Here, in each] |
|
iteration k in the critic step, the approximation parameter is locally updated through a batch of |
|
sampling as in (7), where the batch size is M . Then, agent i will further update the approximation |
|
parameter w[i] through a weighted average of its local and neighboring agents’ parameters as follows: |
|
wk[i] +1 [=][ P]j∈Ni [A][(][i, j][) ˜]wk[i] [.][ This batched sampling update continues for][ K][c][ iterations for each given] |
|
policy πθt. |
|
|
|
2) The Actor Step: As shown in Theorem 1, the advantage function needs to be known to compute |
|
the gradient. However, from the definition in (3), the joint action a also has to be known to compute |
|
the advantage function, whereas in our model, each agent can only observe its own action. As a |
|
result, an estimation of the advantage function is required. Here, we show that the global TD-error |
|
is an unbiased estimate of the advantage function. |
|
|
|
At time t, suppose we have samples st, at, st+1 and the rewards {rt[i]+1[}][i][∈N] [then the advantage] |
|
function is as follows: Advθ(st, at) = E[¯rt+1 − J(θ) + Vθ(St+1) − Vθ(st)|st, at], and the global |
|
TD-error can be computed as follows: δt = ¯rt+1 − µt + Vθ(st+1) − Vθ(st), where µt+1 = (1 − |
|
α) · µt + α · ¯rt+1 is the estimate for the average long term return, and α is the step-size for the actor |
|
step. Hence, we have that the expected global TD-error is the advantage function, i.e., E[δt|st, at] = |
|
Aθ(st, at). Thus, we can use this global TD-error as an unbiased estimate of the advantage function. |
|
For agent i, the local TD-error can be computed as δt[i] [=][ r]t[i]+1 [−] [µ]t[i] [+][ V][θ][(][s][t][+1][)][ −] [V][θ][(][s][t][)][, where] |
|
µ[i]t+1 [= (1][ −] [α][)][µ]t[i] [+][ αr]t[i]+1[. We also note that][ µ][t] [=][ 1]N i∈N [µ]t[i] [.] |
|
|
|
Thus, once each agent knows the global TD-error, the policy parameter can be updated according toP |
|
the policy gradient rule in (4). However, without any communication, each agent only has the knowledge of its own local TD-error. Moreover, we will show that the networked TD-error is actually the |
|
average of the local TD-errors. Specifically, |
|
|
|
|
|
δt = ¯rt+1 − µt + Vθ(st+1) − Vθ(st) = [1] |
|
|
|
|
|
rt[i]+1 [−] [1] |
|
|
|
N |
|
|
|
i∈N |
|
|
|
X |
|
|
|
|
|
µ[i]t [+][ V][θ][(][s][t][+1][)][ −] [V][θ][(][s][t][)] |
|
i∈N |
|
|
|
X |
|
|
|
|
|
= [1] |
|
|
|
|
|
|
|
[rt[i]+1 [−] [µ]t[i] [+][ V][θ][(][s][t][+1][)][ −] [V][θ][(][s][t][)] = 1] |
|
|
|
N |
|
|
|
i∈N |
|
|
|
X |
|
|
|
|
|
δt[i][.] |
|
i∈N |
|
|
|
X |
|
|
|
|
|
----- |
|
|
|
Algorithm 2: Minibatch-TD sharing for Actor Critic Algorithm |
|
|
|
Input : state feature matrix Φ, actor step-size α, Initial parameters θi for all i ∈N |
|
|
|
1 for t = 0, · · ·, T − 1 do |
|
|
|
2 critic update: wt, st,0 = Minibatch-TD-critic in Algorithm 1; |
|
|
|
3 for l = 0, · · ·, B − 1 do |
|
|
|
4 for all i ∈N do |
|
|
|
5 Execute action a[i]t,l [∼] [π]θ[i] t[i] [(][·|][s][t,l][)][;] |
|
|
|
6 Observe the state st,l+1 and reward rt,l[i] +1[;] |
|
|
|
7 Update µ[i]t,l+1 [←] [(1][ −] [α][)][ ·][ µ]t,l[i] [+][ α][ ·][ r]t,l[i] +1[;] |
|
|
|
8 Update δt,l[i] [←] [r]t,l[i] +1 [−] [µ]t,l[i] [+][ φ][T][ (][s][t,l][+1][)][w]t[i] [−] [φ][T][ (][s][t,l][)][w]t[i][;] |
|
|
|
9 Update ψt,l[i] [←∇][θ][i][ log][ π]θ[i] t[i][(][s][t,l][, a]t,l[i] [)][;] |
|
|
|
10 end |
|
|
|
11 end |
|
|
|
δt,[1] 0 - · · δt,[N]0 |
|
. . |
|
|
|
12 Let ∆0 = .. ... .. ; |
|
|
|
δt,B[1] −1 - · · δt,B[N] −1 |
|
|
|
|
|
|
|
13 for i ∈N do |
|
|
|
14 for k = 0 : tgossip − 1 do |
|
|
|
15 ∆k+1(:, i) ← [P]j∈Ni [A][(][i, j][)][ ·][ ∆][k][(:][, j][)][;] |
|
|
|
16 end |
|
|
|
17 end |
|
|
|
|
|
18 for all i ∈N do |
|
|
|
19 Let δ[˜]t,[i] 1:B−1 [= ∆][t]gossip [(:][, i][)][ ;] |
|
|
|
B−1 |
|
|
|
20 Actor Step: θt[i]+1 [←] [θ]t[i] [+][ α]B l=0 δ[˜]t,l[i] [·][ ψ]t,l[i] [;] |
|
|
|
21 end |
|
|
|
P |
|
|
|
22 end |
|
Output: θ ˆT [with][ ˆ]T chosen uniformly from {1, · · ·, T } |
|
|
|
|
|
For any time t, the average of the local TD-errors is an unbiased estimate of the advantage function. |
|
Therefore, we just need to let each agent communicate with its neighbors so that an average of all |
|
local TD-errors can be reached or estimated for all agents. |
|
|
|
From the results in (Nedic & Ozdaglar, 2009), we have limτ →∞ A[τ] (x[1], · · ·, x[N] )[T] = |
|
1 |
|
|
|
N i∈N [x][i][1][. However, this convergence is asymptotic, meaning that the exact estimation can] |
|
only be achieved with infinite iterations (i.e., τ →∞). In practice, since one can only apply fiP |
|
nite iterations, we use δ[˜]t[i] [to denote the estimate of the global TD-error maintained by agent][ i][ after] |
|
tgossip iterations of updates at time t, i.e., δ[˜]t[i] [= [][A][t][gossip] []][i][∆]t[0][, where][ ∆][0]t [= (][δ]t[1][,][ · · ·][, δ]t[N] [)][T][ is the][ N] [-] |
|
dimension vector of local TD-errors at time t. We note that agents do not need to know the weight |
|
information of other agents. Rather, each agent just needs to exchange updated estimate of the local TD-errors with its neighbors for tgossip rounds as shown in Lines 11-16 of Algorithm 2. This |
|
communication among agents is also done in a batch fashion, with batch size being B. This implies |
|
that for each outer iteration t ∈{0 · · ·, T − 1}, only tgossip rounds of communication for every B |
|
samples are needed. |
|
|
|
Combined with a B-batched Markovian sampling, the parameter θ[i] update for agent i ∈N can be |
|
written as follows: θt[i]+1 [=][ θ]t[i] [+][ α]B Bl=0−1 δ[˜]t,l[i] [·][ ψ]t,l[i] [where][ α][ is the step-size for the actor step and] |
|
|
|
ψt,l[i] [=][ ∇][θ][i][ log][ π]θ[i] t[i][(][s][t,l][, a]t,l[i] [)][ is the local score function for agent]P [ i][ using][ l][-th sample at time][ t][. The] |
|
actor step of our algorithm is illustrated in Algorithm 2. |
|
|
|
|
|
----- |
|
|
|
5 THEORETICAL CONVERGENCE ANALYSIS |
|
|
|
In this section, we present the convergence results for both the critic and actor steps in Theorems 2 |
|
and 3, respectively. Due to space limitation, we relegate the proofs to the supplementary material. |
|
|
|
5.1 CONVERGENCE ANALYSIS FOR CRITIC (ALGORITHM 1) |
|
|
|
For a given policy πθ, we define bπθ = Es∼dθ,a∼πθ [φ(s)(¯r(s, a) − J(θ))], where ¯r(s, a) = |
|
1 |
|
|
|
N i∈N [r][i][(][s, a][)][ and][ J][(][θ][)][ are as defined in (2). For all agents, the optimal solution (Wu et al.,] |
|
|
|
2020), (Qiu et al., 2021) of this critic learning is wθ[∗] [=][ −][A]π[−]θ[1][b][π]θ [, where][ A][π]θ [is defined as in (1).] |
|
|
|
P |
|
|
|
The invertiblity of Aπθ is due to the Assumptions 1 and 4 (see more details in (Qiu et al., 2021; |
|
Tsitsiklis & Van Roy, 1999)). Then, the convergence of the critic step is summarized as follows: |
|
Theorem 2 (Convergence and Sample Complexity of the Critic). Suppose that Assumptions 1-5 |
|
hold. For any given policy πθ, consider the iteration generated by Algorithm 1. Recall the definition |
|
of λA in the Assumption 5 and let β < min{ 2(1η−[N]η[−][N][1][−][1]) [,][ λ]128[A] [,] λ4A [}][. It then follows that:] |
|
|
|
N |
|
|
|
E[ ||wK[i] c [−] [w]θ[∗][||][2][]][ ≤] [κ][′]1[N][ 4][γ][2][K][c][ +][ κ][′′]2 [N][ 6][β][2][ +][ κ][3][N][ 5][γ][K][c] [β] |
|
|
|
i=1 |
|
|
|
X |
|
|
|
|
|
+ 2N (1 − [λ][A] w0 − wθ[∗][||][2]2 [+][ κ][4] (8) |
|
|
|
8 [β][)][K][c] [||][ ¯] M [N,] |
|
|
|
where γ := (1−η[N] [−][1])·(1+2β) < 1 and κ[′]1[, κ]2[′′][, κ][3][, κ][4] [are positive constants. If we further let][ K][c] [≥] |
|
|
|
12 [max][{][log][γ][−][1][ 6][κ]1[′]ǫ[N][ 4], log(1− λA8 [β][)][−][1] 12N || ¯w0ǫ−wθ[∗][||][2]2 }, β ≤ min{ N[1][3] 6κǫ[′′]2 [,] 2(1η−[N]η[−][N][1][−][1]) [,][ λ]128[A] [,] λ4A [}] |
|
|
|
6κ4N |
|
and M ≥ ǫ, then we have E[[P][N]i=1 [||][w]K[i] c [−] [w]θ[∗][||][2][]][ ≤] [ǫ][ for all]q[ i][ ∈N][ with total sample] |
|
|
|
complexity given by KcM = O( [N]ǫ [log(][ N]ǫ [))][.] |
|
|
|
Theorem 2 establishes a convergence result for the policy evaluation of a given policy πθ. We can |
|
see that our constant step-size batch-sampled critic process can achieve the same complexity of |
|
O˜(ǫ[−][1]) order-wise as the TD(0) learning in Doan et al. (2019), where diminishing step-sizes were |
|
used. On the other hand, in the single agent average reward setting of Qiu et al. (2021), there exists |
|
a non-vanishing error term in their critic convergence. In contrast, with proper choices of iteration |
|
number Kc and batch size M, our mean-square error can be arbitrarily small. |
|
|
|
5.2 CONVERGENCE ANALYSIS FOR ACTOR-CRITIC ALGORITHM (ALGORITHM 2) |
|
|
|
Define the approximation error introduced by using linear approximation in the critic step, ξapprox[critic] [=] |
|
maxθ∈RPi∈N [mi][ E][s][∼][d][θ] [[][|][V][π][θ] [(][s][)][ −] [V][w]πθ[∗] [(][s][)][|][2][]][. For a given policy][ π][θ][, this error represents the gap] |
|
between ground truth value function under such policy and the value function obtained by the best |
|
possible linear approximation. Such an error term is standard in the literature where linear approximations are adopted (Qiu et al., 2021; Xu et al., 2020). Let Rw, defined in Lemma 4, be an upper |
|
bound on ||wθ|| for all policy parameter θ. |
|
Theorem 3 (Overall Convergence Rate and Sample Complexity). Suppose that Assumption 1-6 |
|
hold. Consider the actor-critic algorithm in Algorithm 2. Let step-size α = 4L1J [. It then holds that:] |
|
|
|
|
|
T N |
|
t=1 i=1 [||][w]t[i] [−] [w]θ[∗]t [||][2] |
|
|
|
E[||∇θJ(θ ˜T [)][||][2][]][ ≤] [16][L][J] [r][max] + 72N + 18κ3N [3](1 − η[N] [−][1])[2][t][gossip] |
|
|
|
T T |
|
|
|
P P |
|
|
|
+ 72ξapprox[critic] [+ 288(][r][max][ +][ R][w][)][2][[1 + (2][κ][ −] [1)][ρ][]] N, (9) |
|
|
|
B(1 − ρ) |
|
|
|
|
|
, B ≥ 576 [(][r][max][+2][R][w](1[)]−[2][[1+(2]ρ) [κ][−][1)][ρ][]] |
|
|
|
|
|
where κ3 is a positive constant. Furthermore, let T ≥ [64][L][J]ǫ[r][max] |
|
|
|
|
|
and the communication round among the agents tgossip ≥ [1]2 [log][(1][−][η][N] [−][1][)][−][1][ 64][κ]ǫ[3][N][ 3] . Suppose for the |
|
|
|
same setting as in Theorem 1 holds so that E[[P][N]i=1 [||][w]K[i] c [−] [w]θ[∗][||][2][]][ ≤] 288ǫ N [for all][ 0][ ≤] [t][ ≤] |
|
|
|
T, then we have E[||∇θJ(θt)||[2]] ≤ ǫ + O(ξapprox[critic] [)][,][ with a total complexity of][ (][B][ +][ MK][c][)][T][ =] |
|
|
|
O( [N]ǫ[2][ 2][ log(][ N]ǫ [))][. And the communication complexity is][ (][K][c][ +][ t][gossip][)][T][ =][ O][(][ 1]ǫ [log(][ N]ǫ [))][.] |
|
|
|
|
|
----- |
|
|
|
(a) System size N = 6. (b) System size N = 10. (c) System size N = 15. |
|
|
|
Figure 1: Our TD-sharing algorithm vs classical MARL algorithm. |
|
|
|
Theorem 3 concludes the overall sample complexity of our proposed actor-critic algorithm. The |
|
sample complexity of O(ǫ[−][2] log(ǫ[−][1])) matches the state-of-the-art single-agent actor-critic RL by |
|
Xu et al. (2020) and the discounted MARL by Chen et al. (2021). |
|
|
|
We note that the overall communication complexity also matches that of (Chen et al., 2021) in the |
|
discounted reward setting. However, our work is still an improvement compared to the classical |
|
MARL in Zhang et al. (2018) for the average reward setting. Specifically, Zhang et al. (2018) needed |
|
a communication round after each sampling. By contrast, in this paper, we only need a communication round per O(ǫ[−][1]) sampling. This is thanks to the use of batch sampling in the actor step. Also |
|
because of the batch sampling, we are able to use a constant step-size for both actor and critic steps. |
|
Here, the overall communication cost is measured by the number of communication rounds rather |
|
than the size of bits transmitted over the network. We follow the standard definition of communication complexity in the literature, which is widely adopted in the literature, see (Chen et al., 2018) |
|
(Zhang et al., 2019). However, we note that tgossip, actor step batch size B and critic iteration rounds |
|
Kc, scale with O(log 1/ǫ), O(1/ǫ) and O(log 1/ǫ) respectively as indicated in the Theorem 3 and |
|
2. The amount of information (in terms of bits) is (KcNK + BNtgossip)T = O( ǫ[N][2][ log][ N]ǫ [)][.] |
|
|
|
6 EXPERIMENTAL RESULTS |
|
|
|
In this section, we conduct experiments to compare our proposed consensus-based TD-sharing |
|
MARL algorithm with the most related MARL algorithm 1 in Zhang et al. (2018) that also studied average reward. To our knowledge, this is the only work that is directly comparable to ours. The |
|
key difference is that the knowledge of joint action is assumed in Zhang et al. (2018), but not in our |
|
work. We vary the system size from N = 6, 10 to 15. The blue curve is our TD-sharing algorithm |
|
and the red curve is classical MARL algorithm in Zhang et al. (2018). The curves represent the |
|
average results of 10 trials and the 95% confidence intervals are also plotted. For the details, see A.1 |
|
in Appendix. The results in Figure 1 show that for different system sizes, both algorithms converge |
|
to a reasonable objective value. Note that if we use uniformly random policy as the baseline policy, |
|
then the objective values will be around 2 due to the setting of our experiments. All simulation |
|
results are above this threshold and our TD-sharing algorithm converge to a better objective value. |
|
See Section A.2 for addition experiment results. |
|
|
|
7 CONCLUSION AND DISCUSSION |
|
|
|
In this paper, we studied fully decentralized MARL in average reward setting and proposed a |
|
batch-sampled actor-critic algorithm. Our main contribution is to establish the first finite time convergence result for fully decentralized MARL in average reward setting, where the complexity is |
|
O(ǫ[−][2] log(ǫ[−][1])), which matches that of the state-of-the-art single agent RL. The algorithm reaches |
|
such convergence with a better communication efficiency. However, it is still in the preliminary |
|
stage of the convergence analysis of the MARL since we only used the vanilla average. The future |
|
direction will be how to design a more scalable algorithm in terms of system size. |
|
|
|
|
|
----- |
|
|
|
ACKNOWLEDGMENTS |
|
|
|
Hairi and Jia Liu’s work has been supported in part by NSF grants CAREER CNS-2110259, CNS2112471, CNS-2102233, CCF-2110252, and a Google Faculty Research Award. |
|
|
|
REFERENCES |
|
|
|
Rajendra Bhatia. Matrix analysis, volume 169 of. Graduate texts in mathematics, 1997. |
|
|
|
Tianyi Chen, Georgios B Giannakis, Tao Sun, and Wotao Yin. Lag: Lazily aggregated gradient for |
|
communication-efficient distributed learning. arXiv preprint arXiv:1805.09965, 2018. |
|
|
|
Ziyi Chen, Yi Zhou, Rongrong Chen, and Shaofeng Zou. Sample and communication-efficient |
|
decentralized actor-critic algorithms with finite-time analysis. arXiv preprint arXiv:2109.03699, |
|
2021. |
|
|
|
Thinh Doan, Siva Maguluri, and Justin Romberg. Finite-time analysis of distributed td (0) with linear |
|
function approximation on multi-agent reinforcement learning. In International Conference on |
|
Machine Learning, pp. 1626–1635. PMLR, 2019. |
|
|
|
Jakob Foerster, Gregory Farquhar, Triantafyllos Afouras, Nantas Nardelli, and Shimon Whiteson. |
|
Counterfactual multi-agent policy gradients. In Proceedings of the AAAI Conference on Artificial |
|
Intelligence, volume 32, 2018. |
|
|
|
Zuyue Fu, Zhuoran Yang, and Zhaoran Wang. Single-timescale actor-critic provably finds globally |
|
optimal policy. arXiv preprint arXiv:2008.00483, 2020. |
|
|
|
Xin Guo, Anran Hu, and Junzi Zhang. Theoretical guarantees of fictitious discount algorithms |
|
for episodic reinforcement learning and global convergence of policy gradient methods. arXiv |
|
preprint arXiv:2109.06362, 2021. |
|
|
|
Sham Machandranath Kakade. On the sample complexity of reinforcement learning. University of |
|
London, University College London (United Kingdom), 2003. |
|
|
|
Donghwan Lee, Niao He, Parameswaran Kamalaruban, and Volkan Cevher. Optimization for reinforcement learning: From a single agent to cooperative agents. IEEE Signal Processing Magazine, |
|
37(3):123–135, 2020. |
|
|
|
David A Levin and Yuval Peres. Markov chains and mixing times, volume 107. American Mathematical Soc., 2017. |
|
|
|
Angelia Nedic and Asuman Ozdaglar. Distributed subgradient methods for multi-agent optimization. IEEE Transactions on Automatic Control, 54(1):48–61, 2009. |
|
|
|
Shuang Qiu, Zhuoran Yang, Jieping Ye, and Zhaoran Wang. On finite-time convergence of actorcritic algorithm. IEEE Journal on Selected Areas in Information Theory, 2(2):652–664, 2021. |
|
|
|
Martin Riedmiller, Andrew Moore, and Jeff Schneider. Reinforcement learning for cooperating and |
|
communicating reactive agents in electrical power grids. In Workshop on Balancing Reactivity |
|
and Social Deliberation in Multi-Agent Systems, pp. 137–149. Springer, 2000. |
|
|
|
Shai Shalev-Shwartz, Shaked Shammah, and Amnon Shashua. Safe, multi-agent, reinforcement |
|
learning for autonomous driving. arXiv preprint arXiv:1610.03295, 2016. |
|
|
|
David Silver, Thomas Hubert, Julian Schrittwieser, Ioannis Antonoglou, Matthew Lai, Arthur Guez, |
|
Marc Lanctot, Laurent Sifre, Dharshan Kumaran, Thore Graepel, et al. A general reinforcement |
|
learning algorithm that masters chess, shogi, and go through self-play. Science, 362(6419):1140– |
|
1144, 2018. |
|
|
|
John N Tsitsiklis and Benjamin Van Roy. Average cost temporal-difference learning. Automatica, |
|
35(11):1799–1808, 1999. |
|
|
|
John N Tsitsiklis and Benjamin Van Roy. On average versus discounted reward temporal-difference |
|
learning. Machine Learning, 49(2):179–191, 2002. |
|
|
|
|
|
----- |
|
|
|
Yue Wu, Weitong Zhang, Pan Xu, and Quanquan Gu. A finite time analysis of two time-scale actor |
|
critic methods. arXiv preprint arXiv:2005.01350, 2020. |
|
|
|
Tengyu Xu, Zhe Wang, and Yingbin Liang. Improving sample complexity bounds for (natural) |
|
actor-critic algorithms. arXiv preprint arXiv:2004.12956, 2020. |
|
|
|
Zhuoran Yang, Yongxin Chen, Mingyi Hong, and Zhaoran Wang. On the global convergence of actor-critic: A case for linear quadratic regulator with ergodic cost. arXiv preprint |
|
arXiv:1907.06246, 2019. |
|
|
|
Chao Yu, Xin Wang, Xin Xu, Minjie Zhang, Hongwei Ge, Jiankang Ren, Liang Sun, Bingcai Chen, |
|
and Guozhen Tan. Distributed multiagent coordinated learning for autonomous driving in highways based on dynamic coordination graphs. IEEE Transactions on Intelligent Transportation |
|
Systems, 21(2):735–748, 2019. |
|
|
|
Kaiqing Zhang, Zhuoran Yang, Han Liu, Tong Zhang, and Tamer Basar. Fully decentralized multiagent reinforcement learning with networked agents. In International Conference on Machine |
|
Learning, pp. 5872–5881. PMLR, 2018. |
|
|
|
Kaiqing Zhang, Zhuoran Yang, and Tamer Bas¸ar. Multi-agent reinforcement learning: A selective |
|
overview of theories and algorithms. Handbook of Reinforcement Learning and Control, pp. |
|
321–384, 2021a. |
|
|
|
Xin Zhang, Jia Liu, Zhengyuan Zhu, and Elizabeth S Bentley. Compressed distributed gradient |
|
descent: Communication-efficient consensus over networks. In IEEE INFOCOM 2019-IEEE |
|
Conference on Computer Communications, pp. 2431–2439. IEEE, 2019. |
|
|
|
Xin Zhang, Zhuqing Liu, Jia Liu, Zhengyuan Zhu, and Songtao Lu. Taming communication and |
|
sample complexities in decentralized policy evaluation for cooperative multi-agent reinforcement |
|
learning. In Advances Neural Information Processing Systems (NeurIPS), Virtual Event, December 2021b. |
|
|
|
A APPENDIX |
|
|
|
In this section, we provide lemmas that lead to the proof of both Theorem 2 and Theorem 3. |
|
|
|
In this paper, we use || · || for 2-norm and || · ||T V for total variance norm. ⟨·, ·⟩ denotes the inner |
|
product. Superscript i in quantity x, i.e. x[i], denotes the x quantity correspond to agent i ∈N . |
|
λ(·) and σ(·) denote the eigenvalues and singular values of the corresponding matrix respectively. |
|
All vectors are assumed to be column vector, unless specified. (·)[T] is the transpose of an matrix or |
|
vector. We use 1 to denote all-1 vector with a proper dimension. For a matrix A, [A]i represents |
|
the i-th row of matrix A. Let [⃗]δ = [δ[1], · · ·, δ[N] ][T], i.e. the column vector for local TD-errors. In |
|
comparison, δ denotes the scalar global TD-error, i.e. δ = N[1] i∈N [δ][i][.] |
|
|
|
First, we explain the detail of our experiment setup. P |
|
|
|
A.1 EXPERIMENT SETUP |
|
|
|
We considered the same setting as in the Section 6.1 of Zhang et al. (2018). There are N agents, |
|
each has a binary-valued action space, i.e. A[i] = {0, 1}, for all i ∈N . In addition, in all the results |
|
shown here, we set |S| = 5 states. The elements in the transition matrix are uniformly sampled from |
|
the interval [0, 1] and normalized to be stochastic. We also added 10[−][5] onto each element to ensure |
|
ergodicity of the MDP such that the Assumption 1 is satisfied. For each agent i and state action pair |
|
(s, a), the mean reward R[i](s, a) is sampled uniformly from [0, 4]. The instantaneous rewards rt[i] [are] |
|
sampled from the uniform distribution [R[i](s, a) − 0.5, R[i](s, a) + 0.5]. The policy is parameterized |
|
following the Bolzmann policies, i.e., |
|
|
|
exp(qs,a[T] [i] [θ][i][)] |
|
πθ[i] [i] [(][s, a][i][) =] |
|
|
|
b[i]∈A[i][ exp(][q]s,b[T] [i] [θ][i][)] |
|
|
|
|
|
----- |
|
|
|
where qs,b[T] [i][ is the feature vector with the same dimension as][ θ][i][, for all][ s][ ∈S][ and][ i][ ∈N] [. Here,] |
|
we set m1 = · · · = mN = 5. The elements of qs,bi are uniformly sampled from [0, 1e3]. We set |
|
the dimension for state features K = 3. The feature matrix Φ are insured to have full column. The |
|
stepsizes for classical MARL are set as βw,t = t[0]1[.][51][ and][ β][θ,t][ =] t[0]1[.][52][ . The network matrix as chosen] |
|
|
|
as a ring network with diagonal elements being 0.4 and off diagonal elements 0.3 . |
|
|
|
For our algorithm, we used step-sizes α = 1 and β = 0.1. The batch sizes are B = 10, M = 10 |
|
and the critic iterations are 10 and actor iterations are T = 100. |
|
|
|
A.2 ADDITIONAL EXPERIMENT RESULTS |
|
|
|
We have modified classical MARL into i)constant stepsize MARL, ii)batch MARL and iii)batch |
|
constant stepsize MARL. For classical MARL and batch MARL, we chose the stepsizes as βw = |
|
1 1 |
|
|
|
t[0][.][65][ and][ β][θ][ =] t[0][.][85][ as in the paper (Zhang et al., 2018), and for the constant steptize MARL and] |
|
|
|
batch constant stepsize MARL, we chose stepsizes to be βw = 0.9 and βθ = 0.01. For batch MARL |
|
and batch constant stepsize MARL, we used batch size as 10. Other parameters are the same as in |
|
the A.1. We vary the system sizes from N = 5 to N = 15 and the empirical comparison results are |
|
in Figure 2 (a)-(c). In addition, we provide the comparison between average reward setting with the |
|
discounted counterpart of our algorithms for discounting factors ranging from γ = 0.1 to γ = 0.999. |
|
As we can see from Figure 2 (a)-(c), our TD-sharing algorithm performs well compared to the baseline algorithms. Among the modified algorithms, batch constant stepsize MARL shows improvement compared to the classical MARL in all three cases. Moreover in (b) and (c), when system size |
|
is larger, specifically N = 10 and N = 15, either batch modification or constant stepsize modification seem to improve the classical MARL. Yet in (a), for smaller system size, i.e. N = 5, only |
|
modifying to the batch size or constant stepsizes don’t seem to improve the performance. |
|
In addition, in (d), for discounted setting, as γ increases and gets closer to 1, the objective value |
|
is closer to the average reward setting. It is because as the discounting factor approaches 1, the |
|
effective horizon, which scales with O( [log]1−[ ǫ]γ[−][1] [)][ (Kakade, 2003), to an][ ǫ][ close stationary point gets] |
|
|
|
larger and larger. As a result, it will get closer to the average reward setting. However, we can see |
|
average setting value converges to a significantly higher value. More importantly, one advantage |
|
of the average reward setting is that with more samples, the policies can potentially keep updating |
|
and so is the objective value. From Figure 2(d), we can see that as the number of sample increases, |
|
the average reward setting objective value still evolves, which means the policies are keep updating. |
|
However, for the discounted reward case, the extra sample doesn’t affect the objective value. |
|
|
|
In Figure 3, we have shown the results of different network structures on the performance when |
|
system size is 10. We compared the ring network, small world network and 2-regular network. The |
|
small world network is generated with mean node degrees being 4 and rewiring probability being |
|
0.2. The entries of matrix A, for both small world network and regular network, are set as the way |
|
discussed after Assumption 3. Different network structures exhibit different performances, but all |
|
are better than baseline value 2. Among these three structures, within given sample numbers, ring |
|
network yields the best result. |
|
|
|
A.3 SUPPORTING LEMMAS FOR THEOREM 2 |
|
|
|
Because of the Assumption 1, by (Levin & Peres, 2017, Theorem 4.9), for aperiodic and irreducible |
|
Makrov chains, we can guarantee the following lemma holds: |
|
|
|
Lemma 1. For any policy parameter θ ∈ RPi∈N [m][i], consider the MDP with policy πθ and transition kernel P (·|s, a). Let dθ be the stationary distribution of the MDP. There exist constants κ > 0 |
|
and ρ ∈ (0, 1) such that sups∈S ||P (st|s0 = s) − dθ||T V ≤ κρ[t], ∀t ≥ 0. |
|
|
|
This lemma has been adopted directly as an assumption in many related works in theoretical analysis |
|
of RL (Xu et al., 2020; Chen et al., 2021; Qiu et al., 2021). |
|
|
|
As a result of the Lemma 1, the Markov chain of state-action pair {st, at}t≥0 for policy πθ also has |
|
the property of ergodicity. We state this result as the following lemma. |
|
|
|
|
|
----- |
|
|
|
(a) N = 5 (b) N = 10 |
|
|
|
2.5 |
|
|
|
|
|
1.5 |
|
|
|
1 |
|
|
|
0.5 |
|
|
|
|Col1|Col2|Col3|Col4|Col5|Col6|Col7| |
|
|---|---|---|---|---|---|---| |
|
|||||||| |
|
|||||||| |
|
||||TD-sharing Discounted Discounted|MARL Counter Counter|part with part with|=0.1 =0.5| |
|
|||||||| |
|
|||||||| |
|
|||||||| |
|
||||Discounted Discounted Discounted|Counter Counter Counter|part with part with part with|=0.9 =0.99 =0.999| |
|
|||||||| |
|
|||||||| |
|
|
|
|
|
TD-sharing MARL |
|
Discounted Counterpart with =0.1 |
|
|
|
Discounted Counterpart with =0.5 |
|
|
|
Discounted Counterpart with =0.9 |
|
|
|
Discounted Counterpart with =0.99 |
|
|
|
Discounted Counterpart with =0.999 |
|
|
|
|
|
Sample Number 10[5] |
|
|
|
(d) average vs discounted |
|
|
|
|
|
(c) N = 15 |
|
|
|
|
|
Figure 2: The empirical comparisons of algorithms |
|
|
|
|
|
2.25 |
|
|
|
2.2 |
|
|
|
|
|
2.15 |
|
|
|
2.1 |
|
|
|
|
|
2.05 |
|
|
|
2 |
|
|
|
1.95 |
|
|
|
|
|
1.9 |
|
|
|
|Col1|Col2|Col3|Col4|Col5|Col6|Col7| |
|
|---|---|---|---|---|---|---| |
|
|||||||| |
|
|||||||| |
|
|||||||| |
|
|||||||| |
|
||||Rin|||| |
|
|||||Rin|g Network|| |
|
||||Sm Reg|Sm Reg|all World N ular Netw|etwork ork| |
|
|||||||| |
|
|
|
|
|
Ring Network |
|
Small World Network |
|
Regular Network |
|
|
|
|
|
Sample Number 10[5] |
|
|
|
Figure 3: Different Network Structures on Performance When N = 10 |
|
|
|
|
|
----- |
|
|
|
Lemma 2. Suppose the Assumption 1 hold, let νθ be the stationary distribution of the state-action |
|
pair MDP. Then, we have |
|
|
|
sup ||P (st, at|s0 = s) − νθ||T V ≤ κρ[t], ∀t ≥ 0. (10) |
|
s∈S |
|
|
|
Proof. For any given s0 ∈S, by definition, we have |
|
|
|
|
|
||P (st, at|s0) − νθ||T V = [1] |
|
|
|
2 |
|
|
|
= [1] |
|
|
|
2 |
|
|
|
= [1] |
|
|
|
2 |
|
|
|
= [1] |
|
|
|
2 |
|
|
|
= [1] |
|
|
|
|
|
|P (st = s, at = a|s0) − νθ(s, a)| |
|
(s,a)∈S×A |
|
|
|
X |
|
|
|
|P (st = s|s0)πθ(at = a|st = s) − νθ[s][(][s][)][π][θ][(][a][|][s][)][|] |
|
(s,a)∈S×A |
|
|
|
X |
|
|
|
|(P (st = s|s0) − νθ[s][(][s][))][π][θ][(][a][|][s][)][|] |
|
(s,a)∈S×A |
|
|
|
X |
|
|
|
πθ(a|s)|P (st = s|s0) − νθ[s][(][s][)][|] |
|
(s,a)∈S×A |
|
|
|
X |
|
|
|
|P (st = s|s0) − νθ[s][(][s][)][|] |
|
s∈S |
|
|
|
X |
|
|
|
|
|
= ||P (st|s0) − νθ[s][||][T V] [≤] [κρ][t][.] |
|
|
|
Since it holds for all s0 ∈S, equation 10 holds. |
|
|
|
As a result of Assumption 5, we have the following lemmas. |
|
|
|
Lemma 3. For all policy πθ, we have |
|
|
|
⟨w, Aπθ w⟩≤− [λ]2[A] [||][w][||][2] (11) |
|
|
|
where λA is defined in Assumption 5. |
|
|
|
Proof. Because of the fact w[T] Aπθ w = w[T] A[T]πθ [w][, we have] |
|
|
|
|
|
w[T] Aπθ w = [1] πθ [w][)] |
|
|
|
2 [(][w][T][ A][π][θ] [w][ +][ w][T][ A][T] |
|
|
|
= [1] πθ [)][w] |
|
|
|
2 [w][T][ (][A][π][θ][ +][ A][T] |
|
|
|
|
|
≤− [λ][A] (12) |
|
|
|
2 [w][T][ w][ =][ −] [λ]2[A] [||][w][||][2][.] |
|
|
|
In fact, interestingly enough, the Assumption 5 and Lemma 3 are equivalent in a sense that if the |
|
statement in the Lemma 3 is taken as the assumption, the statement in the Assumption 5 can be |
|
obtained as a result. And the paper (Xu et al., 2020) used the statement in the Lemma 3 as an |
|
assumption, whereas in ours and (Qiu et al., 2021), we assumed Assumption 5. |
|
|
|
Lemma 4. For any given policy πθ, the corresponding optimal value function approximation parameter wθ[∗] [is uniformly bounded, specifically, there exists][ R][w][ :=][ 4][r]λ[max]A > 0 such that |
|
|
|
||wθ[∗][|| ≤] [R][w][.] (13) |
|
|
|
Proof. It’s easy to see that J(θ) ≤ rmax from equation 2 and ¯r(s, a) ≤ rmax for any (s, a) pair. |
|
Then, we have |
|
|
|
||wθ[∗][||][ =][ || −] [A][−]πθ[1][b][π]θ [||] |
|
|
|
|
|
----- |
|
|
|
−1 |
|
= || − Es∼dθ(s),s′∼P (·|s)[(φ(s[′]) − φ(s))φ[T] (s)] - Es∼dθ,a∼πθ [φ(s)(¯r(s, a) − J(θ))]|| |
|
|
|
−1 |
|
≤|| − Es∼dθ(s),s′∼P (·|s)[(φ(s[′]) − φ(s))φ[T] (s)] || · ||Es∼dθ,a∼πθ [φ(s)(¯r(s, a) − J(θ))]|| |
|
|
|
1 |
|
= - ||Es∼dθ,a∼πθ [φ(s)(¯r(s, a) − J(θ))]|| |
|
|
|
σmin −Es∼dθ(s),s′∼P (·|s)[(φ(s[′]) − φ(s))φ[T] (s)] |
|
|
|
r(s, a) − J(θ))]|| |
|
≤ [2][||][E][s][∼] [d][θ][,a][∼][π][θ] [[][φ][(][s][)(¯] |
|
|
|
λmin(−Aπθ − A[T]πθ [)] |
|
|
|
|
|
r(s, a)| + |J(θ)|)] |
|
≤ [2][E][s][∼][d][θ][,a][∼][π][θ] [[][||][φ][(][s][)][|| ·][ (][|][¯] |
|
|
|
|
|
= [4][r][max] |
|
|
|
λA |
|
|
|
|
|
where the third equality used the fact ||A[−][1]|| = σmin1(A) [and the second from the last inequality is] |
|
|
|
from Bhatia (1997) (Proposition III 5.1) . |
|
|
|
Note that, for agent i ∈N, the estimated long term average reward µ[i]k,τ [at sample][ τ][ of iteration][ k] |
|
in equation 5 can be written as |
|
|
|
|
|
µ[i]k,τ [=][ β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l[i] [+ (1][ −] [β][)][τ] [µ]0[i] [.] (14) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
Lemma 5. For any i ∈N and t ≥ 0, step size 0 < β < 1, for the estimated long term average |
|
reward for agent i, we have |
|
|
|
µ[i]t+1 [= (1][ −] [β][)][µ]t[i] [+][ βr]t[i] (15) |
|
|
|
is bounded by rmax, i.e. |µ[i]t+1[| ≤] [r][max][.] |
|
|
|
Proof. WLOG, we suppose that 0 < µ[i]0 [≤] [r][max][, we have] |
|
|
|
|µ[i]t+1[|][ =][ |][(1][ −] [β][)][µ]t[i] [+][ βr]t[i][|] |
|
|
|
≤ (1 − β)|µ[i]t[|][ +][ β][|][r]t[i][|][.] (16) |
|
|
|
By the supposition, we have |µ[i]0[| ≤] [r][max][. We assume][ |][µ]t[i][| ≤] [r][max] [holds for iteration][ t >][ 0][, then] |
|
for t + 1, by equation 16 |
|
|
|
|µ[i]t+1[| ≤] [(1][ −] [β][)][|][µ]t[i][|][ +][ β][|][r]t[i][| ≤] [(1][ −] [β][)][r][max] [+][ βr][max] [=][ r][max][.] |
|
|
|
Therefore, Lemma 5 holds by mathematical induction. |
|
|
|
For a given policy πθ, to establish a bound on the difference between the optimal approximation |
|
parameter wθ[∗][, we first derive a bound the difference between parameter][ w]k[i] [and the average among] |
|
all agents ¯wk at time k. Then, we derive a bound for the difference between average ¯wk and the |
|
optimal wθ[∗][.] |
|
|
|
We have following notations for the analysis. Given an agent i ∈N, we consider the consensus |
|
error at time k and we denote Q[i]k [=][ w]k[i] [−] w[¯]k, where ¯wk := N1 i∈N [w]k[i] [. Then, we denote the] |
|
|
|
matrix form as Qk = [Q[1]k[,][ · · ·][, Q][N]k []][ ∈] [R][K][×][N][ Then, we have the following lemma.] |
|
|
|
P |
|
|
|
Lemma 6. Suppose the Assumption 2 and 3 hold. For the consensus error matrix, we have |
|
|
|
||Qk+1|| ≤κ1N [2]γ[k]||Q0|| + κ[′]2[N][ 3][β] (17) |
|
|
|
where γ := (1 − η[N] [−][1]) · (1 + 2β) < 1, κ1 = 2(1 + 2β)(1 + η[−][(][N] [−][1)]) and κ[′]2 [=][ 4(1+][η][−]1[(][N]−[−]γ[1)][)][r][max] . |
|
|
|
Proof. By equation 7, the parameter update ˜wk[i] [for agent][ i][ at iteration][ k][ can be written as follows] |
|
|
|
|
|
w˜k[i] [=][ w]k[i] [+][ β] |
|
|
|
M |
|
|
|
= wk[i] [+][ β] |
|
|
|
|
|
δk,τ[i] [·][ φ][(][s][k,τ] [)] |
|
τ =0 |
|
|
|
X |
|
|
|
M −1 |
|
|
|
|
|
|
|
[rk,τ[i] +1 [−] [β] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l[i] [−] [(1][ −] [β][)][τ] [µ]0[i] |
|
l=1 |
|
|
|
X |
|
|
|
|
|
----- |
|
|
|
+ φ(sk,τ +1)[T] wk[i] [−] [φ][(][s][k,τ] [)][T][ w]k[i] []][ ·][ φ][(][s][k,τ] [)] |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] wk[i] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
= wk[i] [+][ β] |
|
|
|
M |
|
|
|
|
|
M −1 |
|
|
|
(rk,τ[i] +1 [−] [β] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l[i] [−] [(1][ −] [β][)][τ] [µ]0[i] [)][φ][(][s][k,τ] [)][.] (18) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
After a consensus step, the update will be |
|
|
|
|
|
wk[i] +1 [=] |
|
|
|
|
|
A(i, j) · ˜wk[j] |
|
j∈Ni |
|
|
|
X |
|
|
|
A(i, j) · [wk[j] [+][ β] |
|
|
|
M |
|
|
|
j∈Ni |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] wk[j] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
(rk,τ[j] +1 [−] [β] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l[j] [−] [(1][ −] [β][)][τ] [µ]0[j] [)][φ][(][s][k,τ] [)]] |
|
l=1 |
|
|
|
X |
|
|
|
M −1 |
|
|
|
|
|
A(i, j) · wk[j] [+][ β] |
|
|
|
M |
|
|
|
j∈Ni |
|
|
|
X |
|
|
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T][ X] A(i, j) · wk[j] |
|
τ =0 j∈Ni |
|
|
|
X |
|
|
|
τ |
|
|
|
|
|
M −1 |
|
|
|
+ [β] φ(sk,τ ) |
|
|
|
M |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
A(i, j) · (rk,τ[j] +1 [−] [β] |
|
j∈Ni |
|
|
|
X |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l[j] [−] [(1][ −] [β][)][τ] [µ]0[j] [)] |
|
l=1 |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] wk · (Ai,:)[T] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
= wk · (Ai,:)[T] + [β] |
|
|
|
|
|
M −1 |
|
|
|
+ [β] φ(sk,τ )Ai,:(rk,τ +1 − β |
|
|
|
M |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0) (19) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
where wk = [wk[1][, w]k[2][,][ · · ·][, w]k[N] []][ ∈] [R][K][×][N][ is the matrix form of all parameters at time][ k][,][ A][i,][:][ is] |
|
the i-th row of matrix A and rk,l = (rk,l[1] [,][ · · ·][, r]k,l[N] [)][. Now we consider the average dynamics of the] |
|
algorithm. Recall ¯wk = N[1] i∈N [w]k[i] [, then using equation 19 we have] |
|
|
|
|
|
w¯k+1 = [1] |
|
|
|
N |
|
|
|
= [1] |
|
|
|
|
|
wk[i] +1 |
|
i∈N |
|
|
|
X |
|
|
|
wk · (Ai,:)[T] + [β] |
|
|
|
M |
|
|
|
i∈N |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] wk · (Ai,:)[T] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )Ai,:(rk,τ +1 − β |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T][ 1] |
|
|
|
N [w][k][ ·][ 1] |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
= [1] |
|
|
|
N [w][k][ ·][ 1][ +][ β]M |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ ) [1] |
|
|
|
N |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
Ai,:(rk,τ +1 − β |
|
i∈N |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] w¯k |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
= ¯wk + [β] |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ ) [1] |
|
|
|
N [1][T][ (][r][k,τ] [+1][ −] [β] |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
----- |
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] w¯k |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
= ¯wk + [β] |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )(¯rk,τ +1 − β |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]r¯k,l − (1 − β)[τ] µ0) (20) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
where ¯rk,l = N[1] |
|
|
|
|
|
i∈N [r]k,l[i] [.] |
|
|
|
|
|
Given an agent i ∈N, we consider the consensus error at time k and recall Q[i]k [=][ w]k[i] [−] w[¯]k. Then, |
|
we have |
|
|
|
Q[i]k+1 [=][w]k[i] +1 [−] w[¯]k+1 |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] (wk · (Ai,:)[T] − w¯k) |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
=wk · (Ai,:)[T] − w¯k + [β] |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )(Ai,: − [1] |
|
|
|
N [1][T][ )(][r][k,τ] [+1][ −] [β] |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0). (21) |
|
l=1 |
|
|
|
X |
|
|
|
|
|
Then, for the matrix form Qk = [Q[1]k[,][ · · ·][, Q][N]k []][ ∈] [R][K][×][N] [, we have] |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] QkA[T] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
Qk+1 =QkA[T] + [β] |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[(A − [1] |
|
|
|
N [11][T][ )(][r][k,τ] [+1][ −] [β] |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
(1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0)][T] |
|
l=1 |
|
|
|
X |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[φ(sk,τ +1) − φ(sk,τ )][T] QkA[T] |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
=QkA[T] + [β] |
|
|
|
|
|
M −1 |
|
|
|
φ(sk,τ )[rk,τ +1 − β |
|
τ =0 |
|
|
|
X |
|
|
|
|
|
+ [β] |
|
|
|
|
|
τ |
|
|
|
(1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0][T] (A − N[1] [11][T][ )] |
|
l=1 |
|
|
|
X |
|
|
|
|
|
(22) |
|
|
|
where the first equality is due to A being doubly stochastic. For convenience, denote Bk = |
|
1 M −1 1 M −1 |
|
|
|
M τ =0 [φ][(][s][k,τ] [)[][φ][(][s][k,τ] [+1][)][ −] [φ][(][s][k,τ] [)]][T][ and][ C][k][ =] M τ =0 [φ][(][s][k,τ] [)[][r][k,τ] [+1][ −] [β][ P]l[τ]=1[(1][ −] |
|
|
|
β)[τ] [−][l]rk,l − (1 − β)[τ] µ0][T] . Then, iteratively we have |
|
|
|
P P |
|
|
|
Qk+1 =(I + βBk)QkA[T] + βCk(A − N[1] [11][T][ )] |
|
|
|
|
|
(I + βBt˜[)][C][t][(][A][ −] [1] (23) |
|
|
|
N [11][T][ )][A][k][−][t] |
|
|
|
t=0 t>t˜ |
|
|
|
X Y |
|
|
|
|
|
(I + βBt)Q0(A[T] )[k][+1] + β |
|
t=0 |
|
|
|
Y |
|
|
|
|
|
Then, the norm of the consensus error is following |
|
|
|
|
|
k |
|
|
|
(I + βBt˜[)][C][t][(][A][ −] [1] |
|
|
|
N [11][T][ )][A][k][−][t][||] |
|
|
|
t>t˜ |
|
|
|
Y |
|
|
|
|
|
(I + βBt)Q0(A[T] )[k][+1] + β |
|
t=0 |
|
|
|
Y |
|
|
|
|
|
||Qk+1|| =|| |
|
|
|
≤|| |
|
|
|
|
|
t=0 |
|
|
|
|
|
k |
|
|
|
||(I + βBt˜[)][|| · ||][C][t][|| · ||][(][A][ −] [1] |
|
|
|
N [11][T][ )][A][k][−][t][||][.] |
|
|
|
t>t˜ |
|
|
|
Y |
|
|
|
(24) |
|
|
|
|
|
(I + βBt)|| · ||Q0(A[T] )[k][+1]|| + β |
|
t=0 t=0 |
|
|
|
Y X |
|
|
|
|
|
Note that since A is doubly stochastic, so is A[T] . Let (A[T] )[k]:,i[+1] be the i-th column of matrix (A[T] )[k][+1]. |
|
From Nedic & Ozdaglar (2009), we know that |
|
|
|
|
|
||Q0(A[T] )[k]:,i[+1][||][ =][ ||][Q][0][(][A][T][ )]:[k],i[+1] − Q0 |
|
|
|
|
|
N [1][||] |
|
|
|
|
|
----- |
|
|
|
((A[T] )[k]j,i[+1] − [1] 0[||] |
|
|
|
N [)][Q][j] |
|
|
|
j∈N |
|
|
|
X |
|
|
|
|
|
= || |
|
|
|
|
|
|(A[T] )[k]j,i[+1] − [1] 0[||] |
|
|
|
N [| · ||][Q][j] |
|
|
|
j∈N |
|
|
|
X |
|
|
|
|
|
≤ N · 2 [1 +][ η][−][(][N] [−][1)] 0[||] |
|
|
|
1 − η[N] [−][1][ (1][ −] [η][N] [−][1][)][k][+1][ ·][ max]j∈N [||][Q][j] |
|
|
|
≤ 2N (1 + η[−][(][N] [−][1)])(1 − η[N] [−][1])[k] - ||Q0||. (25) |
|
|
|
Hence, ||Q0A[k][+1]|| ≤ 2N [2](1 + η[−][(][N] [−][1)])(1 − η[N] [−][1])[k] - ||Q0||. To bound the first term in the RHS |
|
of equation 24, we have |
|
|
|
|
|
|| (I + βBt)|| ≤ |
|
|
|
t=0 |
|
|
|
Y |
|
|
|
|
|
||I + βBt|| |
|
t=0 |
|
|
|
Y |
|
|
|
k |
|
|
|
(||I|| + β||Bt||) |
|
t=0 |
|
|
|
Y |
|
|
|
k |
|
|
|
(1 + 2β) = (1 + 2β)[k][+1]. (26) |
|
|
|
t=0 |
|
|
|
Y |
|
|
|
|
|
In order to make sure it converges to the neighborhood of a consensus in the limit, the step size has |
|
η[N] [−][1] |
|
to satisfy γ := (1 − η[N] [−][1]) · (1 + 2β) < 1, which results in β < 2(1−η[N] [−][1]) [. Hence, for the first] |
|
|
|
term in the RHS of equation 24, we have |
|
|
|
|
|
|| (I + βBt)|| · ||Q0A[k][+1]|| ≤ κ1N [2]γ[k]||Q0|| |
|
|
|
t=0 |
|
|
|
Y |
|
|
|
|
|
where κ1 = 2(1 + 2β)(1 + η[−][(][N] [−][1)]). |
|
|
|
Furthermore, we have |
|
|
|
||(A − [1] (27) |
|
|
|
N [11][T][ )][A][k][−][t][||][ =][ ||][A][k][−][t][+1][ −] N[1] [11][T][ || ≤] [2][N][ 2][(1 +][ η][−][(][N] [−][1)][)(1][ −] [η][N] [−][1][)][k][−][t][.] |
|
|
|
|
|
And ||Ct|| ≤ 2Nrmax, where rmax = supi,s,a r[i](s, a). This is because |
|
|
|
M −1 τ |
|
|
|
||Ck|| = || [1] φ(sk,τ )[rk,τ +1 − β (1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0][T] || |
|
|
|
M |
|
|
|
τ =0 l=1 |
|
|
|
X X |
|
|
|
M −1 τ |
|
|
|
≤ [1] ||φ(sk,τ )[rk,τ +1 − β (1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0][T] || |
|
|
|
M |
|
|
|
τ =0 l=1 |
|
|
|
X X |
|
|
|
M −1 τ |
|
|
|
≤ [1] ||φ(sk,τ )|| · ||rk,τ +1 − β (1 − β)[τ] [−][l]rk,l − (1 − β)[τ] µ0|| |
|
|
|
M |
|
|
|
τ =0 l=1 |
|
|
|
X X |
|
|
|
M −1 τ |
|
|
|
≤ [1] [||rk,τ +1|| + β (1 − β)[τ] [−][l]||rk,l|| + (1 − β)[τ] ||µ0||] |
|
|
|
M |
|
|
|
τ =0 l=1 |
|
|
|
X X |
|
|
|
M −1 |
|
|
|
≤ [1] [Nrmax + β [1][ −] [(1][ −] [β][)][τ] |
|
|
|
M 1 − (1 − β) [Nr][max][ + (1][ −] [β][)][τ] [Nr][max][] = 2][Nr][max][.] |
|
|
|
τ =0 |
|
|
|
X |
|
|
|
|
|
Then for the second term of equation 24, we have |
|
|
|
|
|
k |
|
|
|
||(I + βBt˜[)][|| · ||][C][t][|| · ||][(][A][ −] [1] |
|
|
|
N [11][T][ )][A][k][−][t][||] |
|
|
|
t>t˜ |
|
|
|
Y |
|
|
|
|
|
t=0 |
|
|
|
|
|
----- |
|
|
|
(1 + 2β)[k][−][t] - 2Nrmax · 2N [2](1 + η[−][(][N] [−][1)])(1 − η[N] [−][1])[k][−][t] |
|
t=0 |
|
|
|
X |
|
|
|
|
|
≤β |
|
|
|
|
|
k |
|
|
|
γ[k][−][t] = κ2N [3]β [1][ −] [γ][k][+1] (28) |
|
|
|
1 − γ |
|
|
|
t=0 |
|
|
|
X |
|
|
|
|
|
=κ2β |
|
|
|
|
|
where κ2 = 4(1 + η[−][(][N] [−][1)])rmax. Further, for consensus error equation 24, we have |
|
|
|
||Qk+1|| ≤ κ1N [2]γ[k]||Q0|| + κ2N [3]β [1][ −] [γ][k][+1] ≤κ1N [2]γ[k]||Q0|| + κ[′]2[N][ 3][β] (29) |
|
|
|
1 − γ |
|
|
|
|
|
where κ[′]2 [=] 1κ−2γ [.] |
|
|
|
The corresponding average parameter under the policy πθ will converge to the solution to |
|
the following equation wθ[∗] = A[−]θ [1][b][θ][, where][ A][θ] = Es∼dθ(s)[φ(s)(φ(s[′]) − φ(s))[T] ], bθ = |
|
E(s,a)∼dθ(s,a)[(¯r(s, a) − J(θ))φ(s)] and ¯r(s, a) = N[1] i∈N [r][i][(][s, a][)][.] |
|
|
|
Now consider || ¯wk − w[∗]||. Recall the from the average parameter equation equation 20 and theP |
|
corresponding ODE is |
|
|
|
w˙ θ = Aθwθ + bθ. (30) |
|
|
|
For the difference between average dynamics and the optimal value, we have the following lemma. |
|
|
|
|
|
λ4A [}][ and][ M][ ≥] [(][ 1]λA [+][ β][)][ 6144[1+(](1−ρ)[κ]λ[−]A[1)][ρ][]], we have |
|
|
|
|
|
Lemma 7. For β ≤ min{ 128[λ][A] [,] |
|
|
|
|
|
w [+][ r]max[2] [)[1 + (][κ][ −] [1)][ρ][]] |
|
|
|
E[|| ¯wK − wθ[∗][||][2]2[]][ ≤] [(1][ −] [λ][A] w0 − wθ[∗][||][2]2 [+ ( 1] + β) [1536(4][R][2] . |
|
|
|
8 [β][)][K] [||][ ¯] λA (1 − ρ)λAM |
|
|
|
(31) |
|
|
|
Proof. The proof follows from verifying the Assumption 3 of Xu et al. (2020), then we can apply |
|
the results from Theorem 4 of Xu et al. (2020). |
|
|
|
1. For item 1 in the assumption, it’s easy to check that ||Aθ|| ≤ CA = 4 by Assumption 4 and |
|
bθ ≤ Cb = 2rmax. |
|
|
|
2. For item 2, it holds because of Lemma 3. |
|
|
|
3. For item 3, it holds because of Lemma 1. |
|
|
|
And we recall that the bound on ||wθ[∗][||][ is][ R][w][ =][ 4][r]λ[max]A [. Hence, by Theorem 4 of Xu et al. (2020),] |
|
|
|
β ≤ min{ 8[λ]C[A]A[2] [,] λ4A [}][ and][ M][ ≥] [(][ 2]λA [+ 2][β][)][ 192][C](1A[2] −[[1+(]ρ)λ[κ]A[−][1)][ρ][]], equation 31 holds. |
|
|
|
A.4 PROOF OF THEOREM 2 |
|
|
|
As a result from Lemma 4 and Lemma 7, we provide the following proof for Theorem 2. |
|
|
|
Proof. Therefore, by equation 17 and equation 31, we have |
|
|
|
|
|
E[ ||wK[i] [−] [w]θ[∗][||][2][]] |
|
|
|
i=1 |
|
|
|
X |
|
|
|
N N |
|
|
|
≤ 2E[ ||wK[i] [−] w[¯]K ||[2]] + 2E[ || ¯wK − wθ[∗][||][2][]] |
|
|
|
i=1 i=1 |
|
|
|
X X |
|
|
|
≤ 2||QK ||[2] + 2N E[|| ¯wK − wθ[∗][||][2][]] |
|
|
|
|
|
≤ 2(κ1N [2]γ[K] ||Q0|| + κ[′]2[N][ 3][β][)][2][ + 2][N] [(1][ −] [λ][A] w0 − wθ[∗][||][2]2 |
|
|
|
8 [β][)][K] [||][ ¯] |
|
|
|
|
|
----- |
|
|
|
A[R]w[2] [+][ C]b[2][)[1 + (][κ][ −] [1)][ρ][]] |
|
|
|
+ 2N ( [2] + 2β) [192(][C] [2] |
|
|
|
λA (1 − ρ)λAM |
|
|
|
|
|
≤ κ[′]1[N][ 4][γ][2][K][ +][ κ][′′]2 [N][ 6][β][2][ +][ κ][3][N][ 5][γ][K] [β][ + 2][N] [(1][ −] [λ][A] w0 − wθ[∗][||][2]2 [+][ κ][4] |
|
|
|
8 [β][)][K] [||][ ¯] M [N] |
|
|
|
where κ[′]1 = 2κ[2]1[||][Q][0][||][2][,] κ”2 = 2κ[′]2[2][,] κ3 = 4κ1κ[′]2[||][Q][0][||][ and][ κ][4] = ( λ[1]A [+] |
|
|
|
β) [3072(4][R]w[2] [+](1[r]−max[2]ρ)λ[)[1+(]A [κ][−][1)][ρ][]] . |
|
|
|
|
|
A.5 PROOF OF THEOREM 3 |
|
|
|
We use wt[∗] [to denote the optimal value function parameter under policy][ θ][t] [at time][ t][ and] |
|
wt = [(wt[1][)][T][,][ · · ·][,][ (][w]t[N] [)][T][ ]][T] ∈ R[N] [×][K], which is the aggregated function approximation parameters from Line 2 of Algorithm 2. Recall the δ[˜]t,l[i] [generated by Line 18 of Algorithm 2,] |
|
which is the i-th agent’s estimate of the global TD-error from sample l at time t after consensus. Let vt[i][(][w][t][) =] B1 Bl=0−1 δ[˜]t,l[i] [(][w][t][)][ ·][ ψ]t,l[i] [and][ v][t][(][w][t][) = [(][v]t[1][(][w][t][))][T][,][ · · ·][,][ (][v]t[N] [(][w][t][))][T][ ]][T][,] |
|
|
|
h[i]t[(][w]t[∗][) =] B1 Bl=0−1 [δ][t,l][(][w]t[∗][)]P[ ·][ ψ]t,l[i] [and][ h][t][(][w]t[∗][) = [(][h]t[1][(][w]t[∗][))][T][,][ · · ·][,][ (][h]t[N] [(][w]t[∗][))][T][ ]][T][, Adv][w][(][s, a][) =] |
|
|
|
Es′∼P (·|s,a),r∼dr(s,a)[δw(s, a, s[′])|s, a] and g[i](w, θ) = E[Advw(s, a)ψθi (s, a[i])], where dr(s, a) is |
|
P |
|
the reward distribution of state-action pair (s, a). By Taylor expansion and the Lipschitz property |
|
from Assumption 6, we have |
|
|
|
|
|
J(θt+1) ≥ J(θt) + ⟨∇θJ(θt), θt+1 − θt⟩− [L][J] |
|
|
|
2 [||][θ][t][+1][ −] [θ][t][||][2] |
|
|
|
= J(θt) + α⟨∇θJ(θt), vt(wt) −∇θJ(θt) + ∇θJ(θt)⟩− [L][J] [α][2] |
|
|
|
|
|
||vt(wt)||[2] |
|
|
|
|
|
= J(θt) + α||∇θJ(θt)||[2] + α⟨∇θJ(θt), vt(wt) −∇θJ(θt)⟩− [L][J] [α][2] |
|
|
|
2 |
|
|
|
−∇θJ(θt) + ∇θJ(θt)||[2] |
|
|
|
|
|
||vt(wt) |
|
|
|
|
|
≥ J(θt) + ( [1] |
|
|
|
2 [α][ −] [L][J] [α][2][)][||∇][θ][J][(][θ][t][)][||][2][ −] [(1]2 [α][ +][ L][J] [α][2][)][||][v][t][(][w][t][)][ −∇][θ][J][(][θ][t][)][||][2] |
|
|
|
where the last inequality is because |
|
|
|
|
|
⟨∇θJ(θt), vt(wt) −∇θJ(θt)⟩≥− [1] |
|
|
|
2 [||∇][θ][J][(][θ][t][)][||][2][ −] 2[1] [||][v][t][(][w][t][)][ −∇][θ][J][(][θ][)][||][2][,] |
|
|
|
|
|
and |
|
|
|
||vt(wt) −∇θJ(θt) + ∇θJ(θt)||[2] ≤ 2||vt(wt) −∇θJ(θt)||[2] + 2||∇θJ(θt)||[2]. (32) |
|
|
|
Taking expectations on both sides conditioned on the filtration Ft and rearranging the terms, we |
|
have |
|
|
|
|
|
( [1] |
|
|
|
2 [α][ −] [L][J] [α][2][)][E][[][||∇][θ][J][(][θ][t][)][||][2][|F][t][]][ ≤] [E][[][J][(][θ][t][+1][)][|F][t][]][ −] [J][(][θ][t][)] |
|
|
|
+ ( [1] (33) |
|
|
|
2 [α][ +][ L][J] [α][2][)][E][[][||][v][t][(][w][t][)][ −∇][θ][J][(][θ][t][)][||][2][|F][t][]][.] |
|
|
|
Then, we establish upper bound on the third term of the RHS. By definition, we have |
|
|
|
||vt(wt) −∇θJ(θt)||[2] |
|
|
|
|
|
= ||vt(wt) − vt(wt[∗][) +][ v][t][(][w]t[∗][)][ −] [h][t][(][w]t[∗][) +][ h][t][(][w]t[∗][)][ −] [g][(][w]t[∗][, θ][t][) +][ g][(][w]t[∗][, θ][t][)][ −∇][θ][J][(][θ][t][)][||][2] |
|
|
|
≤ 6||vt(wt) − vt(wt[∗][)][||][2][ + 6][||][v][t][(][w]t[∗][)][ −] [h][t][(][w]t[∗][)][||][2][ + 6][||][h][t][(][w]t[∗][)][ −] [g][(][w]t[∗][, θ][t][)][||][2] |
|
|
|
+ 6||g(wt[∗][, θ][t][)][ −∇][θ][J][(][θ][t][)][||][2][.] (34) |
|
|
|
We note that ||vt(wt) − vt(wt[∗][)][||][2][ =][ P]i∈N [||][v]t[i][(][w][t][)][ −] [v]t[i][(][w]t[∗][)][||][2][ and the other three terms in equa-] |
|
tion 34 can also be similarly decomposed. For the first term in the RHS of equation 34, we have |
|
|
|
||vt[i][(][w][t][)][ −] [v]t[i][(][w]t[∗][)][||][2] |
|
|
|
|
|
----- |
|
|
|
B−1 |
|
|
|
δ˜t,l[i] [(][w][t][)][ ·][ ψ]t,l[i] [−] [1] |
|
|
|
B |
|
|
|
l=0 |
|
|
|
X |
|
|
|
B−1 |
|
|
|
|
|
B−1 |
|
|
|
δ˜t,l[i] [(][w]t[∗][)][ ·][ ψ]t,l[i] [||][2] |
|
l=0 |
|
|
|
X |
|
|
|
|
|
= || [1] |
|
|
|
B |
|
|
|
= || [1] |
|
|
|
B |
|
|
|
= || [1] |
|
|
|
|
|
|
|
[δ[˜]t,l[i] [(][w][t][)][ −] δ[˜]t,l[i] [(][w]t[∗][)]][ ·][ ψ]t,l[i] [||][2] |
|
l=0 |
|
|
|
X |
|
|
|
|
|
B−1 |
|
|
|
|| [1] [A[t][gossip] ]i(wt − wt[∗][1][T][ )][T][ [][φ][(][s][t,l][+1][)][ −] [φ][(][s][t,l][)]][ ·][ ψ]t,l[i] [||][2] |
|
|
|
B |
|
|
|
l=0 |
|
|
|
X |
|
|
|
max t [1][T][ )][T][ [][φ][(][s][t,l][+1][)][ −] [φ][(][s][t,l][)]][ ·][ ψ]t,l[i] [||][2] |
|
l∈{0,···,B−1} [||][[][A][t][gossip] []][i][(][w][t][ −] [w][∗] |
|
|
|
max t [1][T][ ||][2][ · ||][ψ]t,l[i] [||][2] |
|
l∈{0,···,B−1} [||][[][A][t][gossip] []][i][||][2][||][[][φ][(][s][t,l][+1][)][ −] [φ][(][s][t,l][)]][||][2][||][w][t][ −] [w][∗] |
|
|
|
|
|
≤ 4 · ||wt − wt[∗] [⊗] [1][||][2][ = 4] |
|
|
|
where the third equality is from |
|
|
|
δ˜t,l[i] [(][w][t][)][ −] δ[˜]t,l[i] [(][w]t[∗][)] |
|
|
|
|
|
||wt[i] [−] [w]t[∗][||][2] (35) |
|
i=1 |
|
|
|
X |
|
|
|
|
|
= [A[t][gossip] ]i[⃗]δt,l(wt) − [A[t][gossip] ]i[⃗]δt,l(wt[∗][)] |
|
|
|
= [A[t][gossip] ]i[[⃗]δt,l(wt) − [⃗]δt,l(wt[∗][)]] |
|
|
|
φ[T] (st,l+1)(wt[1] [−] [w]t[∗][)][ −] [φ][T][ (][s][t,l][)(][w]t[1] [−] [w]t[∗][)] |
|
. |
|
|
|
= [A[t][gossip] ]i .. |
|
|
|
φ[T] (st,l+1)(wt[N] [−] [w]t[∗][)][ −] [φ][T][ (][s][t,l][)(][w]t[N] [−] [w]t[∗][)] |
|
|
|
|
|
|
|
|
|
= [A[t][gossip] ]i(wt − wt[∗][1][T][ )][T][ [][φ][(][s][t,l][+1][)][ −] [φ][(][s][t,l][)]][.] |
|
|
|
For the second term in the RHS of equation 34, we have |
|
|
|
||vt[i][(][w]t[∗][)][ −] [h]t[i][(][w]t[∗][)][||][2] |
|
|
|
|
|
B−1 B−1 |
|
|
|
δ˜t,l[i] [(][w]t[∗][)][ ·][ ψ]t,l[i] [−] [1] δt,l(wt[∗][)][ ·][ ψ]t,l[i] [||][2] |
|
|
|
B |
|
|
|
l=0 l=0 |
|
|
|
X X |
|
|
|
B−1 |
|
|
|
[δ[˜]t,l[i] [(][w]t[∗][)][ −] [δ][t,l][(][w]t[∗][)]][ ·][ ψ]t,l[i] [||][2] |
|
l=0 |
|
|
|
X |
|
|
|
|
|
= || [1] |
|
|
|
B |
|
|
|
= || [1] |
|
|
|
B |
|
|
|
= || [1] |
|
|
|
|
|
B−1 |
|
|
|
l=0 |
|
|
|
X |
|
|
|
|
|
|
|
[A[t][gossip] ]i − N[1] [1][T] ⃗δt,l(wt[∗][)][ ·][ ψ]t,l[i] [||][2] |
|
|
|
|
|
|
|
|
|
max [A[t][gossip] ]i − [1] ⃗δt,l(wt[∗][)][ ·][ ψ]t,l[i] [||][2] |
|
l∈{0,···,B−1} [||] N [1][T] |
|
|
|
|
|
max δt,l(wt[∗][)][||][2][ · ||][ψ]t,l[i] [||][2] |
|
l∈{0,···,B−1} [||][[][A][t][gossip] []][i][ −] N[1] [1][T][ ||][2][ · ||][⃗] |
|
|
|
|
|
max δt,l(wt[∗][)][||][2] |
|
l∈{0,···,B−1} [N] [(21 +]1 −[ η][−]η[N][(][N][−][−][1][1)][ (1][ −] [η][N] [−][1][)][t][gossip][+1][)][2][ · ||][⃗] |
|
|
|
|
|
≤ 16N [2]((1 + η[−][(][N] [−][1)])(1 − η[N] [−][1])[t][gossip] )[2](rmax + Rw)[2] = κ3N [2](1 − η[N] [−][1])[2][t][gossip] (36) |
|
|
|
where κ3 = 16(1 + η[−][(][N] [−][1)])[2](rmax + Rw)[2]. We note that the second equality is because |
|
|
|
δ˜t,l[i] [(][w]t[∗][)][ −] [δ][t,l][(][w]t[∗][)] |
|
|
|
= [A[t][gossip] ]i[⃗]δt,l(wt[∗][)][ −] N[1] [1][T] [⃗]δt,l(wt[∗][)] |
|
|
|
|
|
|
|
[A[t][gossip] ]i − N[1] [1][T] ⃗δt,l(wt[∗][)][.] |
|
|
|
|
|
|
|
|
|
----- |
|
|
|
For the last inequality, we note that δt,l(w[∗]) is bounded because rewards and feature vectors are |
|
bounded, µ[i] is bounded for constant step size and wt[∗] [is bounded from the critic step. That is, for] |
|
j ∈N entry in [⃗]δt,l(w[∗]) by definition, |
|
|
|
δt,l[j] [(][w][∗][) =][ r]t,l[j] [−] [µ]t,l[j] [+ [][φ][(][s][t,l][+1][)][ −] [φ][(][s][t,l][)]][T][ w]t[∗][.] |
|
|
|
Hence, its 2-norm bound is |
|
|
|
||δt,l[j] [(][w][∗][)][||][ =][ ||][r]t,l[j] [−] [µ]t,l[j] [+ [][φ][(][s][t,l][+1][)][ −] [φ][(][s][t,l][)]][T][ w]t[∗][||] |
|
|
|
≤||rt,l[j] [||][ +][ ||][µ][j]t,l[||][ +][ ||][φ][(][s][t,l][+1][)][ −] [φ][(][s][t,l][)][|| · ||][w]t[∗][||] |
|
|
|
≤ rmax + rmax + [||φ(st,l+1)|| + ||φ(st,l)||] · Rw ≤ 2rmax + 2Rw. (37) |
|
|
|
For the last term in equation 34, we have |
|
|
|
||g(wt[∗][, θ][t][)][ −∇][θ][J][(][θ][t][)][||][2] |
|
|
|
= ||Edθt (s,a)[Advwt[∗][(][s, a][)][ψ][θ][t] [(][s, a][)]][ −] [E][d][θ]t [(][s,a][)][[][Adv][θ][t][(][s, a][)][ψ][θ][t] [(][s, a][)]][||][2] |
|
|
|
= ||Edθt (s,a)[(Advwt[∗][(][s, a][)][ −] [Adv][θ][t] [(][s, a][))][ψ][θ][t] [(][s, a][)]][||][2] |
|
|
|
≤ (Edθt (s,a)[||(Advwt[∗][(][s, a][)][ −] [Adv][θ][t][(][s, a][))][ψ][θ][t] [(][s, a][)][||][])][2] |
|
|
|
≤ (Edθt (s,a)[||Advwt[∗][(][s, a][)][ −] [Adv][θ][t] [(][s, a][)][|| · ||][ψ][θ][t] [(][s, a][)][||][])][2] |
|
|
|
≤ (Edθt (s,a)[|Advwt[∗][(][s, a][)][ −] [Adv][θ][t][(][s, a][)][|][])][2] |
|
|
|
= (Edθt (s,a)[|E[Vwt[∗] [(][s][′][)][|][s, a][]][ −] [V][w]t[∗][(][s][)][ −] [E][[][V][θ]t[∗][(][s][′][)][|][s, a][] +][ V][θ]t[∗][(][s][)][|][])][2] |
|
|
|
≤ (Edθt (s,a)[|E[Vwt[∗] [(][s][′][)][ −] [V][θ]t[∗][(][s][′][)][|][s, a][]][|][ +][ |][V][w]t[∗] [(][s][)][ −] [V][θ]t[∗][(][s][)][|][])][2] |
|
|
|
≤ (Edθt (s,a)[E[|Vwt[∗] [(][s][′][)][ −] [V][θ]t[∗][(][s][′][)][||][s, a][] +][ |][V][w]t[∗] [(][s][)][ −] [V][θ]t[∗][(][s][)][|][])][2] |
|
|
|
= (E[|Vwt[∗] [(][s][)][ −] [V][θ]t[∗][(][s][)][|][] +][ E][[][|][V][w]t[∗] [(][s][)][ −] [V][θ]t[∗][(][s][)][|][])][2] |
|
|
|
≤ 4(E[|Vwt[∗] [(][s][)][ −] [V][θ]t[∗][(][s][)][|][])][2][ ≤] [4][E][[][|][V][w]t[∗] [(][s][)][ −] [V][θ]t[∗][(][s][)][|][2][]][ ≤] [4][ξ]approx[critic] [.] (38) |
|
|
|
For each i ∈N, we have |
|
|
|
||h[i]t[(][w]t[∗][)][ −] [g][i][(][w]t[∗][, θ]t[i][)][||][2] |
|
|
|
|
|
B−1 |
|
|
|
δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][,][ 1]B |
|
l1=0 |
|
|
|
X |
|
|
|
|
|
B−1 |
|
|
|
δt,l2 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗] [(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩] |
|
l2=0 |
|
|
|
X |
|
|
|
|
|
= ⟨ [1] |
|
|
|
B |
|
|
|
= [1] |
|
|
|
B[2] |
|
|
|
|
|
B−1 |
|
|
|
l1=0 |
|
|
|
X |
|
|
|
B−1 |
|
|
|
|
|
B−1 |
|
|
|
⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗] [(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][, δ][t,l][2] [(][w]t[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗] [(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩] |
|
l2=0 |
|
|
|
X |
|
|
|
|
|
= [1] |
|
|
|
B[2] |
|
|
|
|
|
||δt,l(wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][||][2] |
|
l=0 |
|
|
|
X |
|
|
|
|
|
+ [1] |
|
|
|
B[2] |
|
|
|
|
|
⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][, δ][t,l][2] [(][w]t[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]t,l[i] 2 [(][s, a][i][)]][⟩][.] |
|
l1̸=l2 |
|
|
|
X |
|
|
|
|
|
Taking expectation over the filtration Ft, we have |
|
|
|
E[||h[i]t[(][w]t[∗][)][ −] [g][i][(][w]t[∗][, θ]t[i][)][||][2][|F][t][]] |
|
|
|
|
|
B−1 |
|
|
|
E ||δt,l(wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][||][2][|F][t] |
|
l=0 |
|
|
|
X h |
|
|
|
|
|
= [1] |
|
|
|
B[2] |
|
|
|
|
|
+ [1] |
|
|
|
B[2] |
|
|
|
|
|
E ⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][, δ][t,l][2] [(][w]t[∗][)][ψ]θt[i] [−] [E][s,a][[][A][w]t[∗] [(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩|F][t] |
|
l1̸=l2 |
|
|
|
X h |
|
|
|
|
|
≤ [16] |
|
|
|
B [(][r][max][ +][ R][w][)][2] |
|
|
|
|
|
+ [1] |
|
|
|
B[2] |
|
|
|
|
|
E ⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][, δ][t,l][2] [(][w]t[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩|F][t] |
|
l1̸=l2 |
|
|
|
X h |
|
|
|
|
|
----- |
|
|
|
where the inequality follows from triangle inequality and the facts that |δt,l1 (wt[∗][)][ψ]θt[i] [| ≤] [2][r][max][ +] |
|
2Rw and |Es,a[Advwt[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][|][ =][ |][E][[][δ][t,l][1] [(][w]t[∗][)][ψ]θt[i] []][| ≤] [2][r][max][ + 2][R][w][. WLOG, for the] |
|
following term, we suppose l1 < l2. Then we have |
|
|
|
E ⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][, δ][t,l][2] [(][w]t[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩|F][t] |
|
h i |
|
|
|
=E E ⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] [i] [(][s, a][i][)]][, δ][t,l]2 [(][w]t[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩|F][t,l][1] |Ft |
|
h h i i |
|
|
|
=E ⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] [i] [(][s, a][i][)]][,][ E] δt,l2(wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][|F][t,l][1] ⟩|Ft |
|
h h i i |
|
|
|
=E ⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] [i] [(][s, a][i][)]][,][ E][[][δ][t,l]2 [(][w]t[∗][)][ψ]θt[i] [|F][t,l][1] []][ −] [E][s,a][[][Adv][w]t[∗] [(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩|F][t] |
|
h i |
|
|
|
=E ⟨δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] [i] [(][s, a][i][)]][,][ E][[][Adv][w]t[∗] [(][s][t,l][2] [, a][t,l][2] [)][ψ][θ]t[i] [|F][t,l][1] []][ −] [E][s,a][[][Adv][w]t[∗] [(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][⟩|F][t] |
|
h i |
|
|
|
≤E ||δt,l1 (wt[∗][)][ψ]θt[i] [−] [E][s,a][[][Adv][w]t[∗] [(][s, a][)][ψ]θ[i] [i] [(][s, a][i][)]][|| · ||][E][[][Adv][w]t[∗] [(][s][t,l][2] [, a][t,l][2] [)][ψ][θ]t[i] [|F][t,l][1] []][ −] [E][s,a][[][Adv][w]t[∗] [(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][|||F][t] |
|
h |
|
|
|
≤2(2rmax + 2Rw)E ||E[Advwt[∗][(][s][t,l][2] [, a][t,l][2] [)][ψ][θ]t[i] [|F][t,l][1][]][ −] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][|||F][t] |
|
h i |
|
|
|
≤16(rmax + Rw)[2]κρ[l][2][−][l][1] |
|
|
|
where the last inequality follows from |
|
|
|
||E[Advwt[∗][(][s][t,l][2] [, a][t,l][2] [)][ψ][θ]t[i] [|F][t,l][1][]][ −] [E][s,a][[][Adv][w]t[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)]][||] |
|
|
|
|
|
=|| |
|
|
|
|
|
Advwt[∗][(][s][t,l][2] [, a][t,l][2] [)][ψ][θ]t[i] [(][s][t,l][2] [, a][t,l][2] [)][P] [(][s][t,l][2] [, a][t,l][2] [|F][t,l][1] [)][ −] |
|
(st,l2,at,l2 ) |
|
|
|
X |
|
|
|
|
|
Advwt[∗][(][s, a][)][ψ][θ]t[i] [(][s, a][)][ν][θ][t] [(][s, a][)][||] |
|
(s,a) |
|
|
|
X |
|
|
|
|
|
≤ ||Advwt[∗][(][s, a][)][ψ]θ[i] t[i] [(][s, a][i][)][|| · |][P][ l][2][−][l][1][(][s, a][|F][t,l][2] [)][ −] [ν][θ][t][(][s, a][)][|] |
|
|
|
s,a |
|
|
|
X |
|
|
|
≤2(2rmax + 2Rw) · ||P [l][2][−][l][1](s, a|Ft,l2 ) − ν(s, a)||T V |
|
≤4(rmax + Rw)κρ[l][2][−][l][1]. |
|
|
|
Then, we have |
|
|
|
E[||h[i]t[(][w]t[∗][)][ −] [g][i][(][w]t[∗][, θ]t[i][)][||][2][|F][t][]] |
|
|
|
≤ [1] |
|
|
|
B[2][ [16][B][(][r][max][ +][ R][w][)][2][ + 16(][r][max][ +][ R][w][)][2][κ] |
|
|
|
|
|
ρ[l][2][−][l][1]] |
|
|
|
l2̸=l1 |
|
|
|
X |
|
|
|
|
|
≤ [1] ] |
|
|
|
B[2][ [16][B][(][r][max][ +][ R][w][)][2][ + 32(][r][max]1[ +] −[ R]ρ[w][)][2][κρB] |
|
|
|
≤ [16(][r][max][ +][ R][w][)][2][[1 + (2][κ][ −] [1)][ρ][]] . (39) |
|
|
|
B(1 − ρ) |
|
|
|
|
|
Then, we have |
|
|
|
E[||vt(wt) −∇θJ(θt)||[2]] ≤24N |
|
|
|
|
|
||wt[i] [−] [w]t[∗][||][2][ + 6][κ][3][N][ 3][(1][ −] [η][N] [−][1][)][2][t][gossip] |
|
i=1 |
|
|
|
X |
|
|
|
|
|
+ 24ξapprox[critic] [+ 96(][r][max][ +][ R][w][)][2][[1 + (2][κ][ −] [1)][ρ][]] N (40) |
|
|
|
B(1 − ρ) |
|
|
|
As a result, substituting equation equation 40 into equation 33 and taking expectation over Ft on |
|
both sides, we have |
|
|
|
|
|
( [1] |
|
|
|
2 [α][ −] [L][J] [α][2][)][E][[][||∇][θ][J][(][θ][t][)][||][2][]] |
|
|
|
≤ E[J(θt+1)] − E[J(θt)] + ( [1] |
|
|
|
2 [α][ +][ L][J] [α][2][)] |
|
|
|
|
|
||wt[i] [−] [w]t[∗][||][2] |
|
i=1 |
|
|
|
X |
|
|
|
|
|
24N |
|
|
|
|
|
----- |
|
|
|
+6κ3N [3](1 − η[N] [−][1])[2][t][gossip] + 24ξapprox[critic] [+ 96(][r][max][ +][ R][w][)][2][[1 + (2][κ][ −] [1)][ρ][]] N . (41) |
|
|
|
B(1 − ρ) |
|
|
|
|
|
|
|
By considering step size α = |
|
|
|
|
|
1 |
|
|
|
4LJ [, and dividing both sides of previous equation by] |
|
|
|
|
|
1 |
|
|
|
16LJ [, we have] |
|
|
|
|
|
E[||∇θJ(θt)||[2]] ≤16LJ (E[J(θt+1)] − E[J(θt)]) + 72N |
|
|
|
|
|
||wt[i] [−] [w]t[∗][||][2] |
|
i=1 |
|
|
|
X |
|
|
|
|
|
+ 18κ3N [3](1 − η[N] [−][1])[2][t][gossip] + 72ξapprox[critic] [+ 288][N][ (][r][max][ +][ R][w][)][2][[1 + (2][κ][ −] [1)][ρ][]] |
|
|
|
B(1 − ρ) |
|
|
|
(42) |
|
|
|
Let T[ˆ] be a random variable that takes value uniformly among {1, · · ·, T }. Taking summation over |
|
t = {1, · · ·, T } and dividing by T, we have |
|
|
|
|
|
E[||∇θJ(θ ˆT [)][||][2][] = 1] |
|
|
|
T |
|
|
|
|
|
E[||∇θJ(θt)||[2]] |
|
t=1 |
|
|
|
X |
|
|
|
|
|
T N |
|
t=1 i=1 [||][w]t[i] [−] [w]t[∗][||][2] |
|
|
|
≤ [16][L][J] [(][E][[][J][(][θ][T][ )]][ −] [E][[][J][(][θ][0][)])] + 72N |
|
|
|
T T |
|
P P |
|
|
|
+ 18κ3N [3](1 − η[N] [−][1])[2][t][gossip] + 72ξapprox[critic] [+ 288(][r][max][ +][ R][w][)][2][[1 + (2][κ][ −] [1)][ρ][]] N |
|
|
|
B(1 − ρ) |
|
|
|
N |
|
i=1 [||][w]t[i] [−] [w]t[∗][||][2] |
|
|
|
≤ [16][L][J] [E][[][J][(][θ][T][ )]] + 72N |
|
|
|
T T |
|
P |
|
|
|
+ 18κ3N [3](1 − η[N] [−][1])[2][t][gossip] + 72ξapprox[critic] [+ 288(][r][max][ +][ R][w][)][2][[1 + (2][κ][ −] [1)][ρ][]] N. |
|
|
|
B(1 − ρ) |
|
|
|
|
|
----- |
|
|
|
|