Title: MULTI-AGENT INTERACTIONS MODELING WITH COR-RELATED POLICIES

Abstract: In multi-agent systems, complex interacting behaviors arise due to the high correlations among agents. However, previous work on modeling multi-agent interactions from demonstrations is primarily constrained by assuming the independence among policies and their reward structures. In this paper, we cast the multiagent interactions modeling problem into a multi-agent imitation learning framework with explicit modeling of correlated policies by approximating opponents' policies, which can recover agents' policies that can regenerate similar interactions. Consequently, we develop a Decentralized Adversarial Imitation Learning algorithm with Correlated policies (CoDAIL), which allows for decentralized training and execution. Various experiments demonstrate that CoDAIL can better regenerate complex interactions close to the demonstrators and outperforms state-of-the-art multi-agent imitation learning methods. Our code is available at https://github.com/apexrl/CoDAIL.

1. INTRODUCTION:

Modeling complex interactions among intelligent agents from the real world is essential for understanding and creating intelligent multi-agent behaviors, which is typically formulated as a multiagent learning (MAL) problem in multi-agent systems. When the system dynamics are agnostic and non-stationary due to the adaptive agents with implicit goals, multi-agent reinforcement learning (MARL) is the most commonly used technique for MAL. MARL has recently drawn much attention and achieved impressive progress on various non-trivial tasks, such as multi-player strategy games (OpenAI, 2018; Jaderberg et al., 2018) , traffic light control (Chu et al., 2019) , taxi-order dispatching (Li et al., 2019) 

A central challenge in MARL is to specify a good learning goal, as the agents' rewards are correlated and thus cannot be maximized independently (Bu et al., 2008) . Without explicit access to the reward signals, imitation learning could be the most intuitive solution for learning good policies directly from demonstrations. Conventional solutions such as behavior cloning (BC) (Pomerleau, 1991) learn the policy in a supervised manner by requiring numerous data while suffering from compounding error (Ross & Bagnell, 2010; Ross et al., 2011) . Inverse reinforcement learning (IRL) (Ng et al., 2000; Russell, 1998) alleviates these shortcomings by recovering a reward function but is always expensive to obtain the optimal policy due to the forward reinforcement learning procedure in an inner loop. Generative adversarial imitation learning (GAIL) (Ho & Ermon, 2016 ) leaves a better candidate for its model-free structure without compounding error, which is highly effective and scalable. However, real-world multi-agent interactions could be much challenging to imitate because of the strong correlations among adaptive agents' policies and rewards. Consider if a football coach wants to win the league, he must make targeted tactics against various opponents, in addition to the situation of his team. Moreover, the multi-agent environment tends to give rise to more severe compounding errors with more expensive running costs.

Motivated by these challenges, we investigate the problem of modeling complicated multi-agent interactions from a pile of off-line demonstrations and recover their on-line policies, which can regenerate analogous multi-agent behaviors. Prior studies for multi-agent imitation learning typically limit the complexity in demonstrated interactions by assuming isolated reward structures (Barrett et al., 2017; Le et al., 2017; Lin et al., 2014; Waugh et al., 2013) and independence in per-agent policies that overlook the high correlations among agents (Song et al., 2018; Yu et al., 2019) . In this paper, we cast the multi-agent interactions modeling problem into a multi-agent imitation learning framework with correlated policies by approximating opponents' policies, in order to reach inaccessible opponents' actions due to concurrently execution of actions among agents when making decisions. Consequently, with approximated opponents model, we develop a Decentralized Adversarial Imitation Learning algorithm with Correlated policies (CoDAIL) suitable for learning correlated policies under our proposed framework, which allows for decentralized training and execution. We prove that our framework treats the demonstrator interactions as one of -Nash Equilibrium ( -NE) solutions under the recovered reward.

In experiments, we conduct multi-dimensional comparisons for both the reward gap between learned agents and demonstrators, along with the distribution divergence between demonstrations and regenerated interacted trajectories from learned policies. Furthermore, the results reveal that CoDAIL can better recover correlated multi-agent policy interactions than other state-of-the-art multi-agent imitation learning methods in several multi-agent scenarios. We further illustrate the distributions of regenerated interactions, which indicates that CoDAIL yields the closest interaction behaviors to the demonstrators.

2. PRELIMINARIES:



2.1. MARKOV GAME AND -NASH EQUILIBRIUM:

Markov game (MG), or stochastic game (Littman, 1994) , can be regarded as an extension of Markov Decision Process (MDP). Formally, we define an MG with N agents as a tuple N, S, A (1) , . . . , A (N ) , P, r (1) , . . . , r (N ) , ρ 0 , γ , where S is the set of states, A (i) represents the action space of agent i, where i ∈ {1, 2, . . . , N }, P :

S × A (1) × A (2) × • • • × A (N ) × S →

R is the state transition probability distribution, ρ 0 : S → R is the distribution of the initial state s 0 , and γ ∈ [0, 1] is the discounted factor. Each agent i holds its policy π (i) (a (i) |s) : S × A (i) → [0, 1] to make decisions and receive rewards defined as r (i) : S × A (1) × A (2) × • • • × A (N ) → R. We use -i to represent the set of agents except i, and variables without superscript i to denote the concatenation of all variables for all agents (e.g., π represents the joint policy and a denotes actions of all agents). For an arbitrary function f : s, a → R, there is a fact that

E π [f (s, a)] = E s∼P,a∼π [f (s, a)] E [ ∞ t=0 γ t f (s t , a t )]

