Title: LEARNING TO SCHEDULE COMMUNICATION IN MULTI-AGENT REINFORCEMENT LEARNING

Abstract: Many real-world reinforcement learning tasks require multiple agents to make sequential decisions under the agents' interaction, where well-coordinated actions among the agents are crucial to achieve the target goal better at these tasks. One way to accelerate the coordination effect is to enable multiple agents to communicate with each other in a distributed manner and behave as a group. In this paper, we study a practical scenario when (i) the communication bandwidth is limited and (ii) the agents share the communication medium so that only a restricted number of agents are able to simultaneously use the medium, as in the state-of-the-art wireless networking standards. This calls for a certain form of communication scheduling. In that regard, we propose a multi-agent deep reinforcement learning framework, called SchedNet, in which agents learn how to schedule themselves, how to encode the messages, and how to select actions based on received messages. SchedNet is capable of deciding which agents should be entitled to broadcasting their (encoded) messages, by learning the importance of each agent's partially observed information. We evaluate SchedNet against multiple baselines under two different applications, namely, cooperative communication and navigation, and predator-prey. Our experiments show a non-negligible performance gap between SchedNet and other mechanisms such as the ones without communication and with vanilla scheduling methods, e.g., round robin, ranging from 32% to 43%.

1. INTRODUCTION:

Reinforcement Learning (RL) has garnered renewed interest in recent years. Playing the game of Go (Mnih et al., 2015) , robotics control (Gu et al., 2017; Lillicrap et al., 2015) , and adaptive video streaming (Mao et al., 2017) constitute just a few of the vast range of RL applications. Combined with developments in deep learning, deep reinforcement learning (Deep RL) has emerged as an accelerator in related fields. From the well-known success in single-agent deep reinforcement learning, such as Mnih et al. (2015) , we now witness growing interest in its multi-agent extension, the multi-agent reinforcement learning (MARL), exemplified in Gupta et al. (2017) ; Lowe et al. (2017) ; Foerster et al. (2017a) ; Omidshafiei et al. (2017) ; Foerster et al. (2016) ; Sukhbaatar et al. (2016) ; Mordatch & Abbeel (2017) ; Havrylov & Titov (2017) ; Palmer et al. (2017) ; Peng et al. (2017) ; Foerster et al. (2017c) ; Tampuu et al. (2017) ; Leibo et al. (2017) ; Foerster et al. (2017b) . In the MARL problem commonly addressed in these works, multiple agents interact in a single environment repeatedly and improve their policy iteratively by learning from observations to achieve a common goal. Of particular interest is the distinction between two lines of research: one fostering the direct communication among agents themselves, as in Foerster et al. (2016) ; Sukhbaatar et al. (2016) and the other coordinating their cooperative behavior without direct communication, as in Foerster et al. (2017b) ; Palmer et al. (2017) ; Leibo et al. (2017) .

In this work, we concern ourselves with the former. We consider MARL scenarios wherein the task at hand is of a cooperative nature and agents are situated in a partially observable environment, but each endowed with different observation power. We formulate this scenario into a multi-agent sequential decision-making problem, such that all agents share the goal of maximizing the same discounted sum of rewards. For the agents to directly communicate with each other and behave as a coordinated group rather than merely coexisting individuals, they must carefully determine the information they exchange under a practical bandwidth-limited environment and/or in the case of high-communication cost. To coordinate this exchange of messages, we adopt the centralized training and distributed execution paradigm popularized in recent works, e.g., Foerster et al. (2017a) ; Lowe et al. (2017) ; Sunehag et al. (2018) ; Rashid et al. (2018) ; Gupta et al. (2017) .

In addition to bandwidth-related constraints, we take the issues of sharing the communication medium into consideration, especially when agents communicate over wireless channels. The stateof-the-art standards on wireless communication such as Wi-Fi and LTE specify the way of scheduling users as one of the basic functions. However, as elaborated in Related work, MARL problems involving scheduling of only a restricted set of agents have not yet been extensively studied. The key challenges in this problem are: (i) that limited bandwidth implies that agents must exchange succinct information: something concise and yet meaningful and (ii) that the shared medium means that potential contenders must be appropriately arbitrated for proper collision avoidance, necessitating a certain form of communication scheduling, popularly referred to as MAC (Medium Access Control) in the area of wireless communication. While stressing the coupled nature of the encoding/decoding and the scheduling issue, we zero in on the said communication channel-based concerns and construct our neural network accordingly.

Contributions In this paper, we propose a new deep multi-agent reinforcement learning architecture, called SchedNet, with the rationale of centralized training and distributed execution in order to achieve a common goal better via decentralized cooperation. During distributed execution, agents are allowed to communicate over wireless channels where messages are broadcast to all agents in each agent's communication range. This broadcasting feature of wireless communication necessitates a Medium Access Control (MAC) protocol to arbitrate contending communicators in a shared medium. CSMA (Collision Sense Multiple Access) in Wi-Fi is one such MAC protocol. While prior work on MARL to date considers only the limited bandwidth constraint, we additionally address the shared medium contention issue in what we believe is the first work of its kind: which nodes are granted access to the shared medium. Intuitively, nodes with more important observations should be chosen, for which we adopt a simple yet powerful mechanism called weight-based scheduler (WSA), designed to reconcile simplicity in training with integrity of reflecting real-world MAC protocols in use (e.g., . We evaluate SchedNet for two applications: cooperative communication and navigation and predator/prey and demonstrate that SchedNet outperforms other baseline mechanisms such as the one without any communication or with a simple scheduling mechanism such as round robin. We comment that SchedNet is not intended for competing with other algorithms for cooperative multi-agent tasks without considering scheduling, but a complementary one. We believe that adding our idea of agent scheduling makes those algorithms much more practical and valuable.

