Title: RELATIONAL FORWARD MODELS FOR MULTI-AGENT LEARNING

Abstract: The behavioral dynamics of multi-agent systems have a rich and orderly structure, which can be leveraged to understand these systems, and to improve how artificial agents learn to operate in them. Here we introduce Relational Forward Models (RFM) for multi-agent learning, networks that can learn to make accurate predictions of agents' future behavior in multi-agent environments. Because these models operate on the discrete entities and relations present in the environment, they produce interpretable intermediate representations which offer insights into what drives agents' behavior, and what events mediate the intensity and valence of social interactions. Furthermore, we show that embedding RFM modules inside agents results in faster learning systems compared to non-augmented baselines. As more and more of the autonomous systems we develop and interact with become multi-agent in nature, developing richer analysis tools for characterizing how and why agents make decisions is increasingly necessary. Moreover, developing artificial agents that quickly and safely learn to coordinate with one another, and with humans in shared environments, is crucial.
RELATIONAL ANALYSIS OF MARL SYSTEMS
Our RFM is based on graph networks (GN) (Battaglia et al., 2018) , and is trained by supervised learning to predict the dynamics of multi-agent systems. Our model takes as input a semantic description of the state of the environment, and outputs either an action prediction for each agent, or a prediction of the cumulative reward each agent will receive until the end of the episode. We show that our model performs well at these tasks, and crucially, produces interpretable intermediate representations that are useful both as analysis tools, and as inputs to artificial agents who can exploit these predictions to improve their decision-making.

1. INTRODUCTION:

The study of multi-agent systems has received considerable attention in recent years and some of the most advanced autonomous systems in the world today are multi-agent in nature (e.g. assembly lines and warehouse management systems). In particular, research in multi-agent reinforcement learning (MARL), where multiple learning agents perceive and act in a shared environment, has produced impressive results (Jaderberg et al., 2018; Pachocki et al., 2018; Leibo et al., 2017; Hughes et al., 2018; Peysakhovich & Lerer, 2017a; Lerer & Peysakhovich, 2017; Bansal et al., 2017; Lanctot et al., 2017) .

One of the outstanding challenges in this domain is how to foster coordinated behavior among learning agents. In hand-engineered multi-agent systems (e.g. assembly lines), it is possible to obtain coordination by design, where expert engineers carefully orchestrate each agent's behavior and role in the system. This, however, rules out situations where either humans or artificial learning agents are present in the environment. In learning-based systems, there have been some successes by introducing a centralized controller (D'Andrea, 2012; Foerster et al., 2016; 2017; Hong et al., 2017; Lowe et al., 2017) . However, these cannot scale to large number of agents or to mixed human-robot ensembles. There is thus an increasing focus on multi-agent systems that learn how to coordinate on their own (Jaderberg et al., 2018; Pachocki et al., 2018; Perolat et al., 2017) .

Alongside the challenges of learning coordinated behaviors, there are also the challenges of measuring them. In learning-based systems, the analysis tools currently available to researchers focus on the functioning of each single agent, and are ill-equipped to characterize systems of diverse agents as a whole. Moreover, there has been little development of tools for measuring the contextual interdependence of agents' behaviors in complex environment, which will be valuable for identifying the conditions under which agents are successfully coordinating.

Here we address these two challenges by developing Relational Forward Models (RFM) for multiagent systems. We build on recent advances in neural networks that effectively perform relational reasoning with graph networks (GN) (Battaglia et al., 2018) to construct models that learn to predict the forward dynamics of multi-agent systems. First, we show that our models can surpass previous top methods on this task (Kipf et al., 2018; Hoshen, 2017) . Perhaps more importantly, they produce intermediate representations that support the social analysis of multi-agent systems: we use our models to propose a new way to characterize what drives each agent's behavior, track when agents influence each other, and identify which factors in the environment mediate the presence and valence of social interactions. Finally, we embed our models inside agents and use them to augment the host agent's observations with predictions of others' behavior. Our results show that this leads to agents that learn to coordinate with one another faster than non-augmented baselines.

1.1. RELATED WORK:

Relational reasoning has received considerable attention in recent years and researchers have developed deep learning models that operate on graphs, rather than vectors or images, and structure their computations accordingly. These methods have been successfully applied to learning the forward dynamics of systems comprised of multiple entities and a rich relational structure, like physics simulation, multi-object scenes, visual question answering and motion-capture data (Scarselli et al., 2009; Battaglia et al., 2016; Raposo et al., 2017; Santoro et al., 2017; Gilmer et al., 2017; Watters et al., 2017; Kipf et al., 2018; Zambaldi et al., 2018) . Recently, this class of methods have been shown to successfully predict the forward dynamics of multi-agent systems, like basketball or soccer games, and to some extent, to provide insights into the relational and social structures present in the data (Hoshen, 2017; Kipf et al., 2018; Zhan et al., 2018; Zheng et al., 2017) .

With the recent renaissance of deep-learning methods in general, and of deep reinforcement learning (RL) in particular, considerable attention has been devoted to developing analysis tools that allow researchers to understand what drives agents' behavior, what are the most common failure modes and provide insights into the inner workings of learning systems (Zeiler & Fergus, 2013; Yosinski et al., 2015; Olah et al., 2017; Morcos et al., 2018) . In particular, Rabinowitz et al. (2018) embed entire behavioral trajectories of single RL agents as points in an unstructured, high-dimensional space, relying on the inherent structure of the data to yield interpretable representations of whole behavioral motifs.

Finally, coordination in multi-agent system has been a topic of major interest as of late and some of the most advanced MARL systems rely on the emergence of coordination among teammates to complete the task at hand (Jaderberg et al., 2018; Pachocki et al., 2018) . Despite these recent successes, coordination is still considered a hard problem and several attempts have been made to promote the emergence of coordination by relaxing some assumptions (Foerster et al., 2016; 2017; Sukhbaatar et al., 2016; Raileanu et al., 2018; He et al., 2016a) . Here we show that, by embedding RFM modules in RL agents, they can learn to coordinate with one another faster than baseline agents, analogous to imagination-augmented agents in single-agent RL settings (Hamrick et al., 2017; Pascanu et al., 2017; Weber et al., 2017) .