, where s 0 ∼ ρ 0 , a t ∼ π, s t+1 ∼ P (s t+1 |a t , s t ). The objective of agent i is to maximize its own total expected return R

(i) E π [r (i) (s, a)] = E ∞ t=0 γ t r (i) (s t , a t ) .

In Markov games, however, the reward function for each agent depends on the joint agent actions. Such a fact implies that one's optimal policy must also depend on others' policies. For the solution to the Markov games, -Nash equilibrium ( -NE) is a commonly used concept that extends Nash equilibrium (NE) (Nash, 1951) .

Definition 1. An -NE is a strategy profile (π (i) * , π (-i) *

) such that ∃ > 0:

EQUATION

where

v (i) (s, π (i) , π (-i) ) = E π (i) ,π (-i) ,s0=s r (i) (s t , a (i) t , a (-i) t

) is the value function of agent i under state s, and Π (i) is the set of policies available to agent i.

-NE is weaker than NE, which can be seen as sub-optimal NE. Every NE is equivalent to an -NE where = 0.

2.2. GENERATIVE ADVERSARIAL IMITATION LEARNING:

Imitation learning aims to learn the policy directly from expert demonstrations without any access to the reward signals. In single-agent settings, such demonstrations come from behavior trajectories sampled with the expert policy, denoted as τ E = {(s t , a

(i) t )} ∞ t=0 .

However, in multi-agent settings, demonstrations are often interrelated trajectories, that is, which are sampled from the interactions of policies among all agents, denoted as

Ω E = {(s t , a (1) t , ..., a (N ) t )} ∞ t=0 .

For simplicity, we will use the term interactions directly as the concept of interrelated trajectories, and we refer to trajectories for a single agent.

Typically, behavior cloning (BC) and inverse reinforcement learning (IRL) are two main approaches for imitation learning. Although IRL theoretically alleviates compounding error and outperforms to BC, it is less efficient since it requires resolving an RL problem inside the learning loop. Recent proposed work aims to learn the policy without estimating the reward function directly, notably, GAIL (Ho & Ermon, 2016) , which takes advantage of Generative Adversarial Networks (GAN (Goodfellow et al., 2014)) , showing that IRL is the dual problem of occupancy measure matching. GAIL regards the environment as a black-box, which is non-differentiable but can be leveraged through Monte-Carlo estimation of policy gradients. Formally, its objective can be expressed as min

EQUATION

where D is a discriminator that identifies the expert trajectories with agents' sampled from policy π, which tries to maximize its evaluation from D; H is the causal entropy for the policy and λ is the hyperparameter.

2.3. CORRELATED POLICY:

In multi-agent learning tasks, each agent i makes decisions independently while the resulting reward r (i) (s t , a

(i) t , a (-i) t

) depends on others' actions, which makes its cumulative return subjected to the joint policy π. One common joint policy modeling method is to decouple the π with assuming conditional independence of actions from different agents (Albrecht & Stone, 2018):

EQUATION

However, such a non-correlated factorization on the joint policy is a vulnerable simplification which ignores the influence of opponents (Wen et al., 2019) . And the learning process of agent i lacks stability since the environment dynamics depends on not only the current state but also the joint actions of all agents (Tian et al., 2019) . To solve this, recent work has taken opponents into consideration by decoupling the joint policy as a correlated policy conditioned on state s and a (-i) as

EQUATION

where π (i) (a (i) |s, a (-i) ) is the conditional policy, with which agent i regards all potential actions from its opponent policies π (-i) (a (-i) |s), and makes decisions through the marginal policy

π (i) (a (i) |s) = a (-i) π (i) (a (i) |s, a (-i) )π (-i) (a (-i) |s) da (-i) = E a (-i) π (i) (a (i) |s, a (-i) ).

3. METHODOLOGY:



3.1. GENERALIZE CORRELATED POLICIES TO MULTI-AGENT IMITATION LEARNING:

In multi-agent settings, for agent i with policy π (i) , it seeks to maximize its cumulative reward against demonstrator opponents who equip with demonstrated policies π

(-i) E

via reinforcement learning:

EQUATION

where H(π (i) ) is the γ-discounted entropy (Bloem & Bambos, 2014; Haarnoja et al., 2017) of policy π (i) and λ is the hyperparameter. By coupling with Eq. ( 5), we define an IRL procedure to find a reward function r (i) such that the demonstrated joint policy outperforms all other policies, with the regularizer ψ : R S×A (1) ×•••×A (N ) → R:

EQUATION

It is worth noting that we cannot obtain the demonstrated policies from the demonstrations directly.

To solve this problem, we first introduce the occupancy measure, namely, the unnormalized distribution of s, a pairs correspond to the agent interactions navigated by joint policy π:

EQUATION

With the definition in Eq. ( 7), we can further formulate ρ π from agent i's perspective as

EQUATION

where a (i) ∼ π (i) and a (-i) ∼ π (-i) . Furthermore, with the support of Eq. ( 8), we have

EQUATION

In analogy to the definition of occupancy measure of that in a single-agent environment, we follow the derivation from Ho & Ermon (2016) and state the conclusion directly1 . Proposition 1. The IRL regarding demonstrator opponents is a dual form of a occupancy measure matching problem with regularizer ψ, and the induced optimal policy is the primal optimum, specifically, the policy learned by RL on the reward recovered by IRL can be characterize by the following equation:

EQUATION

)

With setting the regularizer ψ = ψ GA similar to Ho & Ermon (2016) , we can obtain a GAIL-like imitation algorithm to learn π

(i) E from π E given demonstrator counterparts π (-i) E

by introducing the adversarial training procedures of GANs which lead to a saddle point (π (i) , D (i) ): min 11) where D (i) denotes the discriminator for agent i, which plays a role of surrogate cost function and guides the policy learning.

EQUATION

However, such an algorithm is not practical, since we are unable to access the policies of demonstrator opponents π

(-i) E

because the demonstrated policies are always given through sets of interactions data. To alleviate this deficiency, it is necessary to deal with accessible counterparts. Thereby we propose Proposition 2.

Proposition 2. Let µ be an arbitrary function such that µ holds a similar form as π (-i) , then

E π (i) ,π (-i) [•] = E π (i) ,µ ρ π (i) ,π (-i) (s,a (i) ,a (-i) ) ρ π (i) ,µ (s,a (i) ,a (-i) )

• .

Proof. Substituting π (-i) with µ in Eq. ( 9) by importance sampling.

Proposition 2 raises an important point that a term of importance weight can quantify the demonstrator opponents. By replacing π

(-i) E

with π (-i) , Eq. ( 11) is equivalent with

EQUATION

) where α = ρ π (i) ,π (-i) E (s,a (i) ,a (-i) ) ρ π (i) ,π (-i) (s,a (i) ,a (-i) )

is the importance sampling weight. In practice, it is challenging to estimate the densities and the learning methods might suffer from large variance. Thus, we fix α = 1 in our implementation, and as the experimental results have shown, it has no significant influences on performance. Besides, a similar approach can be found in Kostrikov et al. (2018) .

So far, we've built a multi-agent imitation learning framework, which can be easily generalized to correlated or non-correlated policy settings. No prior has to be considered in advance since the discriminator is able to learn the implicit goal for each agent.

3.2. LEARN WITH THE OPPONENTS MODEL:

With the objective shown in Eq. ( 11), demonstrated interactions can be imitated by updating discriminators to offer surrogate rewards and learning their policies alternately. Formally, the update of discriminator for each agent i can be expressed as:

EQUATION

and the update of policy is:

EQUATION

where discriminator D (i) is parametrized by ω, and the policy π (i) is parametrized by θ. It is worth noting that the agent i considers opponents' action a (-i) while updating its policy and discriminator, with integrating all its possible decisions to find the optimal response. However, it is unrealistic to have the access to opponent joint policy π(a (-i) |s) for agent i. Thus, it is possible to estimate opponents' actions via approximating π (-i) (a (-i) |s) using opponent modeling. To that end, we construct a function σ (i) (a (-i) 

|s) : S × A (1) × • • • × A (i-1) × A (i+1) × • • • × A (N ) → [0, 1]

N -1 , as the approximation of opponents for each agent i. Then we rewrite Eq. ( 13) and Eq. ( 14) as:

EQUATION

and

∇ θ J π (θ) ≈ E s∼P,â (-i) ∼σ (i) ,a (i) ∼π (i) θ ∇ θ (i) log π (i) θ (a (i) |s, â(-i) )A (i) (s, a (i) , â(-i) ) -λ∇ θ (i) H(π (i)

θ ) (16) respectively. Therefore, each agent i must infer the opponents model σ (i) to approximate the unobservable policies π (-i) , which can be achieved via supervised learning. Specifically, we learn in discrete action space by minimizing a cross-entropy (CE) loss, and a mean-square-error (MSE) loss in continuous action space:

L = 1 2 E s∼p σ (i) (a (-i) |s) -π (-i) (a (-i) |s)

2 , continuous action space

EQUATION

)

With opponents modeling, agents are able to be trained in a fully decentralized manner. We name our algorithm as Decentralized Adversarial Imitation Learning with Correlated policies (Correlated DAIL, a.k.a. CoDAIL) and present the training procedure in Appendix Algo. 1, which can be easily scaled to a distributed algorithm. As a comparison, we also present a non-correlated DAIL algorithm with non-correlated policy assumption in Appendix Algo. 2.

3.3. THEORETICAL ANALYSIS:

In this section, we prove that the reinforcement learning objective against demonstrator counterparts shown in the last section is essentially equivalent to reaching an -NE.

Since we fix the policies of agents -i as π

(-i)

E , the RL procedure mentioned in Eq. ( 5) can be regarded as a single-agent RL problem. Similarly, with a fixed π (-i) E , the IRL process of Eq. ( 6) is cast to a single-agent IRL problem, which recovers an optimal reward function r (i) * which achieves the best performance following the joint action π E . Thus we have

EQUATION

)

We can also rewrite Eq. ( 18) as λH(π

(i) E ) + E π (i) E ,π (-i) E [r (i) (s, a (i) , a (-i) )] ≥ λH(π (i) ) + E π (i) ,π (-i) E [r (i) (s, a (i) , a (-i) )] (19)

for all π (i) ∈ Π (i) , which is equivalent to

EQUATION

t , a

(-i) t ) + λ(H(π (i) ) -H(π (i) E )), ∀π (i) ∈ Π (i) .

Given the value function defined in Eq. ( 1) for each agent i, for H(π (i) )-H(π

EQUATION

)