Related work:

We now discuss the body of relevant literature. Busoniu et al. (2008) and Tan (1993) have studied MARL with decentralized execution extensively. However, these are based on tabular methods so that they are restricted to simple environments. Combined with developments in deep learning, deep MARL algorithms have emerged (Tampuu et al., 2017; Foerster et al., 2017a; Lowe et al., 2017) . Tampuu et al. (2017) uses a combination of DQN with independent Q-learning. This independent learning does not perform well because each agent considers the others as a part of environment and ignores them. It is widely accepted that communication can further enhance the collective intelligence of learning agents in their attempt to complete cooperative tasks. To this end, a number of papers have previously studied the learning of communication protocols and languages to use among multiple agents in reinforcement learning. We explore those bearing the closest resemblance to our research. Foerster et al. (2016) ; Sukhbaatar et al. (2016) ; Peng et al. (2017) ; Guestrin et al. (2002) , and Zhang & Lesser (2013) train multiple agents to learn a communication protocol, and have shown that communicating agents achieve better rewards at various tasks. Mordatch & Abbeel (2017) and Havrylov & Titov (2017) investigate the possibility of the artificial emergence of language. Coordinated RL by Guestrin et al. (2002) is an earlier work demonstrating the feasibility of structured communication and the agents' selection of jointly optimal action.

Only DIAL (Foerster et al., 2016) and Zhang & Lesser (2013) explicitly address bandwidth-related concerns. In DIAL, the communication channel of the training environment has a limited bandwidth, such that the agents being trained are urged to establish more resource-efficient communication protocols. The environment in Zhang & Lesser (2013) also has a limited-bandwidth channel in effect, due to the large amount of exchanged information in running a distributed constraint optimization algorithm. Recently, Jiang & Lu (2018) proposes an attentional communication model that allows some agents who request additional information from others to gather observation from neighboring agents. However, they do not explicitly consider the constraints imposed by limited communication bandwidth and/or scheduling due to communication over a shared medium.

To the best of our knowledge, there is no prior work that incorporates an intelligent scheduling entity in order to facilitate inter-agent communication in both a limited-bandwidth and shared medium access scenarios. As outlined in the introduction, intelligent scheduling among learning agents is pivotal in the orchestration of their communication to better utilize the limited available bandwidth as well as in the arbitration of agents contending for shared medium access.

2. BACKGROUND:

Reinforcement Learning We consider a standard RL formulation based on Markov Decision Process (MDP). An MDP is a tuple < S, A, r, P, γ > where S and A are the sets of states and actions, respectively, and γ ∈ [0, 1] is the discount factor. A transition probability function P : S × A → S maps states and actions to a probability distribution over next states, and r : S × A → R denotes the reward. The goal of RL is to learn a policy π : S → A that solves the MDP by maximizing the expected discounted return

R t = E[ ∞ k=0 γ k r t+k |π]. The policy induces a value function V π (s) = E π [R t |s t = s], and an action value function Q π (s, a) = E π [R t |s t = s, a t = a].

Actor-critic Method The main idea of the policy gradient method is to optimize the policy, parametrized by θ π , in order to maximize the objective J(θ) = E s∼p π ,a∼π θ [R] by directly adjusting the parameters in the direction of the gradient. By the policy gradient theorem Sutton et al. (2000) , the gradient of the objective is:

EQUATION

where ρ π is the state distribution. Our baseline algorithmic framework is the actor-critic approach Konda & Tsitsiklis (2003) . In this approach, an actor adjusts the parameters θ of the policy π θ (s) by gradient ascent. Instead of the unknown true action-value function Q π (s, a), its approximated version Q w (s, a) is used with parameter w. A critic estimates the action-value function Q w (s, a) using an appropriate policy evaluation algorithm such as temporal-difference learning Tesauro (1995) . To reduce the variance of the gradient updates, some baseline function b(s) is often subtracted from the action value, thereby resulting in Q π (s, a) -b(s) Sutton & Barto (1998). A popular choice for this baseline function is the state value V (s), which indicates the inherent "goodness" of the state. This difference between the action value and the state value is often dubbed as the advantage A(s, a) whose TD-error-based substitute δ t = r t + γV (s t+1 ) -V (s t ) is an unbiased estimate of the advantage as in Mnih et al. (2016) . The actor-critic algorithm can also be applied to a deterministic policy µ θ : S → A. By the deterministic policy gradient theorem Silver et al. (2014) , we update the parameters as follows:

EQUATION

MARL: Centralized Critic and Distributed Actor (CCDA) We formalize MARL using DEC-POMDP (Oliehoek et al., 2016) , which is a generalization of MDP to allow a distributed control by multiple agents who may be incapable of observing the global state. A DEC-POMDP is described by a tuple < S, A, r, P, Ω, O, γ >. We use bold-face fonts in some notations to highlight the context of multi-agents. Each agent i ∈ N chooses an action a i ∈ A, forming a joint action vector a = [a i ] ∈ A n and has partial observations o i ∈ Ω according to some observation function O(s, i) : S × N → Ω. P (s |s, a) : S × A n → [0, 1] is the transition probability function. All agents share the same reward r(s, u) : S × A n → R. Each agent i takes action a i based on its own policy π i (a i |o i ). As mentioned in Section 1, our particular focus is on the centralized training and distributed execution paradigm, where the actor-critic approach is a good fit to such a paradigm. Since the agents should execute in a distributed setting, each agent, say i, maintains its own actor that selects i's action based only on what is partially observed by i. The critic is naturally responsible for centralized training, and thus works in a centralized manner. Thus, the critic is allowed to have the global state s as its input, which includes all agents' observations and extra information from the environment. The role of the critic is to "criticize" individual agent's actions. This centralized nature of the critic helps in providing more accurate feedback to the individual actors with limited observation horizon. In this case, each agent's policy, π i , is updated by a variant of (1) as:

∇ θ J(π i θ ) = E s∼ρ π ,a∼π θ [∇ θ log π i θ (a i |o i )(r + γV (s t+1 ) -V (s t ))].

(3)

3. METHOD:



3.1. COMMUNICATION ENVIRONMENT AND PROBLEM:

In practical scenarios where agents are typically separated but are able to communicate over a shared medium, e.g., a frequency channel in wireless communications, two important constraints are imposed: bandwidth and contention for medium access (Rappaport, 2001) . The bandwidth constraint entails a limited amount of bits per unit time, and the contention constraint involves having to avoid collision among multiple transmissions due to the natural aspect of signal broadcasting in wireless communication. Thus, only a restricted number of agents are allowed to transmit their messages each time step for a reliable message transfer. In this paper, we use a simple model to incorporate that the aggregate information size per time step is limited by L band bits and that only K sched out of n agents may broadcast their messages.

Weight-based Scheduling Noting that distributed execution of agents is of significant importance, there may exist a variety of scheduling mechanisms to schedule K sched agents in a distributed manner. In this paper, we adopt a simple algorithm that is weight-based, which we call WSA (Weightbased Scheduling Algorithm). Once each agent decides its own weight, the agents are scheduled based on their weights following a class of the pre-defined rules. We consider the following two specific ones among many different proposals due to simplicity, but more importantly, good approximation of wireless scheduling protocols in practice.

• Top(k). Selecting top k agents in terms of their weight values.

• Softmax(k). Computing softmax values σ i (w) = e w i n j=1 e w j for each agent i, and then randomly selecting k agents acoording to the probability distribution [σ i (w)] n i=1 . Since distributed execution is one of our major operational constraints in SchedNet or other CTDEbased MARL algorithms, Top(k) and Softmax(k) should be realizable via a weight-based mechanism in a distributed manner. In fact, this has been an active research topic to date in wireless networking, where many algorithms exist (Tassiulas & Ephremides, 1992; Yi et al., 2008; Jiang & Walrand, 2010) . Due to space limitation, we present how to obtain distributed versions of those two rules based on weights in our supplementary material. To summarize, using so-called CSMA (Carrier Sense Multiple Access) (Kurose, 2005), which is a fully distributed MAC scheduler and forms a basis of Wi-Fi, given agents' weight values, it is possible to implement Top(k) and Softmax(k).

Our goal is to train agents so that every time each agent takes an action, only K sched agents can broadcast their messages with limited size L band with the goal of receiving the highest cumulative reward via cooperation. Each agent should determine a policy described by its scheduling weights, encoded communication messages, and actions.

3.2. ARCHITECTURE:

To this end, we propose a new deep MARL framework with scheduled communications, called SchedNet, whose overall architecture is depicted in Figure 1 . SchedNet consists of the following three components: (i) actor network, (ii) scheduler, and (iii) critic network. This section is devoted to presenting the architecture only, whose details are presented in the subsequent sections.

Neural networks:

The actor network is the collection of n per-agent individual actor networks, where each agent i's individual actor network consists of a triple of the following networks: a

𝑜 3 𝑜 2 𝑚 3 WSA (𝑘) WG 1 𝑚 1 ENC 1 𝑢 1 𝒐 AS 1 𝒘 (𝑜 1 , 𝒎) Update Critic Actor 𝑜 1 Scheduler 𝒄 Environment 𝒔 𝒎 = 𝑚 𝑖 𝑛 ⊗ 𝒄 𝑤 1 WG 2 WG 3 𝑜 2 𝑤 2 𝑤 3 𝑜 1 ENC 2 ENC 3 𝑜 3 (𝑜 2 , 𝒎) (𝑜 3 , 𝒎) AS 2 AS 3 𝑢 2 𝑢 3

Update:

Figure 1: Architecture of SchedNet with three agents.

Agents 1 and 3 have been scheduled for this time step.

message encoder, an action selector, and a weight generator, as specified by:

message encoder f i enc : o i → m i , action selector f i as : (o i , m ⊗ c) → u i , weight generator f i wg : o i → w i .

Here

, m = [m i ] n 1 is the vector of each i's encoded message m i . An agent schedule vector c = [c i ] n , c i ∈ {0, 1} represents whether each agent is scheduled. Note that agent i's encoded message m i is generated by a neural network f i enc : o i → m i .

The operator "⊗" concatenates all the scheduled agents' messages. For example, for m = [010, 111, 101] and c = [110], m ⊗ c = 010111. This concatenation with the schedule profile c means that only those agents scheduled in c may broadcast their messages to all other agents. We denote by θ i as , θ i wg , and θ i enc the parameters of the action selector, the weight generator, and the encoder of agent i, respectively, where we let θ as = [θ i as ] n , and similarly define θ wg and θ enc . Coupling: Actor and Scheduler Encoder, weight generator and the scheduler are the modules for handling the constraints of limited bandwidth and shared medium access. Their common goal is to learn the state-dependent "importance" of individual agent's observation, encoders for generating compressed messages and the scheduler for being used as a basis of an external scheduling mechanism based on the weights generated by per-agent weight generators. These three modules work together to smartly respond to time-varying states. The action selector is trained to decode the incoming message, and consequently, to take a good action for maximizing the reward. At every time step, the schedule profile c varies depending on the observation of each agent, so the incoming message m comes from a different combination of agents. Since the agents can be heterogeneous and they have their own encoder, the action selector must be able to make sense of incoming messages from different senders. However, the weight generator's policy changes, the distribution of incoming messages also changes, which is in turn affected by the pre-defined WSA. Thus, the action selector should adjust to this changed scheduling. This also affects the encoder in turn. The updates of the encoder and the action selector trigger the update of the scheduler again. Hence, weight generators, message encoders, and action selectors are strongly coupled with dependence on a specific WSA, and we train those three networks at the same time with a common critic.