G t hid < l a t e x i t s h a 1 _ b a s e 6 4 = " N 3 D O N K a V T W i a O J Q f n k E e G T R 4 4 8 8 = " > A A A B + 3 i c b V D L S s N A F L 3 x W e s r 1 q W b Y B F c l U Q E X b g o u N B l B f u A N o b J Z N o O n Z m E m Y l Y Q n 7 F j Q t F 3 P o j 7 v w b J 2 0 W 2 n p g 4 H D O v d w z J 0 w Y V d p 1 v 6 2 V 1 b X 1 j c 3 K V n V 7 Z 3 d v 3 z 6 o d V S c S k z a O G a x 7 I V I E U Y F a W u q G e k l k i A e M t I N J 9 e F 3 3 0 k U t F Y 3 O t p Q n y O R o I O K U b a S I F d u 3 n Q Q T b g S I 8 l z 8 Y 0 y v P A r r s N d w Z n m X g l q U O J V m B / D a I Y p 5 w I j R l S q u + 5 i f Y z J D X F j O T V Q a p I g v A E j U j f U I E 4 U X 4 2 y 5 4 7 J 0 a J n G E s z R P a m a m / N z L E l Z r y 0 E w W I d W i V 4 j / e f 1 U D y / 9 j I o k 1 U T g + a F h y h w d O 0 U R T k Q l w Z p N D U F Y U p P V w W M k E d a m r q o p w V v 8 8 j L p n D U 8 t + H d n d e b V 2 U d F T i C Y z g F D y 6 g C b f Q g j Z g e I J n e I U 3 K 7 d e r H f r Y z 6 6 Y p U 7 h / A H 1 u c P u c + U 2 w = = < / l a t e x i t > < l a t e x i t s h a 1 _ b a s e 6 4 = " N 3 D O N K a V T W i a O J Q f n k E e G T R 4 4 8 8 = " > A A A B + 3 i c b V D L S s N A F L 3 x W e s r 1 q W b Y B F c l U Q E X b g o u N B l B f u A N o b J Z N o O n Z m E m Y l Y Q n 7 F j Q t F 3 P o j 7 v w b J 2 0 W 2 n p g 4 H D O v d w z J 0 w Y V d p 1 v 6 2 V 1 b X 1 j c 3 K V n V 7 Z 3 d v 3 z 6 o d V S c S k z a O G a x 7 I V I E U Y F a W u q G e k l k i A e M t I N J 9 e F 3 3 0 k U t F Y 3 O t p Q n y O R o I O K U b a S I F d u 3 n Q Q T b g S I 8 l z 8 Y 0 y v P A r r s N d w Z n m X g l q U O J V m B / D a I Y p 5 w I j R l S q u + 5 i f Y z J D X F j O T V Q a p I g v A E j U j f U I E 4 U X 4 2 y 5 4 7 J 0 a J n G E s z R P a m a m / N z L E l Z r y 0 E w W I d W i V 4 j / e f 1 U D y / 9 j I o k 1 U T g + a F h y h w d O 0 U R T k Q l w Z p N D U F Y U p P V w W M k E d a m r q o p w V v 8 8 j L p n D U 8 t + H d n d e b V 2 U d F T i C Y z g F D y 6 g C b f Q g j Z g e I J n e I U 3 K 7 d e r H f r Y z 6 6 Y p U 7 h / A H 1 u c P u c + U 2 w = = < / l a t e x i t > < l a t e x i t s h a 1 _ b a s e 6 4 = " N 3 D O N K a V T W i a O J Q f n k E e G T R 4 4 8 8 = " > A A A B + 3 i c b V D L S s N A F L 3 x W e s r 1 q W b Y B F c l U Q E X b g o u N B l B f u A N o b J Z N o O n Z m E m Y l Y Q n 7 F j Q t F 3 P o j 7 v w b J 2 0 W 2 n p g 4 H D O v d w z J 0 w Y V d p 1 v 6 2 V 1 b X 1 j c 3 K V n V 7 Z 3 d v 3 z 6 o d V S c S k z a O G a x 7 I V I E U Y F a W u q G e k l k i A e M t I N J 9 e F 3 3 0 k U t F Y 3 O t p Q n y O R o I O K U b a S I F d u 3 n Q Q T b g S I 8 l z 8 Y 0 y v P A r r s N d w Z n m X g l q U O J V m B / D a I Y p 5 w I j R l S q u + 5 i f Y z J D X F j O T V Q a p I g v A E j U j f U I E 4 U X 4 2 y 5 4 7 J 0 a J n G E s z R P a m a m / N z L E l Z r y 0 E w W I d W i V 4 j / e f 1 U D y / 9 j I o k 1 U T g + a F h y h w d O 0 U R T k Q l w Z p N D U F Y U p P V w W M k E d a m r q o p w V v 8 8 j L p n D U 8 t + H d n d e b V 2 U d F T i C Y z g F D y 6 g C b f Q g j Z

g e I J n e I U 3 K 7 d e r H f r Y z 6 6 Y p U 7 h / A H 1 u c P u c + U 2 w = = < / l a t e x i t > < l a t e x i t s h a 1 _ b a s e 6 4 = " N 3 

D O N K a V T W i a O J Q f n k E e G T R 4 4 8 8 = " > A A A B + 3 i c b V D L S s N A F L 3 x W e s r 1 q W b Y B F c l U Q E X b g o u N B l B f u A N o b J Z N o O n Z m E m Y l Y Q n 7 F j Q t F 3 P o j 7 v w b J 2 0 W 2 n p g 4 H D O v d w z J 0 w Y V d p 1 v 6 2 V 1 b X 1 j c 3 K V n V 7 Z 3 d v 3 z 6 o d V S c S k z a O G a x 7 I V I E U Y F a W u q G e k l k i A e M t I N J 9 e F 3 3 0 k U t F Y 3 O t p Q n y O R o I O K U b a S I F d u 3 n Q Q T b g S I 8 l z 8 Y 0 y v P A r r s N d w Z n m X g l q U O J V m B / D a I Y p 5 w I j R l S q u + 5 i f Y z J D X F j O T V Q a p I g v A E j U j f U I E 4 U X 4 2 y 5 4 7 J 0 a J n G E s z R P a m a m / N z L E l Z r y 0 E w W I d W i V 4 j / e f 1 U D y / 9 j I o k 1 U T g + a F h y h w d O 0 U R T k Q l w Z p N D U F Y U p P V w W M k E d a m r q o p w V v