For H(π (i) ) -H(π

EQUATION

which is exactly the -NE defined in Definition 1. We can always prove that is bounded in small values such that the -NE solution concept is meaningful. Generally, random policies that keep vast entropy are not always considered as sub-optimal solutions or demonstrated policies π

(i)

E in most reinforcement learning environments. As we do not require those random policies, we can remove them from the candidate policy set Π (i) , which indicates that H(π (i) ) is bounded in small values, so as . Empirically, we adopt a small λ, and attain the demonstrator policy π E with an efficient learning algorithm to become a close-to-optimal solution. Thus, we conclude that the objective of our CoDAIL assumes that demonstrated policies institute an -NE solution concept (but not necessarily unique) that can be controlled the hyperparameter λ under some specific reward function, from which the agent learns a policy. It is worth noting that Yu et al. (2019) claimed that NE is incompatible with maximum entropy inverse reinforcement learning (MaxEnt IRL) because NE assumes that the agent never takes sub-optimal actions. Nevertheless, we prove that given demonstrator opponents, the multi-agent MaxEnt IRL defined in Eq. ( 6) is equivalent to finding an -NE.

4. RELATED WORK:

Albeit non-correlated policy learning guided by a centralized critic has shown excellent properties in couple of methods, including MADDPG (Lowe et al., 2017) , COMA (Foerster et al., 2018) , MA Soft-Q (Wei et al., 2018) , it lacks in modeling complex interactions because its decisions making relies on the independent policy assumption which only considers private observations while ignores the impact of opponent behaviors. To behave more rational, agents must take other agents into consideration, which leads to the studies of opponent modeling (Albrecht & Stone, 2018) where an agent models how its opponents behave based on the interaction history when making decisions (Claus & Boutilier, 1998; Greenwald et al., 2003; Wen et al., 2019; Tian et al., 2019) .

For multi-agent imitation learning, however, prior works fail to learn from complicated demonstrations, and many of them are bounded with particular reward assumptions. For instance, Bhattacharyya et al. (2018) proposed Parameter Sharing Generative Adversarial Imitation Learning (PS-GAIL) that adopts parameter sharing trick to extend GAIL to handle multi-agent problems directly, but it does not utilize the properties of Markov games with strong constraints on the action space and the reward function. Besides, there are many works built-in Markov games that are restricted under tabular representation and known dynamics but with specific prior of reward structures, as fully cooperative games (Barrett et al., 2017; Le et al., 2017; Šošic et al., 2016; Bogert & Doshi, 2014) , two-player zero-sum games (Lin et al., 2014) , two-player general-sum games (Lin et al., 2018) , and linear combinations of specific features (Reddy et al., 2012; Waugh et al., 2013) .

Recently, some researchers take advantage of GAIL to solve Markov games. Inspired by a specific choice of Lagrange multipliers for a constraint optimization problem (Yu et al., 2019) , Song et al. (2018) derived a performance gap for multi-agent from NE. It proposed multi-agent GAIL (MA-GAIL), where they formulated the reward function for each agent using private actions and observations. As an improvement, Yu et al. ( 2019) presented a multi-agent adversarial inverse reinforcement learning (MA-AIRL) based on logistic stochastic best response equilibrium and MaxEnt IRL. However, both of them are inadequate to model agent interactions with correlated policies with independent discriminators. By contrast, our approach can generalize correlated policies to model the interactions from demonstrations and employ a fully decentralized training procedure without to get access to know the specific opponent policies.

Except for the way of modeling multi-agent interactions as recovering agents' policies from demonstrations, which can regenerate similar interacted data, some other works consider different effects of interactions. Grover et al. (2018) proposed to learn a policy representation function of the agents based on their interactions and sets of generalization tasks using the learned policy embeddings. They regarded interactions as the episodes that contain only k (in the paper they used 2 agents), which constructs an agent-interaction graph. Different from us, they focused on the potential relationships among agents to help characterize agent behaviors. Besides, Kuhnt et al. (2016) and Gindele et al. (2015) proposed to use the Dynamic Bayesian Model that describes physical relationships among vehicles and driving behaviors to model interaction-dependent behaviors in autonomous driving scenario.

Correlated policy structures that can help agents consider the influence of other agents usually need opponents modeling (Albrecht & Stone, 2018) to infer others' actions. Opponent modeling has a rich history in MAL (Billings et al., 1998; Ganzfried & Sandholm, 2011) , and lots of researches have recently worked out various useful approaches for different settings in deep MARL, e.g., DRON (He et al., 2016) and ROMMEO (Tian et al., 2019) . In this paper, we focus on imitation learning with correlated policies, and we choose a natural and straightforward idea of opponent modeling that learning opponents' policies in the way of supervised learning with historical trajectories. Opponent models are used both in the training and the execution stages.

5. EXPERIMENTS:



5.1. EXPERIMENTAL SETTINGS:

Environment Description We test our method on the Particle World Environments (Lowe et al., 2017) , which is a popular benchmark for evaluating multi-agent algorithms, including several cooperative and competitive tasks. Specifically, we consider two cooperative scenarios and two com- petitive ones as follows: 1) Cooperative-communication, with 2 agents and 3 landmarks, where an unmovable speaker knowing the goal, cooperates with a listener to reach a particular landmarks who achieves the goal only through the message from the speaker; 2) Cooperative-navigation, with 3 agents and 3 landmarks, where agents must cooperate via physical actions and it requires each agent to reach one landmark while avoiding collisions; 3) Keep-away, with 1 agent, 1 adversary and 1 landmark, where the agent has to get close to the landmark, while the adversary is rewarded by pushing away the agent from the landmark without knowing the target; 4) Predator-prey, with 1 prey agent with 3 adversary predators, where the slower predactor agents must cooperate to chase the prey agent that moves faster and try to run away from the adversaries.