Scheduling logic:

The schedule profile c is determined by the WSA module, which is mathematically a mapping from all agents' weights w (generated by f i wg ) to c. Typical examples of these mappings are Top(k) and Softmax(k), as mentioned above. The scheduler of each agent is trained appropriately depending on the employed WSA algorithm. In the centralized training with distributed execution, for a given WSA, we include all components and modules in Figure 1 to search for θ as , θ wg , and θ enc , whereas in execution, each agent i runs a certain shared medium access mechanism, well-modeled by a weightbased scheduler, and just needs three agent-specific parameters θ i as , θ i wg , and θ i enc .

3.3. TRAINING AND EXECUTION:



𝑤 Ԧ 𝑆:



3.3.1. CENTRALIZED TRAINING:



Centralized critic:

The actor is trained by dividing it into two parts: (i) message encoders and action selectors, and (ii) weight generators. This partitioning is motivated by the fact that it is hard to update both parts with one backpropagation since WSA is not differentiable. To update the actor, we use a centralized critic parametrized by θ c to estimate the state value function V θc (s) for the action selectors and message encoders, and the action-value function Q π θc (s, w) for the weight generators. The critic is used only when training, and it can use the global state s, which includes the observation of all agents. All networks in the actor are trained with gradient-based on temporal difference backups. To share common features between V θc (s) and Q π θc (s, w) and perform efficient training, we use shared parameters in the lower layers of the neural network between the two functions, as shown in Figure 2 .

Weight generators We consider the collection of all agents' WGs as a single neural network µ θwg (o) mapping from o to w, parametrized by θ wg . Noting that w i is a continuous value, we apply the DDPG algorithm (Lillicrap et al., 2015) , where the entire policy gradient of the collection of WGs is given by:

∇ θwg J(θ wg , •) = E w∼µ θwg [∇ θwg µ θwg (o)∇ w Q θc (s, w)| w=µ θwg (o) ].

We sample the policy gradient for sufficient amount of experience in the set of all scheduling profiles, i.e., C = {c | ci ≤ k}. The values of Q θc (s, w) are estimated by the centralized critic, where s is the global state corresponding to o in a sample.

Message encoders and action selectors:

The observation of each agent travels through the encoder and the action selector. We thus serialize f i enc and f i as together and merge the encoders and actions selectors of all agents into one aggregate network π θu (u|o, c), which is parametrized by θ u = {θ enc , θ as }. This aggregate network π θu learns via backpropagation of actor-critic policy gradients, described below. The gradient of this objective function, which is a variant of (3), is given by

EQUATION

where s and s are the global states corresponding to the observations at current and next time step. We can get the value of state V θc (s) from the centralized critic and then adjust the parameters θ u via gradient ascent accordingly.

3.3.2. DISTRIBUTED EXECUTION:

In execution, each agent i should be able to determine the scheduling weight w i , encoded message m i , and action selection u i in a distributed manner. This process must be based on its own observation, and the weights generated by its own action selector, message encoder, and weight generator with the parameters θ i as , θ i enc , and θ i wg , respectively. After each agent determines its scheduling weight, K sched agents are scheduled by WSA, which leads the encoded messages of scheduled agents to be broadcast to all agents. Finally, each agent finally selects an action by using received messages. This process is sequentially repeated under different observations over time.

4. EXPERIMENT:

Environments To evaluate SchedNet2 , we consider two different environments for demonstrative purposes: Predator and Prey (PP) which is used in Stone & Veloso (2000) , and Cooperative Communication and Navigation (CCN) which is the simplified version of the one in Lowe et al. (2017) . The detailed experimental environments are elaborated in the following subsections as well as in supplementary material. We take the communication environment into our consideration as follows. k out of all agents can have the chance to broadcast the message whose bandwidth3 is limited by l.

Tested algorithms and setup:

We perform experiments in aforementioned environments. We compare SchedNet with a variant of DIAL,4 (Foerster et al., 2016) which allows communication with limited bandwidth. During the execution of DIAL, the limited number (k) of agents are scheduled following a simple round robin scheduling algorithm, and the agent reuses the outdated messages of non-scheduled agents to make a decision on the action to take, which is called DIAL(k). The other baselines are independent DQN (IDQN) (Tampuu et al., 2017) and COMA (Foerster et al., 2017a) in which no agent is allowed to communicate. To see the impact of scheduling in SchedNet, we compare SchedNet with (i) RR (round robin), which is a canonical scheduling method in communication systems where all agents are sequentially scheduled, and (ii) FC (full communication), which is the ideal configuration, wherein all the agents can send their messages without any scheduling or bandwidth constraints. We also diversify the WSA in SchedNet into: (i) Sched-Softmax( 1) and (ii) Sched-Top(1) whose details are in Section 3.1. We train our models until convergence, and then evaluate them by averaging metrics for 1,000 iterations. The shaded area in each plot denotes 95% confidence intervals based on 6-10 runs with different seeds.

4.1. PREDATOR AND PREY:

In this task, there are multiple agents who must capture a randomly moving prey. Agents' observations include position of themselves and the relative positions of prey, if observed. We employ four agents, and they have different observation horizons, where only agent 1 has a 5 × 5 view while agents 2, 3, and 4 have a smaller, 3 × 3 view. The predators are rewarded when they capture the prey, and thus the performance metric is the number of time steps taken to capture the prey.