A N n J C p Z r U C R M 6 Q Z R f O 8 5 b + M 6 0 = " > A A A B / H i c b V D L S s N A F J 3 4 r P U V 7 d J N s A i u S i K C L l w U X O i y g n 1 A G 8 N k O m m H z k z C z I 1 Q Q v w V N y 4 U c e u H u P N v n L R Z a O

u B g c M 5 9 z L n n j D h T I P r f l s r q 2 v r G 5 u V r e r 2 z u 7 e v n 1 w 2 N F x q g h t k 5 j H q h d i T T m T

t A 0 M O O 0 l i m I R c t o N J 9 e F 3 3 2 k S r N Y 3 s M 0 o b 7 A I 8 k i R j A Y K b B r N w 8 Z 5 E E 2 E B j G S m R M 5 n l g 1 9 2 G O 4 O z T L y S 1 F G J V m B / D Y Y x S Q W V Q D j W u u + 5 C f g Z V s A I p 3 l 1 k G q a Y D L B I 9 o 3 V G J B t Z / N w u f O i V G G T h Q r 8 y Q 4 M / X 3 R o a F 1 l M R m s k i o 1 7 0 C v E / r 5 9 C d O m b g 5 I U q C T z j 6 K U O x A 7 R R P O k C l K g E 8 N w U Q x k 9 U h Y 6 w w A d N X 1 Z T g L Z 6 8 T D p n D c 9 t e H f n 9 e Z V W U c F H a F j d I o 8 d I G a 6 B a 1 U B s R N E X P 6 B W 9 W U / W i / V u f c x H V 6 x y p 4 b + w P r 8 A d H Q l X 8 = < / l a t e x i t > < l a t e x i t s h a 1 _ b a s e 6 4 = " p o A N n J C p Z r U C R M 6 Q Z R f O 8 5 b + M 6 0 = " > A A A B / H i c b V D L S s N A F J 3 4 r P U V 7 d J N s A i u S i K C L l w U X O i y g n 1 A G 8 N k O m m H z k z C z I 1 Q Q v w V N y 4 U c e u H u P N v n L R Z a O

u B g c M 5 9 z L n n j D h T I P r f l s r q 2 v r G 5 u V r e r 2 z u 7 e v n 1 w 2 N F x q g h t k 5 j H q h d i T T m T c) RFM-augmented agents, the output of the the RFM module is appended to the original observation input to the policy network. The on-board RFM module is trained with full supervision and alongside the policy network.

t A 0 M O O 0 l i m I R c t o N J 9 e F 3 3 2 k S r N Y 3 s M 0 o b 7 A I 8 k i R j A Y K b B r N w 8 Z 5 E E 2 E B j G S m R M 5 n l g 1 9 2 G O 4 O z T L y S 1 F G J V m B / D Y Y x S Q W V Q D j W u u + 5 C f g Z V s A I p 3 l 1 k G q a Y D L B I 9 o 3 V G J B t Z / N w u f O i V G G T h Q r 8 y Q 4 M / X 3 R o a F 1 l M R m s k i o 1 7 0 C v E / r 5 9 C d O m b g 5 I U q C T z j 6 K U O x A 7 R R P O k C l K g E 8 N w U Q x k 9 U h Y 6 w w A d N X 1 Z T g L Z 6 8 T D p n D c 9 t e H f n 9 e Z V W U c F H a F j d I o 8 d I G a 6 B a 1 U B s R N E X P 6 B W 9 W U / W i / V u f c x H V 6 x y p 4 b + w P r 8 A d H Q l X 8 = < / l a t e x i t > < l a t e x i t s h a 1 _ b a s e 6 4 = " p o A N n J C p Z r U C R M 6 Q Z R f O 8 5 b + M 6 0 = " > A A A B / H i c b V D L S s N A F J 3 4 r P U V 7 d J N s A i u S i K C L l w U X O i y g n 1 A G 8 N k O m m H z k z C z I 1 Q Q v w V N y 4 U c e u H u P N v n L R Z a O u B g c M 5 9 z L n n j D h T I P r f l s r q 2 v r G 5 u V r e r 2 z u 7 e v n 1 w 2 N F x q g h t k 5 j H q h d i T T m T t A 0 M O O 0 l i m I R c t o N J 9 e F 3 3 2 k S r N Y 3 s M 0 o b 7 A I 8 k i R j A Y K b B r N w 8 Z 5 E E 2 E B j G S m R M 5 n l g 1 9 2 G O 4 O z T L y S 1 F G J V m B / D Y Y x S Q W V Q D j W u u + 5 C f g Z V s A I p 3 l 1 k G q a Y D L B I 9 o 3 V G J B t Z / N w u f O i V G G T h Q r 8 y Q 4 M / X 3 R o a F 1 l M R m s k i o 1 7 0 C v E / r 5 9 C d O m b g 5 I U q C T z j 6 K U O x A 7 R R P O k C l K g E 8 N w U Q x k 9 U h Y 6 w w A d N X 1 Z T g L Z 6 8 T D p n D c 9 t e H f n 9 e Z V W U c F H a F j d I o 8 d I G a 6 B a 1 U B s R N E X P 6 B W 9 W U / W i / V u f c x H V 6 x y p 4 b + w P r 8 A d H Q l X 8 = < / l a t e x i t > < l a t e x i t s h a 1 _ b a s e 6 4 = " p o A N n J C p Z r U C R M 6 Q Z R f O 8 5 b + M 6 0 = " > A A A B / H i c b V D L S s N A F J 3 4 r P U V 7 d J N s A i u S i K C L l w U X O i y g n 1 A G 8 N k O m m H z k z C z I 1 Q Q v w V N y 4 U c e u H u P N v n L R Z a O u B g c M 5 9 z L n n j D h T I P r f l s r q 2 v r G 5 u V r e r 2 z u 7 e v n 1 w 2 N F x q g h t k 5 j H q h d i T T m T t A 0 M O O 0 l i m I R c t o N J 9 e F 3 3 2 k S r N Y 3 s M 0 o b 7 A I 8 k i R j A Y K b B r N w 8 Z 5 E E 2 E B j G S m R M 5 n l g 1 9 2 G O 4 O z T L y S 1 F G J V m B / D Y Y x S Q W V Q D j W u u + 5 C f g Z V s A I p 3 l 1 k G q a Y D L B I 9 o 3 V G J B t Z / N w u f O i V G G T h Q r 8 y Q 4 M / X 3 R o a F 1 l M R m s k i o 1 7 0 C v E / r 5 9 C d O m b g 5 I U q C T z j 6 K U O x A 7 R R P O k C l K g E 8 N w U Q x k 9 U h Y 6 w w A d N X 1 Z T g L Z 6 8 T D p n D c 9 t e H f n 9 e Z V W U c F H a F j d I o 8 d I G a 6 B a 1 U B s R N E X P 6 B W 9 W U / W i / V u f c x H V