Experimental Details:

We aim to compare the quality of interactions modeling in different aspects.

To obtain the interacted demonstrations sampled from correlated policies, we train the demonstrator agent via a MARL learning algorithm with opponents modeling to regard others' policies into one's decision making, since the ground-truth reward in those simulated environments is accessible. Specifically, we modify the multi-agent version ACKTR (Wu et al., 2017; Song et al., 2018) , an efficient model-free policy gradient algorithm, by keeping an auxiliary opponents model and a conditioned policy for each agent, which can transform the original centralized on-policy learning algorithm to be decentralized. Note that we do not necessarily need experts that can do well in our designated environments. Instead, any demonstrator will be treated as it is from an -NE strategy concept under some unknown reward functions, which will be recovered by the discriminator. In our training procedure, we first obtain demonstrator policies induced by the ground-truth rewards and then generate demonstrations, i.e., the interactions data for imitation training. Then we train the agents through the surrogate rewards from discriminators. We compare CoDAIL with MA-AIRL, MA-GAIL, non-correlated DAIL (NC-DAIL) (the only difference between MA-GAIL and NC-DAIL is whether the reward function depends on joint actions or individual action) and a random agent. We do not apply any prior to the reward structure for all tasks to let the discriminator learn the implicit goals. All training procedures are pre-trained via behavior cloning to reduce the sample complexity, and we use 200 episodes of demonstrations, each with a maximum of 50 timesteps.

5.2. REWARD GAP:

Tab. 1 and Tab. 2 show the averaged absolute differences of reward for learned agents compared to the demonstrators in cooperative and competitive tasks, respectively. The learned interactions are considered superior if there are smaller reward gaps. Since cooperative tasks are reward-sharing, we show only a group reward for each task in Tab. 1. Compared to the baselines, CoDAIL achieves smaller gaps in both cooperative and competitive tasks, which suggests that our algorithm has a robust imitation learning capability of modeling the demonstrated interactions. It is also worth noting that CoDAIL achieves higher performance gaps in competitive tasks than cooperative ones, for which we think that conflict goals motivate more complicated interactions than a shared goal. Besides, MA-GAIL and NC-DAIL are about the same, indicating that less important is the surrogate reward structure on these multi-agent scenarios. To our surprise, MA-AIRL does not perform well in some environments, and even fails in Predator-prey. We list the raw obtained rewards in Appendix C, and we provide more hyperparameter sensitivity results in Appendix D.

5.3. DIVERGENCE OVER INTERACTIONS:

Since we aim to recover the interactions of agents generated by the learned policies, it is proper to evaluate the relevance between distributions of regenerated interactions and demonstration data. Specifically, we collect positions of agents over hundreds of state-action tuples, which can be regarded as the low-dimension projection of the state-action interactions. We start each episode from a different initial state but the same for each algorithm in one episode. We run all the experiments under the same random seed, and collect positions of each agent in the total 100 episodes, each with a maximum of 50 timesteps.

We first estimate the distribution of position (x, y) via Kernel Density Estimation (KDE) (Rosenblatt, 1956) with Gaussian kernel to compute the Kullback-Leibler (KL) divergence between the generated interactions with the demonstrated ones, shown in Tab. 3. It is evident that in terms of the KL divergence between regenerated interactions with demonstrator interactions, CoDAIL generates the interaction data that obtains the minimum gap with the demonstration interaction, and highly outperforms other baseline methods. Besides, MA-GAIL and NC-DAIL reflect about-thesame performance to model complex interactions, while MA-AIRL behaves the worst, even worse than random agents on Predator-prey.

5.4. VISUALIZATIONS OF INTERACTIONS:

To further understand the interactions generated by learned policies compared with the demonstrators, we visualize the interactions for demonstrator policies and all learned ones. We plot the density distribution of positions, (x, y) and marginal distributions of x-position and y-position. We illustrate the results conducted on Keep-away in Fig. 1 , other scenarios can be found in the Appendix E.

Higher frequency positions in collected data are colored darker in the plane, and higher the value with respect to its marginal distributions.

As shown in Fig. 1 , the interaction densities of demonstrators and CoDAIL agents are highly similar (and with the smallest KL divergence), which tend to walk in the right-down side. In contrast, other learned agents fail to recover the demonstrator interactions. It is worth noting that even different policies can interact to earn similar rewards, but still keep vast differences among their generated interactions. Furthermore, such a result reminds us that the real reward is not the best metric to evaluate the quality of modeling the demonstrated interactions or imitation learning (Li et al., 2017) . 

6. CONCLUSION:

In this paper, we focus on modeling complex multi-agent interactions via imitation learning on demonstration data. We develop a decentralized adversarial imitation learning algorithm with correlated policies (CoDAIL) with approximated opponents modeling. CoDAIL allows for decentralized training and execution and is more capable of modeling correlated interactions from demonstrations shown by multi-dimensional comparisons against other state-of-the-art multi-agent imitation learning methods on several experiment scenarios. In the future, we will consider covering more imitation learning tasks and modeling the latent variables of policies for diverse multi-agent imitation learning.

Appendices A ALGORITHM OUTLINES:

A.1 CODAIL ALGORITHM Algo. 1 demonstrates the outline for our CoDAIL algorithm with non-correlated policy structure defined in Eq. ( 4), where we approximate the opponents model σ (i) (a (-i) |s), improve the discriminator D (i) and the policy π (i) iteratively.

Algorithm 1 CoDAIL Algorithm 1: Input: Expert interactive demonstrations Ω E ∼ π E , N policy parameters θ (1) , ..., θ (N ) , N value parameters φ (1) , ..., φ (N ) , N opponents models parameters ψ (1) , ..., ψ (N ) and N discriminator parameters ω (1) , ..., ω (N ) ; 2: for k = 0, 1, 2, . . . do

3::

Sample interactions among N agents Ω k ∼ π, where π is in correlated policy structure and each agent i utilizes its opponents model σ (i) to help making decisions.

4::

for agent i = 1, 2, . . . , N do 5:

Use state-action pairs (s, a (-i) ) ∈ Ω k to update ψ (i) to minimize the objective as shown in Eq. ( 17).

6::

For every state-action pair (s, a (i) ) ∈ Ω k , sample estimated opponent policies from opponents model: â(-i) ∼ σ (i) (a (i) |s), and update ω (i) with the gradient as shown in Eq. ( 15).

7::

Compute advantage estimation A (i) for each tuple (s, a (i) , , â(-i) ) with surrogate reward function r

EQUATION

-V (i)

EQUATION

Update φ (i) to minimize the objective:

EQUATION

Update θ (i) following the gradient shown in Eq. ( 16): Algorithm 2 NC-DAIL Algorithm 1: Input: Expert interactive demonstrations Ω E ∼ π E , N policy parameters θ (1) , ..., θ (N ) , N value parameters φ (1) , ..., φ (N ) and N discriminator parameters ω (1) , ..., ω (N ) ; 2: for k = 0, 1, 2, . . . do

EQUATION

Sample interactions between N agents Ω k ∼ π, where π is in non-correlated policy structure.

for agent i = 1, 2, . . . , N do 5:

Use (s, a (i) , a (-i) ) ∈ Ω k to update ω (i) with the gradient:

EQUATION