Result in PP Figure 3a illustrates the learning curve of 750,000 steps in PP. In FC, since the agents can use full state information even during execution, they achieve the best performance. SchedNet outperforms IDQN and COMA in which communication is not allowed. It is observed that agents first find the prey, and then follow it until all other agents also eventually observe the prey. An agent successfully learns to follow the prey after it observes the prey but that it takes a long time to meet the prey for the first time. If the agent broadcasts a message that includes the location information of the prey, then other agents can find the prey more quickly. Thus, it is natural that SchedNet and DIAL perform better than IDQN or COMA, because they are trained to work with communication. However, DIAL is not trained for working under medium contention constraints.

Although DIAL works well when there is no contention constraints, under the condition where only one agent is scheduled to broadcast the message by a simple scheduling algorithm (i.e., RR), the average number of steps to capture the prey in DIAL(1) is larger than that of SchedNet-Top(1), because the outdated messages of non-scheduled agents is noisy for the agents to decide on actions. Thus, we should consider the scheduling from when we train the agents to make them work in a demanding environment.

Impact of intelligent scheduling:

In Figure 3b , we observe that IDQN, RR, and SchedNet-Softmax(1) lie more or less on a comparable performance tier, with SchedNet-Softmax(1) as the best in the tier. SchedNet-Top(1) demonstrates a non-negligible gap better than the said tier, implying that a deterministic selection improves the agents' collective rewards the best. In particular, SchedNet-Top(1) improves the performance by 43% compared to RR. Figure 3b lets us infer that, while all the agents are trained under the same conditions except for the scheduler, the difference in the scheduler is the sole determining factor for the variation in the performance levels. Thus, ablating away the benefit from smart encoding, the intelligent scheduling element in SchedNet can be accredited with the better performance.

Weight-based Scheduling We attempt to explain the internal behavior of SchedNet by investigating instances of temporal scheduling profiles obtained during the execution. We observe that SchedNet has learned to schedule those agents with a farther observation horizon, realizing the rationale of importance-based assignment of scheduling priority also for the PP scenario. Recall that Agent 1 has a wider view and thus tends to obtain valuable observation more frequently. In Fig-

ure 4, we see that scheduling chances are distributed over (14, 3, 4, 4) where corresponding average weights are (0.74, 0.27, 0.26, 0.26), implying that those with greater observation power tend to be scheduled more often. ). This is because the agent should transfer more informative messages that implicitly include the location of the prey, when it observes the prey. Further analysis of the messages is presented in our supplementary material.

4.2. COOPERATIVE COMMUNICATION AND NAVIGATION:

In this task, each agent's goal is to arrive at a pre-specified destination on its one-dimensional world, and they collect a joint reward when both agents reach their respective destination. Each agent has a zero observation horizon around itself, but it can observe the situation of the other agent. We introduce heterogeneity into the scenario, where the agent-destination distance at the beginning of the task differs across agents. The metric used to gauge the performance is the number of time steps taken to complete the CCN task.

Result in CCN We examine the CCN environment whose results are shown in Figure 3c . Sched-Net and other baselines were trained for 200,000 steps. As expected, IDQN takes the longest time, and FC takes the shortest time. RR exhibits mediocre performance, better than IDQN, because agents at least take turns in obtaining the communication opportunity. Of particular interest is Sched-Net, outperforming both IDQN and RR with a non-negligible gap. We remark that the deterministic selection with SchedNet-Top(1) slightly beats the probabilistic counterpart, SchedNet-Softmax(1). The 32% improved gap between RR and SchedNet clearly portrays the effects of intelligent scheduling, as the carefully learned scheduling method of SchedNet was shown to complete the CCN task faster than the simplistic RR. Scheduling in CCN As Agent 2 is farther from its destination than Agent 1, we observe that Agent 1 is scheduled more frequently to drive Agent 2 to its destination (7 vs. 18), as shown in Figure 6 . This evidences that SchedNet flexibly adapts to heterogeneity of agents via scheduling. Towards more efficient completion of the task, a rationale of more scheduling for more important agents should be implemented. This is in accordance with the results obtained from PP environments: more important agents are scheduled more.

5. CONCLUSION:

We have proposed SchedNet for learning to schedule inter-agent communications in fullycooperative multi-agent tasks. In SchedNet, we have the centralized critic giving feedback to the actor, which consists of message encoders, action selectors, and weight generators of each individual agent. The message encoders and action selectors are criticized towards compressing observations more efficiently and selecting actions that are more rewarding in view of the cooperative task at hand. Meanwhile, the weight generators are criticized such that k agents with apparently more valuable observation are allowed to access the shared medium and broadcast their messages to all other agents. Empirical results and an accompanying ablation study indicate that the learnt encoding and scheduling behavior each significantly improve the agents' performance. We have observed that an intelligent, distributed communication scheduling can aid in a more efficient, coordinated, and rewarding behavior of learning agents in the MARL setting.

SUPPLEMENTARY MATERIAL A SCHEDNET TRAINING ALGORITHM:

The training algorithm for SchedNet is provided in Algorithm 1. The parameters of the message encoder are assumed to be included in the actor network. Thus, we use the notation for t = 1 to T do 6:

f i as (o i , c) = f i as (o i , f i enc (o i ) ⊗ c)

w t ← the priority w i = f i wg (o i ) of each agent i 7:

Get schedule c t from w t 8: Set

u t ← the action u i = f i as (o i , c t )

y k = r k + γ V (s k+1 ) 13: Set ŷk = r k + γ Q(s k+1 , f i wg (o k+1 , c k+1 )) 14:

Update the critic by minimizing the loss:

L = 1 S k ((y k -V (s k )) 2 + (ŷ k -Q(s, w k )) 2 ) 15:

Update the actor along with the encoder using sampled policy gradient:

∇ θu J(•, θ u ) = E s∼ρ π ,u∼π [∇ θu log π(u|o, c)[r + γV θc (s ) -V θc (s)]]

16:

Update scheduler using sampled policy gradient:

∇ θwg J(θ wg , •) = E w∼µ [∇ θwg µ(o)∇ w Q θc (s, w)| w=µ(o) ] 17:

Update target network parameters:

θ wg ← τ θ wg + (1 -τ )θ wg θ c ← τ θ c + (1 -τ )θ c 18:

end for 19: end for Predator and prey We assess SchedNet in this predator-prey setting as in Stone & Veloso (2000) , illustrated in Figure 7a . This setting involves a discretized grid world and multiple cooperating predators who must capture a randomly moving prey. Agents' observations include position of themselves and the relative positions of the prey, if observed. The observation horizon of each predator is limited, thereby emphasizing the need for communication. The termination criterion for the task is that all agents observe the prey, as in the right of Figure 7a . The predators are rewarded when the task is terminated. We note that agents may be endowed with different observation horizons, making them heterogeneous. We employ four agents in our experiment, where only agent 1 has a 5 × 5 view while agents 2, 3, and 4 have a smaller, 3 × 3 view. The performance metric is the number of time steps taken to capture the prey.

B DETAILS OF ENVIRONMENTS AND IMPLEMENTATION:



Cooperative communication and navigation:

We adopt and modify the cooperative communication and navigation task in Lowe et al. (2017) , where we test SchedNet in a simple one-dimensional grid as in Figure 7b . In CCN, each of the two agents resides in its one-dimensional grid world. Each agent's goal is to arrive at a pre-specified destination (denoted by the square with a star or a heart for Agents 1 and 2, respectively), and they collect a joint reward when both agents reach their target destination. Each agent has a zero observation horizon around itself, but it can observe the situation of the other agent. We introduce heterogeneity into the scenario, where the agent-destination distance at the beginning of the task differs across agents. In our example, Agent 2 is initially located at a farther place from its destination, as illustrated in Figure 7b . The metric used to gauge the performance of SchedNet is the number of time steps taken to complete the CCN task.

B.2 EXPERIMENT DETAILS:

Table 1 shows the values of the hyperparameters for the CCN and the PP task. We use Adam optimizer to update network parameters and soft target update to update target network. The structure of the networks is the same across tasks. For the critic, we used three hidden layers, and the critic between the scheduler and the action selector shares the first two layers. For the actor, we use one hidden layer; for the encoder and the weight generator, three hidden layers each. Networks use rectified linear units for all hidden layers. Because the complexity of the two tasks differ, we sized the hidden layers differently. The actor network and the critic network for the CCN have hidden layers with 8 units and 16 units, respectively. The actor network and the critic network for the PP have hidden layers with 32 units and 64 units, respectively. 

C ADDITIONAL EXPERIMENT RESULTS:

C.1 PREDATOR AND PREY Impact of bandwidth (L) and number of schedulable agents (K) Due to communication constraints, only k agents can communicate and scheduled agents can broadcast their message, each of which has a limited size l due to bandwidth constraints. We see the impact of l and k on the performance in Figure 8a . As L increases, more information can be encoded into the message, which can be used by other agents to take action. Since the encoder and the actor are trained to maximize the shared goal of all agents, they can achieve higher performance with increasing l. In Figure 8b , we compare the cases where k = 1, 2, 3, and FC in which all agents can access the medium, with l = 1.

As we can expect, the general tendency is that the performance grows as k increases. Impact of joint scheduling and encoding To study the effect of jointly coupling scheduling and encoding, we devise a comparison against a pre-trained auto-encoder (Bourlard & Kamp, 1988; Hinton & Zemel, 1994 ). An auto-encoder was trained ahead of time, and the encoder part of this auto-encoder was placed in the Actor's ENC module in Figure 1 . The encoder part is not trained further while training the other parts of network. Henceforth, we name this modified Actor "AE". Figure 8c shows the learning curve of AE and other baselines. Table 2 highlights the impact of joint scheduling and encoding. The numbers shown are the performance metric normalized to the FC case in the PP environment. While SchedNet-Top(1) took only 2.030 times as long as FC to finish the PP task, the AE-equipped actor took 3.408 times as long as FC. This lets us ascertain that utilizing a pre-trained auto-encoder deprives the agent of the benefit of joint the scheduler and encoder neural network in SchedNet.

What messages agents broadcast In Section 4.1, we attempted to understand what the predator agents communicate when performing PP task where k = 1 and l = 2. In this section, we look into the message in detail. Figure 9 shows the projections of the messages generated by the scheduled agent based on its own observation. In the PP task, the most important information is the location of the prey, and this can be estimated from the observation of other agents. Thus, we are interested in the location information of the prey and other agents. We classify the message into four classes based on which quadrant the prey and the predator are included, and mark each class with different colors.

Figure 9a shows the messages for different relative location of prey for agents' observation, and Figure 9b shows the messages for different locations of the agent who sends the message. We can observe that there is some general trend in the message according to the class. We thus conclude that if the agents observe the prey, they encode into the message the relevant information that is helpful to estimate the location of the prey. The agents who receive this message interpret the message to select action. 

C.2 PARTIAL OBSERVABILITY ISSUE IN SCHEDNET:

In MARL, partial observability issue is one of the major problems, and there are two typical ways to tackle this issue. First, using RNN structure to indirectly remember the history can alleviate the partial observability issues. Another way is to use the observations of other agents through communication among them. In this paper, we focused more on the latter because the goal of this paper is to show the importance of learning to schedule in a practical communication environment in which the shared medium contention is inevitable. Enlarging the observation through communication is somewhat orthogonal to considering temporal correlation. Thus, we can easily merge SchedNet with RNN which can be appropriate to some partially observable environments. We add one GRU layer into each of individual encoder, action selector, and weight generator of each agent, where each GRU cell has 64 hidden nodes.

Figure 10 shows the result of applying RNN. We implement IDQN with RNN, and the results show that the average steps to complete tasks of IDQN with RNN is slightly smaller than that of IDQN with feed-forward network. In this case, RNN helps to improve the performance by tackling the partial observable issue. On the other hand, SchedNet-RNN and SchedNet achieve similar performance. We think that the communication in SchedNet somewhat resolves the partial observable issues, so the impact of considering temporal correlation with RNN is relatively small. Although applying RNN to SchedNet is not really that helpful in this simple environment, we expect that in a more complex environment, using the recurrent connection is more helpful. In FC, since all agents can broadcast their message during execution, they achieve the best performance. IDQN and COMA in which no communication is allowed, take a longer time to complete the task compared to other baselines. The performances of both are similar because no cooperation can be achieved without the exchange of observations in this environment. As expected, SchedNet and DIAL outperform IDQN and COMA. Although DIAL works well when there is no contention constraint, under the contention constraint, the average number of steps to complete the task in DIAL(1) is larger than that of SchedNet-Top(1). This result shows the same tendency with the result in PP environment.

D SCHEDULER FOR DISTRIBUTED EXECUTION:

Issues. The role of the scheduler is to consider the constraint due to accessing a shared medium, so that only k < n agents may broadcast their encoded messages. k is determined by the wireless communication environment. For example, under a single wireless channel environment where each agent is located in other agents' interference range, k = 1. Although the number of agents that can be simultaneously scheduled is somewhat more complex, we abstract it with a single number k because the goal of this paper lies in studying the importance of considering scheduling constraints.

There are two key challenges in designing the scheduler: (i) how to schedule agents in a distributed manner for decentralized execution, and (ii) how to strike a good balance between simplicity in implementation and training, and the integrity of reflecting the current practice of MAC (Medium Access Control) protocols. Weight-based scheduling To tackle the challenges addressed in the previous paragraph, we propose a scheduler, called weight-based scheduler (WSA), that works based on each agent's individual weight coming from its observation. As shown in Figure 12 , the role of WSA is to map from w = [w i ] n to c. This scheduling is extremely simple, but more importantly, highly amenable to the philosophy of distributed execution. The remaining checkpoint is whether this principle is capable of efficiently approximating practical wireless scheduling protocols. To this end, we consider the following two weight-based scheduling algorithms among many different protocols that could be devised:

• Top(k). Selecting top k agents in terms of their weight values.

• Softmax(k). Computing softmax values σ(w) i = e w i n j=1 e w j for each agent i, and then randomly selecting k agents with probability in proportion to their softmax values. Top(k) can be a nice abstraction of the MaxWeight (Tassiulas & Ephremides, 1992) scheduling principle or its distributed approximation (Yi et al., 2008) , in which case it is known that different choices of weight values result in achieving different performance metrics, e.g., using the amount of messages queued for being transmitted as weight. Softmax(k) can be a simplified model of CSMA (Carrier Sense Multiple Access), which forms a basis of 802.11 Wi-Fi. Due to space limitation, we refer the reader to Jiang & Walrand (2010) for detail. We now present how Top(k) and Softmax(k) work.

D.1 CARRIER SENSE MULTIPLE ACCESS (CSMA):

CSMA is the one of typical distributed MAC scheduling in wireless communication system. To show the feasibility of scheduling Top(k) and Softmax(k) in a distributed manner, we will explain the variant of CSMA. In this section, we first present the concept of CSMA.

How does CSMA work?:

The key idea of CSMA is "listen before transmit". Under a CSMA algorithm, prior to trying to transmit a packet, senders first check whether the medium is busy or idle, and then transmit the packet only when the medium is sensed as idle, i.e., no one is using the channel. To control the aggressiveness of such medium access, each sender maintains a backoff timer, which is set to a certain value based on a pre-defined rule. The timer runs only when the medium is idle, and stops otherwise. With the backoff timer, links try to avoid collisions by the following procedure:

• Each sender does not start transmission immediately when the medium is sensed idle, but keeps silent until its backoff timer expires. • After a sender grabs the channel, it holds the channel for some duration, called the holding time.

Depending on how to choose the backoff and holding times, there can be many variants of CSMA that work for various purposes such as fairness and throughput. Two examples of these, Top(k) and Softmax(k), are introduced in the following sections.

D.2 A VERSION OF Distributed Top(k):

In this subsection, we introduce a simple distributed scheduling algorithm, called Distributed Top(k), which can work with SchedNet-Top(k). It is based on CSMA where each sender determines backoff and holding times as follows. In SchedNet, each agent generates the scheduling weight w based on its own observation. The agent sets its backoff time as 1 -w where w is its schedule weight, and it waits for backoff time before it tries to broadcast its message. Once it successfully broadcasts the message, it immediately releases the channel. Thus, the agent with the highest w can grab the channel in a decentralized manner without any message passing. By repeating this for k times, we can realize decentralized Top(k) scheduling.

To show the feasibility of distributed scheduling, we implemented the Distributed Top(k) on Contiki network simulator (Dunkels et al., 2004) and run the trained agents for the PP task. In our experiment, Top(k) agents are successfully scheduled 98% of the time, and the 2% failures are due to probabilistic collisions in which one of the colliding agents is randomly scheduled by the default collision avoidance mechanism implemented in Contiki. In this case, agents achieve 98.9% performance compared to the case where Top(k) agents are ideally scheduled.