2.1.1. RELATIONAL FORWARD MODELS AND BASELINES ARCHITECTURES:

A GN is a neural network that operates on graphs. The input to a GN is a directed graph, (u, V, E), where u ∈ R du is a graph-level attribute vector (e.g. the score in a football game), V = {v i } i=1:nv is a set of vertices (e.g. the players) with attributes v i ∈ R dv (e.g the players' (x,y) coordinates on the pitch), and E = {(e k , r k , s k )} k=1:ne is a set of directed edges which connect sender vertex, v s k to receiver vertex v r k and have attribute e k ∈ R de (e.g. same team or opponent). The output of a GN is also a graph, with the same connectivity structure as the input graph (i.e., same number of vertices and edges, as well as same sender and receiver for each edge), but updated global, vertex, and edge attributes. See Fig. 1b for an example graph.

The sequence of computations in a GN proceed by updating the edge attributes, followed by the vertex attributes, and finally the global attributes. These computations are implemented via three "update" functions (the φs) and three "aggregation" functions (the ρs),

e k = φ e (e k , v r k , v s k , u) , v i = φ v (ē i , v i , u) , u = φ u (ē , v , u) , ē i = ρ e→v (E i ) , v = ρ v→u (V ) , ē = ρ e→u (E ) (1)

where

E i = {(e k , r k , s k )} r k =i .

The edges are updated by φ e , as a function of the sender vertex, receiver vertex, edge, and global attributes. We term the updated edge attribute the "message", e k . Next, each vertex, i, is updated by aggregating the e k messages for which i = r k , and computing the updated vertex attribute, v i , as a function (φ v ) of these aggregated messages, as well as the current vertex and global attributes. Finally, the global attributes are updated, by φ u , as a function of the current global attribute and all aggregated e k and v i attributes.

Since a GN takes as input a graph and outputs a graph, GN blocks can be composed to form more complex, and powerful, architectures. These architecture can also be made recurrent in time by introducing a state graph, and using recurrent neural networks (RNNs) as the φ functions. GN-based architectures can be optimized with respect to some objective function by gradient descent (using backpropagation through time for recurrent implementations). Here we focus on supervised learning using datasets of input-output pairs. See (Battaglia et al., 2018) for further details.

We construct our RFM architecture by arranging three GN blocks as in Fig. 1a . We selected this specific architecture to allow our model to perform relational reasoning steps both on the raw input data, before time recurrence is included, and then again on the output of our time recurrent block. This allows the recurrent block to construct memories of the relations between entities and not simply of their current state.

Architecture details are as follows: input graphs G t in go through a GN encoder block, a basic GN module whose φ v , φ e and φ u are three separate 64-unit MLPs, with 1 hidden layer, and ReLU activations and whose ρ functions are summations. The output of the GN encoder block is used, in conjunction with a state graph G t-1 hid , in a "GraphGRU", where each φ function is a Gated Recurrent Unit (GRU) (Cho et al., 2014) with a hidden state size of 32 for each of vertices, edges and globals. The GraphGRU's output is then copied into a state graph and an output graph. The state graph is used in the following time step, while the output graph is passed through a GN decoder block. This last block's structure has an identical to the GN encoder's, and outputs the model's predictions (e.g. the actions of each agent).

We compared the prediction performance of our RFM module to two state-of-the-art relational reasoning baselines: Neural Relational Inference networks (Kipf et al., 2018) and Vertex Attention Interaction Networks (Hoshen, 2017) . These architectures are similar to our RFM module. In particular, NRI models operate on graph structured data and, with the exception that the graph connectivity map is not given, but rather estimated from trajectories using an auto-encoder architecture, they are identical to our model. VAIN networks are essentially single feed-forward GN blocks where the φ e and ρ e→v functions take particular and restricted form:

φ e (e k , v r k , v s k , u) = e a(vr k )-a(vs k ) 2 and ρ e→v (E i ) = v i s k e k , with a(•) a learnable function.

We also compared our full RFM against ablated variants, which allowed us to measure the importance of the relational reasoning component, and of time recurrence. In particular we considered a Feedforward model, which had no GraphGRU block, and a No-relation model, which was a full fledged RFM module but operated on graphs with only self-connections (i.e., edges where s i = r i ). Finally, we included a vector-based MLP + LSTM model among our baselines, so as to highlight the advantage of using graphs over vector based modules. This last model operated on the concatenation of the vertex attributes and had a standard Encoder MLP (64-units), LSTM (32-hidden units), Decoder MLP (2 hidden layers, 32-units each) architecture. We matched all models for capacity (with the exception of NRI which has about 3x more parameters than other models because of its autoencoder connectivity map estimator). Models were within 3% of each other in terms of number of parameters (as reported by the TensorFlow checkpoint loader).

2.1.2. MARL ENVIRONMENTS AND AGENT ARCHITECTURE:

We considered three multi-agent environments for our study: Cooperative Navigation (Lowe et al., 2017) , Coin Game (Raileanu et al., 2018) and Stag Hunt (Peysakhovich & Lerer, 2017b) .

Cooperative Navigation (Lowe et al., 2017) . Two agents navigate an empty 6 × 6 arena to cover two tiles. A reward of +1 is given to both agents whenever both tiles are covered, i.e., when each agent is on a tile of its own. Episodes are of fixed length (20 environment steps), to encourage a swift resolution of the underlying assignment problem. The positions of both tiles and the starting positions of each agent are randomized at the start of each episode.

Coin Game (Raileanu et al., 2018) . Two agents roam an 8 × 8 arena populated with 12 coins, 4 of each of 3 colors, for 10 environment steps. Agents can collect coins by stepping on them; out of the 3 coin colors, two colors carried a reward and one a punishment. Crucially, each of the two agents only has access to information about 1 good color. The short episode duration incentivizes agents to quickly infer what the unknown good color is by observing their teammate actions, so that all good coins can be collected. At the end of each episode both agents are rewarded according to how many good coins have been collected by either agent. Conversely, they are penalized according to the number of bad coins collected, again by either agent. The role of each color, coin positions, and starting coordinates for the agents are randomized in each episode.

Stag Hunt (Peysakhovich & Lerer, 2017b) . We implemented a Markov version of the classic Stag Hunt game where two (or four) agents navigate an arena populated with 3 red Stags (each of which is static, and occupies a 2 × 2 tile) and 12 green apples, for 32 environment steps. Agents can collect apples by themselves for a reward of +1 or, by both stepping on the same stag, capture it for a reward of +10. Collected apples and captured stags became unavailable for some time (denoted by dimmed colors), and at each time step have a small probability of becoming available again. All entities' locations are randomized at the start of each episode.

We trained populations of RL agents to convergence on these three tasks using a multi-agent implementation of importance-weighted actor-learner (Jaderberg et al., 2018; Espeholt et al., 2018) , a batched advantage actor-critic (A2C) algorithm. For each episode, a group of agents were randomly sampled, with replacement, from a population of 4 learners; at each time step agents received an ego-centric, top-down view of the environment which was large enough to contain the entire arena, and, in the Coin Game, one of the 2 good colors. Agents then selected one of 5 actions to be performed (move left, move right, move up, move down, and stay). Within each agent, the input image was parsed by a single convolutional layer (3 × 3-filters, 6 output channels) whose output was fed to a 256-unit single-layer MLP. The MLP output vector was concatenated with each player's last reward, and one-hot encoded last action, as well as, for the Coin Game, one of the two coin colors that carried a positive reward. The resulting vector served as input to a 256-hidden-units LSTM whose output was fed into a single soft-max layer. Throughout the learning process, there was no sharing of weights, gradients or any communication channel between the agents, consistent with standard MARL settings.

2.1.3. OFFLINE TRAJECTORIES COLLECTION AND MODEL TRAINING:

To train the forward RFM model, we collected 500,000 episodes of behavioral trajectories of trained agents acting in their respective environments. At each time step, we collected a semantic description of the state of the environment, as well as the action taken by each agent and the reward they received. These descriptions were compiled into a graph, where agents and static entities (i.e., apples, stags, coins, and tiles) were represented by vertices whose attributes, v i , were: the entity's position in the arena; the one-hot encoded type of the entity (e.g. agent, apple, etc.); (when applicable) the entity's state (e.g. available / collected); and (when applicable) the last action taken. When attributes were not applicable (e.g. the last action of an apple), we padded the corresponding attribute features with zeros. Edges connected all non-agent entities to all agents as well as agents to each other. Input edges contained no attributes and were characterized by senders and receivers only (see Fig. 1b for an example environment graph). In order to understand our analysis contributions, it is crucial to note that while the input graph to our RFM module contained no edge attributes, and edges were simply characterized by their sender and receiver vertices, the edges of a RFM's output graph did contain attributes. These attributes were computed by the network itself and amounted to distributed representations of the effect the sender entity had on the receiver agent.

We also collected 2,500 further episode trajectories for performance reporting and analysis. Training of both RFM and baseline models was conducted using gradient descent to minimize the cross-entropy loss between predicted and ground-truth actions. The training procedure was halted after one million steps, during each of which the gradient was estimated using a batch of 128 episodes. Results are presented in Sec. 2.2.1.

2.2. RESULTS:



2.2.1. ACTION PREDICTION PERFORMANCE:

We trained our RFM modules and baseline models to predict the actions of each agent in each of the three games we considered. Models were given a graph representation of the state of the environment, and produced an action prediction for each agent. After training (see Sec. 2.1.3), we used held-out episodes to assess the performance of each model in terms of mean length of perfect roll-out: the mean number of steps during which prediction and ground truth do not diverge. This metric gives us a measure of how long we could simulate the agents' behavior before making a mistake. For completeness, we report next-action classification accuracy in Sec. A.5.

Results are shown in Fig. 2 . As expected, all models achieve similar scores on the Coop Nav game, which is a rather simple environment. Our RFM module outperforms the NRI baseline by a substantial margin on the Coin Game and Stag Hunt environments. Since the two models are identical, except for the initial graph structure inference step, this result suggests that when the importance of some relations is revealed over time, rather than obvious from the start, the graph structure inference step proposed in NRI might not be appropriate. Our RFM consistently outperforms the VAIN model, and on Stag Hunt our Feedforward model does as well. This indicates that, for this particular task, distributed interaction representations are superior to simple attention weights. Finally, the MLP+LSTM and No-relation models performed worst across the board, which suggests that relations between entities, rather than the state of the entities themselves, carry most of the predictive power in these environments.

These results reproduce and advance the conclusion that relational models can be trained to perform action prediction for multi-agent systems, and are superior to non-relational models for this task (Kipf et al., 2018; Hoshen, 2017) .

2.2.2. RELATIONAL ANALYSIS OF THE STAG HUNT GAME: ACTIONS:

Here we introduce our relational analysis tools and use the Stag Hunt game as a case study. While we illustrate our findings on a simple game, these intuitions can be easily transferred to more complex domains.

We propose the Euclidean norm of a message vector (i.e., e k ) as a measure of the influence a sender entity, v s k , has on a receiver, v r k . We validate this suggestion in Fig. 3 (top row), where we show that the edge norm between a sender entity (either a stag or an apple) and a receiver agent is predictive of which entity the agent will move towards, or away from, at the next time step.

This intuition can be developed to discover the events that qualitatively change agents' behavior, as well as the factors that mediate how agents interact with one another. Fig. 3 (middle row), for example, shows how the norm of an edge between a stag and an agent changes over time. The importance of the relation is modulated by the prey's state: when a stag becomes available, the edge norm rises substantially; when a stag is consumed, the edge norm drops. Remarkably, the presence or absence of a stag also influences the edge norm between the two teammates, as shown in Fig. 3 (bottom row): in the time step immediately before they consume a stag, the edge between the two teammates is higher than immediately afterwards. In contrast, this effect does not occur with apples, which do not require coordination between teammates to consume. Finally, as shown in Fig. 3 (bottom row), we find that agents' influence on each other's behavior is higher when there is a scarcity of apples (as agents compete for this resource). We note that while significant changes in edge norm or the rank order of edge norm can be used to discover events that qualitatively change agents behavior and factors that mediate agents' social interaction, the raw values have no intrinsic meaning.

Taken as a whole, these findings highlight how the norm of the edge messages, computed by a RFM which is trained to predict the future actions in a multi-agent system, contain intepretable and quantifiable information about when and how certain entities and relations influence agents' behavior, and about which entities and situations mediate the social influence between agents.

2.2.3. RELATIONAL ANALYSIS OF THE STAG HUNT GAME: RETURN:

A second key finding is that beyond measuring the intensity of a social influence relation, RFM modules can also be used to quantify their valence. We trained a RFM model to predict the return received by each agent (until the end of the episode), rather than their future action. We used this model to measure the marginal utility of the actual social context, i.e., to ask: what would happen to agent 1's return if we didn't know the exact state of agent 2? On the y-axis we plot the average relative displacement between the agent (receiver) and an entity (sender), we order the plot by the rank of the edge activation magnitude; predictive power declines sharply with rank. Middle-row: edge activations discover what agents care about and how this changes over time, here we have time series plots (left and right) of an edge activation norm when a stag becomes available and unavailable, and averages over all time steps grouped by stag state (middle). Bottom row: when stags become available, agents care about each other more than just before that happens, (p < 0.05; middle). Apples becoming available has no effect (p = 0.27; middle). See also control experiments in Fig. 9 . The norm of the edge connecting the two agents is also modulated by scarcity (right), agents compete for apple consumption and the fewer apple there are, the more the two agents influence each other behavior (r = -0.39, p < 0.05).

The proposed approach is to effectively compare two estimators for agent 1's return:

EQUATION

EQUATION

≈ M (s a1 , s a2 , z)p(s a2 |s a1 , z)ds a2 Full graph -Ra1

Pruned Graph right before and right after a stag is captured: agents' influence are most beneficial for each other when they a capture a stag. Episodes ran for 128 steps for this analysis.

where Ra1

Full graph is the model M 's estimate of the return received by agent 1, given the state of both agents 1 and 2, and all other environment variables, z, whereas Ra1

Pruned graph is that same estimate, without knowledge of the state of agent 2 (i.e., marginalizing out s a2 ). In practice, this latter estimate can be obtained by removing the edge connecting the two agents from the input graph 1 .

If we find that, in certain situations, the predicted return decreases when removing information about agent 2 (i.e., Ra1

Full graph > Ra1 Pruned graph ), we would conclude that the actual state of agent 2 results in a better-than-expected return for agent 1, that is, agent 2 is helping agent 1. Conversely, if the predicted return increases we would conclude that agent 2 is hindering agent 1.

We ran this experiment using a set-up identical to the one we used for action prediction, except for three modifications: (1) the target variable and (2) loss function were changed, from cross-entropy between predicted and ground-truth actions, to mean squared error between predicted and true return; and (3) the training set contained an equal proportion of environment graphs with and without edges between teammates. The latter modification ensured that the pruned-graph computations were not out-of-distribution. The ground truth and predicted return (using both the full and pruned graph) for a sample episode are shown in Fig. 4 (left) .

We note that within this setup, both the pruned-graph estimator and the full-graph estimator are produced by a single graph neural network. This network is trained to predict agent 1's return both using the full graph (i.e. knowing the actual state of a 2 ) and the pruned graph (i.e. not knowing the actual state of a 2 ). During training we randomly drop out edges between teammates (to ensure that both full graph and pruned graph are in-distribution for M ). At test time, we then compute the full-graph estimate by using all edges, and the pruned-graph estimator by dropping out edges between teammates.

Similar to the edge-norm relational analysis above, we can find the entities and events that mediate the value of a social interaction. For example, Fig. 4 (middle and right) show the marginal value of a teammate's particular state (i.e. Ra1

Full graph -Ra1 Pruned Graph ) over time and around the time of a stag capture. Thus the model estimates that teammates' specific interactions during this time are beneficial to their return. This also scales to different number of agents. Importantly, the on-board RFM module is trained alongside the policy network, and there is no sharing of parameters or gradients between the agents. We also show curves for training alongside learning teammates in Fig. 8 . Embedding an RFM is also more beneficial than embedding an MLP+LSTM (see Fig. 7 .)