Compute advantage estimation A (i) for (s, a (i) , a (-i) ) ∈ Ω k with surrogate reward function r (i) (s, a (i) , a (-i) ) = log(D (i)

EQUATION

Update φ (i) to minimize the objective:

EQUATION

Update θ (i) by taking a gradient step with:

EQUATION

end for 10: end for

B MODEL ARCHITECTURES:

During our experiments, we use two layer MLPs with 128 cells in each layer, for policy networks, value networks, discriminator networks and opponents model networks on all scenarios. Specifically for opponents models, we utilize a multi-head-structure network, where each head predicts each opponent's action separately, and we get the overall opponents joint action a (-i) by concatenating all actions. The batch size is set to 1000. The policy is trained using K-FAC optimizer (Martens & Grosse, 2015) with learning rate of 0.1 and with a small λ of 0.05. All other parameters for K-FAC optimizer are the same in (Wu et al., 2017) . We train each algorithm for 55000 epochs with 5 random seeds to gain its average performance on all environments.

C RAW RESULTS:

We list the raw obtained rewards of all algorithms in each scenarios. We evaluate how the stability of our algorithm when the hyperparameters change during our experiments on Communication-navigation. Tab. 6 shows the total reward difference between learned agents and demonstrators when we modify the training frequency of D and G (i.e., the policy), which indicates that the frequencies of D and G are more stable when D is trained slower than G, and the result reaches a relative better performance when the frequency is 1:2 or 1:1. Fig. 2 illustrates that the choice of λ has little effect on the total performance. The reason may be derived from the discrete action space in this environment, where the policy entropy changes gently.

:

Note thatHo & Ermon (2016) proved the conclusion under the goal to minimize the cost instead of maximizing the reward of an agent.

Figures and Tables:

Figure 1: The density and marginal distribution of agent positions, in 100 repeated episodes with different initialized states, generated from different learned policies upon Keep-away. The top row of each sub-figure is drawn from state-action pairs of all agents. Meanwhile, the bottom row explains for each individual (KL means the KL divergence between generated interactions shown in the top row and the demonstrators).

step by step NC-DAIL algorithm with non-correlated decomposition of joint policy defined in Eq. (3) in Algo. 2.

Figure 2: Results of different entropy coefficient λ.

Average reward gaps between demonstrators and learned agents in 2 cooperative tasks. Means and standard deviations are taken across different random seeds.

Table content: <table><tr><td>Algorithm</td><td>Coop.-Comm.</td><td>Coop.-Navi.</td></tr><tr><td>Demonstrators</td><td>0 ± 0</td><td>0 ± 0</td></tr><tr><td>MA-AIRL</td><td>0.780 ± 0.917</td><td>6.696 ± 3.646</td></tr><tr><td>MA-GAIL</td><td>0.638 ± 0.624</td><td>7.596 ± 3.088</td></tr><tr><td>NC-DAIL</td><td>0.692 ± 0.597</td><td>6.912 ± 3.971</td></tr><tr><td>CoDAIL</td><td>0.632 ± 0.685</td><td>6.249 ± 2.779</td></tr><tr><td>Random</td><td colspan="2">186.001 ± 16.710 322.1124 ± 15.358</td></tr></table>

Average reward gaps between demonstrators and learned agents in 2 competitive tasks, where 'agent+' and 'agent-' represent 2 teams of agents and 'total' is their sum. Means and standard deviations are taken across different random seeds.

Table content: <table><tr><td>Algorithm</td><td/><td>Keep-away</td><td/><td/><td>Pred.-Prey</td><td/></tr><tr><td/><td>Total</td><td>Agent+</td><td>Agent-</td><td>Total</td><td>Agent+</td><td>Agent-</td></tr><tr><td>Demonstrators</td><td>0 ± 0</td><td>0 ± 0</td><td>0 ± 0</td><td>0 ± 0</td><td>0 ± 0</td><td>0 ± 0</td></tr><tr><td>MA-AIRL</td><td colspan="2">12.273 ± 1.817 4.149 ± 1.912</td><td colspan="4">8.998 ± 4.345 279.535 ± 77.903 35.100 ± 1.891 174.235 ± 73.168</td></tr><tr><td>MA-GAIL</td><td>1.963 ± 1.689</td><td>1.104 ± 1.212</td><td>1.303 ± 0.798</td><td>15.788 ± 10.887</td><td>4.800 ± 2.718</td><td>8.826 ± 3.810</td></tr><tr><td>NC-DAIL</td><td>1.805 ± 1.695</td><td>1.193 ± 0.883</td><td>1.539 ± 1.188</td><td>27.611 ± 14.645</td><td>8.260 ± 7.087</td><td>6.975 ± 5.130</td></tr><tr><td>CoDAIL</td><td>0.269 ± 0.078</td><td>0.064 ± 0.041</td><td>0.219 ± 0.084</td><td>10.456 ± 6.762</td><td>4.500 ± 3.273</td><td>4.359 ± 2.734</td></tr><tr><td>Random</td><td colspan="5">28.272 ± 2.968 25.183 ± 2.150 53.455 ± 2.409 100.736 ± 6.870 37.980 ± 2.396</td><td>13.204 ± 8.444</td></tr></table>

KL divergence of learned agents position distribution and demonstrators position distribution from an individual perspective in different scenarios. 'Total' is the KL divergence for state-action pairs of all agents, and 'Per' is the averaged KL divergence of each agent. Experiments are conducted under the same random seed. Note that unmovable agents are not recorded since they never move from the start point, and there is only one movable agent in Cooperative-communication.

Table content: <table><tr><td>Algorithm</td><td>Coop.-Comm.</td><td colspan="2">Coop.-Navi.</td><td colspan="2">Keep-away</td><td colspan="2">Pred.-Prey</td></tr><tr><td/><td>Total/Per</td><td>Total</td><td>Per</td><td>Total</td><td>Per</td><td>Total</td><td>Per</td></tr><tr><td>Demonstrators</td><td>0</td><td>0</td><td>0</td><td>0</td><td>0</td><td>0</td><td>0</td></tr><tr><td>MA-AIRL</td><td>35.525</td><td>18.071</td><td>47.241</td><td>69.146</td><td colspan="3">98.248 71.568 118.511</td></tr><tr><td>MA-GAIL</td><td>34.681</td><td>15.034</td><td>45.550</td><td>51.721</td><td>69.820</td><td>9.998</td><td>27.116</td></tr><tr><td>NC-DAIL</td><td>38.002</td><td>16.202</td><td>46.040</td><td>46.563</td><td colspan="3">61.780 16.698 33.307</td></tr><tr><td>CoDAIL</td><td>6.427</td><td>9.033</td><td>23.100</td><td>3.113</td><td>5.735</td><td>8.621</td><td>22.600</td></tr><tr><td>Random</td><td>217.456</td><td colspan="6">174.892 221.209 191.344 234.829 37.555 82.361</td></tr></table>

Raw average total rewards in 2 comparative tasks. Means and standard deviations are taken across different random seeds.

Table content: <table><tr><td>Algorithm</td><td>Coop.-Comm.</td><td>Coop.-Navi.</td></tr><tr><td>Demonstrators</td><td>-24.560 ± 1.213</td><td>-178.597 ± 6.383</td></tr><tr><td>MA-AIRL</td><td>-25.366 ± 1.492</td><td>-172.733 ± 5.595</td></tr><tr><td>MA-GAIL</td><td>-25.081 ± 1.421</td><td>-172.169 ± 4.105</td></tr><tr><td>NC-DAIL</td><td>-25.177 ± 1.371</td><td>-171.685 ± 4.591</td></tr><tr><td>CoDAIL</td><td>-25.107 ± 1.486</td><td>-183.846 ± 5.728</td></tr><tr><td>Random</td><td colspan="2">-247.606 ± 17.842 -1139.569 ± 19.192</td></tr></table>

Raw average rewards of each agent in 2 competitive tasks, where agent+ and agent-represent 2 teams of agents and total is their sum. Means and standard deviations are taken across different random seeds. .815 ± 0.909 -12.092 ± 0.617 -6.723 ± 0.430 MA-AIRL -31.088 ± 2.371 -15.367 ± 3.732 -15.721 ± 4.448 MA-GAIL -20.778 ± 0.994 -12.818 ± 1.105 -7.959 ± 0.796 NC-DAIL -20.619 ± 0.957 -12.357 ± 1.424 -8.262 ± 1.310 CoDAIL -19.084 ± 0.882 -12.142 ± 0.578 -6.942 ± 0.433 Random -47.086 ± 2.485 13.091 ± 2.032 -60.177 ± 2.225

Table content: <table><tr><td>Algorithm</td><td/><td>Total</td><td>Keep-away Agent+</td><td>Agent-</td></tr><tr><td colspan="2">Demonstrators -18Algorithm</td><td>Total</td><td>Pred.-Prey Agent+</td><td>Agent-</td></tr><tr><td>Demonstrators</td><td colspan="2">65.202 ± 18.661</td><td>44.820 ± 4.663</td><td>-69.258 ± 5.361</td></tr><tr><td>MA-AIRL</td><td colspan="2">-210.546 ± 80.333</td><td>8.040 ± 3.626</td><td>-234.666 ± 71.165</td></tr><tr><td>MA-GAIL</td><td colspan="2">65.202 ± 18.661</td><td>44.820 ± 4.663</td><td>-69.258 ± 5.361</td></tr><tr><td>NC-DAIL</td><td colspan="2">59.553 ± 30.684</td><td>42.320 ± 10.323</td><td>-67.407 ± 3.700</td></tr><tr><td>CoDAIL</td><td colspan="2">79.445 ± 5.913</td><td>47.480 ± 4.067</td><td>-61.909 ± 6.367</td></tr><tr><td>Random</td><td colspan="2">-31.747 ± 7.865</td><td>5.160 ± 1.170</td><td>-47.227 ± 7.830</td></tr></table>

Results of different training frequency (1:4, 1:2, 1:1, 2:1, 4:1) of D and G on Communicationnavigation. Means and standard deviations are taken across different random seeds.

Table content: <table><tr><td/><td colspan="6">Training Frequency Total Reward Difference</td></tr><tr><td/><td/><td>1:4</td><td/><td colspan="3">2541.144 ± 487.711</td></tr><tr><td/><td/><td>1:2</td><td/><td colspan="2">12.004 ± 5.496</td></tr><tr><td/><td/><td>1:1</td><td/><td colspan="2">6.249 ± 2.779</td></tr><tr><td/><td/><td>2:1</td><td/><td colspan="3">1136.255 ± 1502.604</td></tr><tr><td/><td/><td>4:1</td><td/><td colspan="3">2948.878 ± 1114.528</td></tr><tr><td/><td>14</td><td/><td/><td/><td/></tr><tr><td/><td>12</td><td/><td/><td/><td/></tr><tr><td>Total Reward Difference</td><td>6 8 10</td><td/><td/><td/><td/></tr><tr><td/><td>4</td><td/><td/><td/><td/></tr><tr><td/><td>2</td><td/><td/><td/><td/></tr><tr><td/><td>0.0</td><td>0.2</td><td>0.4</td><td>0.6</td><td>0.8</td><td>1.0</td></tr><tr><td/><td/><td/><td colspan="2">Entropy Coefficient</td><td/></tr></table>

Peer review (iclr2020):
This paper proposes to model interactions in a multi-agent system by considering correlated policies. In order to do so, the work modifies the GAIL framework to derive a learning objective. Similar to GAIL, the discriminator distinguishes between state, action, next state sequences but crucially the actions here are considered for all agents. The paper is a natural extension of GAIL/MA-GAIL. I have two major points that need to be addressed. 1. The exposition and significance of some of the theoretical results is unclear. - The non-correlated and correlated eqns in 2nd and 3rd line in eq. 8 are not equivalent in general, yet connected via an equality. In particular, Proposition 2 considers an importance weighting procedure to reweight state, action, next state triplets. It is unclear how this resolves the shortcomings of pi_E^{-1} being inaccessible. Prop 2 shifts from pi_E^{-1} to pi^{-1} and hence, the expectations in Prop 2 and Eq. 11 are not equivalent. - More importantly, how are the importance weights estimated in Eq. 12? The numerator requires pi_E^{-1}, which is not accessible. If the numerator and denominator are estimated separately, it becomes a chicken-and-egg problem since the denominator is itself intended to be an imitating the expert policy appearing in the numerator? 2. Missing related work There is a huge body of missing work in multi-agent interactions modeling and generative modeling. [1, 2] consider modeling of agent interactions via imitation learning and a principled evaluation framework of generalization in the Markov games setting. By sharing parameters, they are also able to model correlations across agent policies and have strong results on generalization to cooperation/competition with unseen agents with similar policies (which wouldn*t have been possible if correlations were not modeled). Similarly, [3, 4] are other similar works which consider modeling of other agent interactions/diverse behaviors via imitation style approaches. Finally, the idea of correcting for the mismatch in state, action, next state triplets in Proposition 2 has been considered for model-based off-policy evaluation in [5]. They proposed a likelihood-free method to estimate importance weights, which seems might be necessary for this task as well (re: qs. on how are importance weights estimated?). Re:experiments. Results look good and convincing for most parts. I don*t see much value of the qualitative evaluation in Figure 1. If the KL divergence is low, we can expect the marginals to be better estimated. Trying out various levels of generalization as proposed in [2] would significantly strengthen the paper. Typos sec 2.1 Transition dynamics should have range in R+ Proof of Prop 2. mu instead of u References: [1] Learning Policy Representations in Multiagent Systems. ICML 2018. [2] Evaluating Generalization in Multiagent Systems using Agent-Interaction Graphs. AAMAS 2018. [3] Machine Theory of Mind. ICML 2018. [4] Robust imitation of diverse behaviors. NeurIPS 2017. [5] Bias Correction of Learned Generative Models using Likelihood-free Importance Weighting. NeurIPS 2019.