D.3 OCSMA ALGORITHM AND Softmax(k):

In this section, we explain the relation between Softmax(k) and the existing CSMA-based wireless MAC protocols, called oCSMA. When we use Softmax(k) in the case of k = 1, the scheduling algorithm directly relates to the channel selection probability of oCSMA algorithms. First, we explain how it works and show that the resulting channel access probability has a same form with Softmax(k).

How does oCSMA work? It is also based on the basic CSMA algorithm. Once each agent generates its scheduling weight w i , it sets b i and h i to satisfy w i = log(b i h i ). It sets its backoff and holding times following exponential distributions with means 1/b i and h i , respectively. Based on these backoff and holding times, each agent runs the oCSMA algorithm. In this case, if all agents are in the communication range, the probability that agent i is scheduled over time is as follows:

s i (w) = exp(w i ) n j=1 exp(w j ) .

We refer the readers to Jang et al. (2014) for detail.

:

We use[•]n to mean the n-dimensional vector, where n is the number of agents.

The code is available on https://github.com/rhoowd/sched net

The unit of bandwidth is 2 bytes which can express one real value (float16 type)

We train and execute DIAL without discretize/regularize unit (DRU), because in our setting, agents can exchange messages that can express real values.

Figures and Tables:

Foerster et al. (2017a); Lowe et al. (2017); Gupta et al. (2017); Sunehag et al. (2018), and Foerster et al. (2017b) adopt the framework of centralized training with decentralized execution, empowering the agent to learn cooperative behavior considering other agents' policies without any communication in distributed execution.

Figure 2: Architecture of the critic. FC stands for fully connected neural network.

schemes. k = 1, l = 1 Figure 3: Learning curves during the learning of the PP and CCN tasks. The plots show the average time taken to complete the task, where shorter time is better for the agents.

Figure 4: Instances of scheduling results over 25 time steps in PP

Figure 6: Instances of scheduling results over 25 time steps in CCN

to simplify the presentation.Algorithm 1 SchedNet 1: Initialize actor parameters θ u , scheduler parameters θ wg , and critic parameters θ c 2: Initialize target scheduler parameters θ wg , and target critic parameters θ c 3: for episode = 1 to M do

Figure 7: Illustrations of the experimental environment

Figure 8: Performance evaluation of SchedNet. The graphs show the average time taken to complete the task, where shorter time is better for the agents.

Messages for different location of agent.

Figure 9: Projection of encoded messages into 2D plane in PP.

Figure 10: Impact of applying RNN (k = 1 and l = 2)

Figure 11: Comparison with other baselinesResult in CCN Figure11illustrates the learning curve of 200,000 steps in CCN. In FC, since all agents can broadcast their message during execution, they achieve the best performance. IDQN and COMA in which no communication is allowed, take a longer time to complete the task compared to other baselines. The performances of both are similar because no cooperation can be achieved without the exchange of observations in this environment. As expected, SchedNet and DIAL outperform IDQN and COMA. Although DIAL works well when there is no contention constraint, under the contention constraint, the average number of steps to complete the task in DIAL(1) is

Figure 12: Proposed scheduling architecture. Each agent i calculates its scheduling weight wi from weight generator (WG), and the corresponding scheduling profile c ∈ {0, 1} n is determined by the scheduling algorithm (k) (WSA(k)), satisfying the condition ||c||1 = k.

List of hyperparameters

Table content: <table><tr><td>Hyperparameter</td><td>Value</td><td>Description</td></tr><tr><td>training step</td><td colspan="2">750000 Maximum time steps until the end of training</td></tr><tr><td>episode length</td><td>1000</td><td>Maximum time steps per episode</td></tr><tr><td>discount factor</td><td>0.9</td><td>Importance of future rewards</td></tr><tr><td>learning rate for actor</td><td colspan="2">0.00001 Actor network learning rate used by Adam optimizer</td></tr><tr><td>learning rate for critic</td><td>0.0001</td><td>Critic network learning rate used by Adam optimizer</td></tr><tr><td>target update rate</td><td>0.05</td><td>Target network update rate to track learned network</td></tr><tr><td colspan="2">entropy regularization weight 0.01</td><td>Weight of regularization to encourage exploration</td></tr></table>

Performance with/without encoder

Table content: <table><tr><td>FC</td><td>SchedNet -Top(1)</td><td>Schedule w/ auto-encoder</td></tr><tr><td>1</td><td>2.030</td><td>3.408</td></tr></table>

Peer review (iclr2019):
 # overview This paper focuses on multi-agent reinforcement learning tasks that require communication between the agents, and further presupposes that the communication protocol is bandwidth constrained and contentious so that a scheduling mechanism is necessary. To address this they introduce a new learned weighting scheme based scheduler and distributed actor, centralized critic based architecture which is evaluated on a couple of communication driven multi-agent tasks. The two evaluation tasks had their bandwidth artificially constrained, and SchedNet time to convergence was shown to fall somewhere between having no communication and full communication, and somewhat better than a purely round-robin based scheduling scheme, which doesn*t seem particularly informative. From this it is difficult to assess the significance of the contributions. # pros * communication in multi-agent scenarios is an important aspect to consider, and this work shines a spotlight on scenarios in which bandwidth is constrained. * general presentation fairly clear and easy to read # cons * Would have been more impactful to focus experiments on real-world scenarios in which bandwidth is constrained and naturally contentious # other comments * pg. 2 related work, suspect you meant to call out Foerster et al 2017b in second reference not Foerster et al 2017a twice.