3 RFM-AUGMENTED AGENTS

3.1. METHODS:

We have shown that relational reasoning modules capture information about the social dynamics of multi-agent environments. We now detail how these modules' predictions can be useful for improving MARL agents' speed of learning. We extended the agent architecture (described in Sec. 2.1.2) by embedding a RFM module in each agent, and augmenting the policy network's observations with the RFM's output. This agent architecture is depicted in Fig. 1c .

Incorporating an on-board RFM module did not provide the agents with any additional information above and beyond that provided to baseline agents. All games were fully observable, so the additional inputs (i.e. the true last action, and the environment graph, which was provided as input to the embedded RFM) did not add any new information to the original egocentric observations. Similarly, the on-board RFM modules were trained from scratch alongside the policy networks, while the agents were learning to act, so that no additional game structure was given to the augmented agents. Finally, we highlight that each learning agent in the arena had its own RFM module and policy networks; there was never any sharing of weights, gradients or communication between the agents.

Our baseline agent policy network architecture comprised of a CNN that processed the actor's egocentric observation, followed by a MLP+LSTM network that provided action logits (see Sec. 2.1.2 for architecture details). Our augmented agents had an embedded RFM module, which was fed graph representations of the state of the environment, just as in the offline RFM modules in the forward modeling experiments. We trained this module to minimize the cross-entropy loss between its prediction and the last action taken by all fellow agents. We used the prediction output of the on-board RFM module to augment the observation stream at the input of the original policy network. Specifically, the output of the RFM module-predicted action logits for fellow agents-was rendered as image planes whose pixel intensity was proportional to the estimated probability that an agent would be at a certain location at the next time step2 . These image planes were appended to the ego-centric top-down observation and fed to the original policy network.

Our experimental design was relatively straightforward. First, we trained A2C agents (as described in Sec. 2.1.2) to play the three games we considered, as well as a four-player variant of the Stag Hunt game. Second, we paired learning agents with these pre-trained experts: learning agents occupied a single-player slot in each game, while all their teammates were pre-trained experts. We repeated this procedure using both RFM-enhanced agents and baseline A2C agents as learners. During training we recorded the reward received by the singular learning agent in each episode.

Our results show that agents that explicitly model each other using an on-board RFM learn to coordinate with one another faster than baseline agents (Fig. 5 ). In Stag Hunt our RFM-augmented agent achieves a score above 25 after around 600K steps, while baseline agents required around 1M steps. This effect is even more prominent in the 4-player version of the game where these scores are achieved around 500K and 1M steps respectively. Similarly in Coop Nav baseline agents required twice as many steps of experience to consistently score above 25 as our RFM-augmented agents. Moreover, in the Coin Game environment, the faster learning rate of RFM-augmented agents appears to be due to a superior efficiency in learning to interpret the teammate's action and infer the negative coin color in each episode (see Sec. A.1). Finally, we found that augmenting agents with on-board RFM modules was more beneficial to agents learning than using MLP + LSTM models (see Sec. A.2). These results suggest that agents take into account the on-board RFM's predictions when planning their next action, and that this results in agents that learn faster to coordinate with others, and to discover others' preferences from their actions.

4. CONCLUSIONS:

Here we showed that our Relational Forward Model can capture the rich social dynamics of multiagent environments, that its intermediate representations contained valuable interpretable information, and that providing this information to learning agents results in faster learning system.

The analysis tools we introduced allow researchers to answer new questions, which are specifically tailored to multi-agent systems, such as what entities, relations and social interactions drive agents' behaviors, and what environment events or behavior patterns mediate these social and non-social influence signals. Importantly our methods require no access to agents internals, only to behavioral trajectories, making them amenable to analyzing human behavior, sports and ecological systems.

Providing agents with access the output of RFM modules results in agents that learn to coordinate with one another faster than non-augmented baselines. We posit that explicit modeling of teammates and opponents is an important research direction in multi-agent RL, and one that might alleviate the need for communication, parameter sharing or centralized controllers to achieve coordination. Future work will see our methods applied to more complex and varied domains where artificial and non-artificial agents interact and learn in shared environments. We will focus on identifying entire patterns of behavior for in-agent modeling, so as to adapt the host agent policy more efficiently. when artificial learning agents interact with human experts. For completeness, we show in Fig. 8 the corresponding results when RFM-augments agents train alongside other learning agents (i.e. non-experts). In this case, either all agents in the environment were RFM-augmented (green), or all agents were baseline (blue). We use longer episodes in these experiments (128 steps, rather than 32) in order to make training easier (hence total returns were higher). For brevity, we only report results on the two-player versions of the games.

We see a similar result to the main text: allowing agents to model one another explicitly results in faster learning (e.g. in StagHunt RFM + A2C achieves scores around 48 around 30M steps while vanilla A2C requires 45M training steps. Similarly in CoinGame RFM + A2C achieves a score around 15 in 100M steps while vanilla A2C requires almost 200M steps). We note that this setting presents an additional challenge: a learned model of a teammate's behavior can only provide useful information for coordination after the teammate's policy becomes sensible. The advantage conferred by embedding the RFM into the learning agent will thus be delayed relative to the expert teammate condition shown in the main text. Nonetheless, augmenting agents with RFM models still results in faster learning than omitting it altogether.

A.4 CONTROL EXPERIMENTS: WHICH EVENTS MEDIATE INTERDEPENDENT BEHAVIOR:

In the main text we showed that our RFM model reveals how agents' influence on each other is contextual. In particular, we observe in Fig: 3 (bottom row, left and middle panels) that the Euclidean norm of the activation of the edge between the two agents increases when a stag appears. We argue that this indicates that agents coordinate their behavior when stags are available. Here we report control experiments to test alternative hypotheses. In these experiments, we trained agents on two modifications of the StagHunt game, wherein there is no explicit incentive for agents to coordinate:

• Stags carry no rewards. This tests whether the changes in Fig. 3 could be due to arbitrary changes in the environment. Here our hypothesis predicts that stag appearance would have no effect on edge norms, since agents should learn to ignore them. • Stags can be collected by a lone hunter. This tests whether the changes in Fig. 3 could be due to the appearance of new reward-carrying objects that do not require coordination. Again, our hypothesis predicts that stags would have no effect on the edge norm between agents, since collecting a stag does not require coordination, like apples.

In both cases our experimental pipeline was as described in the main text: (1) train A2C agents on the modified version of the StagHunt game;

(2) collect behavioral trajectories from these agents;

(3) train a RFM model to predict the action of each agent given the state of the environment; and (4) report the Euclidean norm of the edge activations in the link connecting the two agents at a time just before and just after a Stag or an Apple appear.

Consistent with our primary hypothesis, there is no significant change in the norm of the activations in the edge connecting two agents when Stags (or apples) appear in these situation. This provides evidence that our RFM model is able to reveal which events in the environment mediate how agents influence each one another. In the main text we show that our RFM model provides longer perfect rollouts than competing models.

Here we provide an alternative metric to measure the relative accuracy of different models. In Fig. 10 , we show the next-step action classification accuracy, in the same manner as Fig. 2 . Model ranking remain unchanged (with the exception of NRI on CoinGame): the RFM outperforms other models on predicting agent behavior.

:

It is worth highlighting that even though the edge from agent

is removed, the estimator Ra 1 Pruned graph can implicitly take advantage of sa 1 and z to effectively form a posterior over sa 2 before marginalizing it out. For this reason, we include p(sa 2 |sa 1 , z) rather than p(sa 2 ) in equation 2.2.3.2 For example, consider a fellow agent at the center of the map, and prediction logits indicating that, at the next time step, it might move up with a probability of 0.3, and down with a probability of 0.7. The additional image plane would be zero everywhere, with the exception of the pixel above the center (which would have a value of 0.3) and the one below the center (which would have an value of 0.7).

Figures and Tables:

8 8 j L p n D U 8 t + H d n d e b V 2 U d F T i C Y z g F D y 6 g C b f Q g j Z g e I J n e I U 3 K 7 d e r H f r Y z 6 6 Y p U 7 h / A H 1 u c P u c + U 2 w = = < / l a t e x i t > G t in < l a t e x i t s h a 1 _ b a s e 6 4 = " p o

Figure 1: (a) The RFM module stacks a GN Encoder, a Graph GRU and a GN Decoder to obtain a relational reasoning module that holds state information across time steps. (b) Example of an environment graph representation. Edges connect agents (magenta and orange) to all entities and are color-coded according to the identity of the receiver. (c) RFM-augmented agents, the output of the the RFM module is appended to the original observation input to the policy network. The on-board RFM module is trained with full supervision and alongside the policy network.

Figure2: Action prediction performance of our RFM module and baseline models. The reported quantity is the mean number of environment steps for which the predicted actions matched the ground truth exactly, for all agents. Mean across 128 episodes, bars indicate standard deviation across episodes. Alternative measures of model performance show similar results (Fig.10).

(a) Edge activation magnitude is predictive of future behavior. (b) Edge activation magnitude reveals changes in what drives agents behavior over time.(c) Edge activation magnitude discovers situations that alter agents' social influence.

Figure3: Edge analysis. Top-row: the norm of output edge activations is predictive of future behavior. On the y-axis we plot the average relative displacement between the agent (receiver) and an entity (sender), we order the plot by the rank of the edge activation magnitude; predictive power declines sharply with rank. Middle-row: edge activations discover what agents care about and how this changes over time, here we have time series plots (left and right) of an edge activation norm when a stag becomes available and unavailable, and averages over all time steps grouped by stag state (middle). Bottom row: when stags become available, agents care about each other more than just before that happens, (p < 0.05; middle). Apples becoming available has no effect (p = 0.27; middle). See also control experiments in Fig.9. The norm of the edge connecting the two agents is also modulated by scarcity (right), agents compete for apple consumption and the fewer apple there are, the more the two agents influence each other behavior (r = -0.39, p < 0.05).

Figure 4: Return analysis: we trained our RFM model to predict the return (until the end of the episode) received by each agent. We trained our model on graphs with and without edges connecting the two agents. (Left) ground truth and predicted return (using both graphs) for a sample episode. (Middle) Ra1 Full graph -Ra1 Pruned Graph around the time a stag is captured. Positive value indicates that the model estimates that the social influence has a positive marginal utility. (Right) Ra1Full graph -Ra1

Figure 5: Training curves for A2C agents with and without on-board RFM modules. Allowing agents to access the output of a RFM module results in agents that learn to coordinate faster than baseline agents.This also scales to different number of agents. Importantly, the on-board RFM module is trained alongside the policy network, and there is no sharing of parameters or gradients between the agents. We also show curves for training alongside learning teammates in Fig.8. Embedding an RFM is also more beneficial than embedding an MLP+LSTM (see Fig.7.)

Figure 8: Agents training with non-expert teammates. Reward shown as the average return per agent, averaged over four agent seeds.

Figure 9: If coordination is not required to collect stags, or if agents are not interested in collecting stags, the edge norm between the two agents is not affected by the appearance of available stags. (Compare to Fig. 3 bottom row left and middle panels.)



Figure 10: Next-step action classification accuracy.

Peer review (iclr2019):
 This paper used graph neural networks to do relational reasoning of multi-agent systems to predict the actions and returns of MARL agents that they call Relational Forward Modeling. They used RFM to analyze and assess the coordination between agents in three different multi-agent environments. They then constructed an RFM-aumented RL agent and showed improved training speeds over non relational reasoning baseline methods. I think the overall approach is interesting and a novel way to address the growing concern of how to access coordination between agents in multi-agent systems. I also like how they authors immediately incorporated the relational reasoning approach to improve the training of the MARL agents. I wonder how dependent this approach is to the semantic representation of the environment. These semantic descriptions are similar to hand crafted features and thus will require some prior knowledge about the environment or task and will be harder to obtain on more difficult environment and tasks. Will this approach work on continuous tasks? For example, the continuous state and action space of the predator-prey tasks that use the multi-agent particle environment from OpenAi. I think one of the biggest selling points from this paper is using this method to assess the coordination/collaboration between agents (i.e. the social influence amongst agents). I would have liked to see more visualizations or analysis into these learned representations. The bottom row of Figure 3 shows that "when stags become available, agents care about each other more than just before that happens". While this is very interesting and an important result, i think that this allows one to see what features of the environment (including other agents) are important to a particular agents decision making but it doesn*t really answer whether the agents are truly coordinated, i.e. whether there are any causal dependencies between agents. For the RFM augmented agents, I like that you are able to train the policy as well as the RFM simultaneously from scratch, however, it seems that this requires you to only train a single agent in the multi-agent environment. If I understand correctly, for a given multi-agent environment, you first pre-trained A2C agents to play the three MARL games and then you paired one of the pre-trained (expert) agents with the RFM-augmented learning agents during training. This seems to limit the practicality and usability of this method as it requires you to have pre-trained agents that have already solved the task. I would like to know why the authors didn*t try to train two (or four) RFM-augmented agents from scratch together. When you use one of the agents as a pre-trained agent, this might make the training of the RFM module a bit easier since you have at least one agent with a fixed policy to predict actions from. It could be challenging when trying to train both RFM modules on two learning agents as the behaviors of learning agents are changing over time and thus the learning might be unstable. Overall, i think this is an interesting approach and especially for probing what information drives agents* behaviors. However, I don*t see the benefit of the RFM-augmented agent provides. It*s clearly shown to learn faster than non RFM-augmented agents (which is good), however, unless I*m mistaken, the RFM-augmented agent requires a pre-trained agent to be able to learn in the first place. --edit: The authors have sufficiently addressed my questions and concerns and have performed additional analysis. My biggest concern of weather or not the RFM-augmented agent was capable of learning without a pre-trained agent has been addressed with additional experiments and analysis (Figure 8). Based on this, i have adjusted my rating to a 7.