UPD E T: U NIVERSAL M ULTI - AGENT R EINFORCEMENT L EARNING VIA P OLICY D ECOUPLING WITH T RANSFORMERS
Anonymous authors
Paper under double-blind review
A BSTRACT
Recent advances in multi-agent reinforcement learning have been largely limited training one model from scratch for every new task.
This limitation occurs due to the restriction of the model architecture related to ﬁxed input and output dimensions, which hinder the experience accumulation and transfer of the learned agent over tasks across diverse levels of difﬁculty (e.g. 3 vs 3 or 5 vs 6 multiagent games).
In this paper, we make the ﬁrst attempt to explore a universal multi-agent reinforcement learning pipeline, designing a single architecture to ﬁt tasks with different observation and action conﬁguration requirements.
Unlike previous RNN-based models, we utilize a transformer-based model to generate a ﬂexible policy by decoupling the policy distribution from the intertwined input observation, using an importance weight determined with the aid of the selfattention mechanism.
Compared to a standard transformer block, the proposed model, which we name Universal Policy Decoupling Transformer (UPDeT), further relaxes the action restriction and makes the multi-agent task’s decision process more explainable.
UPDeT is general enough to be plugged into any multiagent reinforcement learning pipeline and equip it with strong generalization abilities that enable multiple tasks to be handled at a time.
Extensive experiments on large-scale SMAC multi-agent competitive games demonstrate that the proposed
UPDeT-based multi-agent reinforcement learning achieves signiﬁcant improvements relative to state-of-the-art approaches, demonstrating advantageous transfer capability in terms of both performance and training speed (10 times faster).
I NTRODUCTION
Reinforcement Learning (RL) provides a framework for decision-making problems in an interactive environment, with applications including robotics control (Hester et al.
(2010)), auto-driving (Bojarski et al.
and video gaming (Mnih et al.
Cooperative multi-agent reinforcement learning (MARL), a long-standing problem in the RL context, involves organizing multiple agents to achieve a goal, and is thus a key tool used to address many real-world problems, such as mastering multi-player video games (Peng et al.
and studying population dynamics (Yang et al.
A number of methods have been proposed that exploit an action-value function to learn a multiagent model (Du et al.
(2019), Mahajan et al.
(2019), Rashid et al.
, Zhou et al.
(2020), Yang et al.
(2020), Sunehag et al.
(2017), Hostallero et al.
However, current methods have poor representation learning ability and fail to exploit the common structure underlying the tasks this is because they tend to treat observation from different entities in the environment as an integral part of the whole.
Accordingly, they give tacit support to the assumption that neural networks are able to automatically decouple the observation to ﬁnd the best mapping between the whole observation and policy.
Adopting this approach means that they treat all information from other agents or different parts of the environment in the same way.
The most commonly used method involves concatenating the observations from each entity in to a vector that is used as input (Rashid et al.
(2018), Du et al.
(2019), Zhou et al.
In addition, current methods ignore the rich physical meanings behind each action.
Multi-agent tasks feature a close relationship between the observation and output.
If the model does not decouple the observation from the different agents, individual functions maybe
[Figure]
Individual
[Figure]
Credit Assignment G 𝑄 𝑘 (𝑜 1 ,𝑎 𝑡1 )
Individual F
Agent
Agent T 𝑜 𝑡𝑇 … 𝑄 𝑘 (𝑜 𝑇 ,𝑎 𝑡𝑇 ) 𝑄 𝑡𝑜𝑡 (𝑜, 𝑎)
[Figure]
Loss Function
GRU/LSTM 𝑎𝑐𝑡𝑖𝑜𝑛𝑠
F ours
[Figure]
misguided and impede the centralized value function.
Worse yet, conventional models require the input and the output dimensions to be ﬁxed (Shao et al.
(2018), Wang et al.
(2020)), which makes zero-shot transfer impossible.
Thus, the application of current methods is limited in real-world applications.
Our solution to these problems is to develop a multi-agent reinforcement learning (MARL) framework with no limitation on input or output dimension.
Moreover, this model should be general enough to be applicable to any existing MARL methods.
More importantly, the model should be explainable and capable of providing further improvement for both the ﬁnal performance on singletask scenarios and transfer capability on multi-task scenarios.
Inspired by the self-attention mechanism (Vaswani et al.
, we propose a transformer-based
MARL framework, named Universal Policy Decoupling Transformer ( UPDeT ).
There are four key advantages of this approach: 1) Once trained, it can be universally deployed; 2) it provide more robust representation with a policy decoupling strategy; 3) it is more explainable; 4) it is general enough to be applied on any MARL model.
We further design a transformer-based function to handle various observation sizes by treating individual observations as ”observation-entities”.
We match the related observation-entity with action-groups by separating the action space into several action-groups with reference to the corresponding observation-entity, allowing us to get matched observation-entity — action-group pairs set.
We further use a self-attention mechanism to learn the relationship between the matched observation-entity and other observation-entities.
Through the use of self-attention map and the embedding of each observation-entity, UPDeT can optimize the policy at an action-group level.
We refer to this strategy as Policy Decoupling .
By combining the transformer and policy decoupling strategies, UPDeT signiﬁcantly outperforms conventional RNNbased models.
In UPDeT, there is no need to introduce any new parameters for new tasks.
We also prove that it is only with decoupled policy and matched observation-entity — action-group pairs that UPDeT can learn a strong representation with high transfer capability.
Finally, our proposed UPDeT can be plugged into any existing method with almost no changes to the framework architecture required, while still bringing signiﬁcant improvements to the ﬁnal performance, especially in hard and complex multi-agent tasks.
The main contributions of this work are as follows:
First, our UPDeT-based MARL framework outperforms RNN-based frameworks by a large margin in terms of ﬁnal performance on state-ofthe-art centralized functions.
Second, our model has strong transfer capability and can handle a number of different tasks at a time.
Third, our model accelerates the transfer learning speed (total steps cost) to make it roughly 10 times faster compared to RNN-based models in most scenarios.
[Figure]
MAEXP 𝑜 𝑡𝑛 𝑒 𝑡𝑛 𝑞 𝑡𝑚
[Figure]
R ELATED W ORK
Attention mechanisms have become an integral part of models that capture global dependencies.
In particular, self-attention (Parikh et al.
(2016)) calculates the response at a speciﬁc position in a sequence by attending to all positions within this sequence.
Vaswani et al.
(2017) demonstrated that machine translation models can achieve state-of-the-art results solely by using a self-attention model.
Parmar et al.
(2018) proposed an Image Transformer model that applies self-attention to image generation.
Wang et al.
(2018) formalized self-attention as a non-local operation in order to model the spatial-temporal dependencies in video sequences.
In spite of this, self-attention mechanisms have not yet been fully explored in multi-agent reinforcement learning.
Another line of research is multi-agent reinforcement learning (MARL).
Existing work in MARL focuses primarily on building a centralized function to guide the training of individual value function (Lowe et al.
(2017), Sunehag et al.
(2017), Rashid et al.
(2018), Mahajan et al.
, Hostallero et al.
(2019), Yang et al.
(2020), Zhou et al.
Few works have opted to form a better individual functions with strong representation and transfer capability.
In standard reinforcement learning, this generalization has been fully studied (Taylor & Stone (2009), Ammar et al.
(2012), Parisotto et al.
, Gupta et al.
(2017), Da Silva & Costa (2019)).
While multi-agent transfer learning has been proven to be more difﬁcult than the single-agent scenario (Boutsioukis et al.
(2011), Shao et al.
(2018), Vinyals et al.
However, the transfer capability of a multi-agent system is of greater signiﬁcance due to the various number of agents, observations sizes and policy distributions.
To the best of our knowledge, we are the ﬁrst to develop a multi-agent framework capable of handling multiple task at a time.
Moreover, we provide a policy decoupling strategy to further improve the model performance and facilitate the multi-agent transfer learning, which is a signiﬁcant step towards real world multi-agent applications.
M ETHOD
We begin by introducing the notations and basic task settings necessary for our approach.
We then describe a transformer-based individual function and policy decoupling strategy under MARL.
Finally, we introduce different temporal units and assimilate our Universal Policy Decoupling Transformer (UPDeT) into Dec-POMDP.
3. N OTATIONS AND T ASK S ETTINGS
Multi-agent Reinforcement Learning A cooperative multi-agent task is a decentralized partially observable Markov decision process (Oliehoek et al.
with a tuple G = (cid:104) S, A, U, P, r, Z, O, n, γ (cid:105) .
Let S denote the global state of the environment, while A represents the set of n agents and U is the action space.
At each time step t , agent a ∈
A ≡ { 1 , ..., n } selects an action u ∈ U , forming
a joint action u ∈ U ≡ U n , which in turn causes a transition in the environment represented by the state transition function P ( s (cid:48) | s, u ) :
S × U × S → [0 , 1] .
All agents share the same reward function r ( s, u ) : S × U
→ R , while γ ∈ [0 , 1) is a discount factor.
We consider a partially observable scenario in which each agent makes individual observations z ∈ Z according to the observation function O ( s, a ) : S ×
A → Z .
Each agent has an actionobservation history that conditions a stochastic policy π t , creating the following joint action value:
Q
[Equation]
s t ,
[Equation]
t ) =
[Equation]
t +1: ∞ , u t +1: ∞
[ R t | s t , u t ] , where R t = (cid:80) ∞
i =
γ i r t
+ i is the discounted return.
Centralized training with decentralized execution Centralized training with decentralized execution (CTDE) is a commonly used architecture in the MARL context.
Each agent is conditioned only on its own action-observation history to make a decision using the learned policy.
The centralized value function provides a centralized gradient to update the individual function based on its output.
Therefore, a stronger individual value function can beneﬁt the centralized training.
3. T RANSFORMER - BASED I NDIVIDUAL V ALUE F UNCTION
In this section, we present a mathematical formulation of our transformer-based model UPDeT. We describe the calculation of the global Q-function with self-attention mechanism.
First, the observation O is embedded into a semantic embedding to handle the various observation space.
For example, if an agent a i observes k other entities { o i, 1 , ..., o i,k } at time step t , all observation entities are embedded via an embedding layer E as follows:
[Equation]
Here, i is the index of the agent, i ∈ { 1 , ..., n } .
Next, the value functions { Q 1 , ..., Q n } for the n agents for each step are estimated as follows:
[Equation]
We introduce h t − 1 i , the temporal hidden state at the last time step t − 1 , since POMDP policy is highly dependent on the historical information.
e ti denotes the observation embedding, while u ti is the candidate action, u ti ∈ U .
θ i is the parameter that deﬁnes Q i .
Finally, the global Q-function Q π is calculated by all individual value functions, as follows:
[Equation]
F i is the credit assignment function for deﬁned by φ i for each agent a i , as utilized in Rashid et al.
(2018) and Sunehag et al.
For example, in VDN, F is a sum function that can be expressed as F ( q t 1 , .., q tn ) =
(cid:80) n i =1 q t i .
Implement Q-function with Self-attention Vaswani et al.
(2017) adopts three matrices, K , Q , V representing a set of keys, queries and values respectively.
The attention is computed as follows:
[Equation]
where d k is a scaling factor equal to the dimension of the key.
In our method, we adopt the selfattention to learn the features and relationships from the observation entity embedding and the global temporal information.
To learn the independent policy in decentralized multi-agent learning, we deﬁne K i , Q i and V i as the key, query and value metrics for each agent a i .
We further consider the query, key and value for the same matrices R l i
= K i
= Q i
= V i , where l ∈ { 1 , ..., L } is the number of layers of the transformer.
Thus, we formulate our transformer as follows:
[Equation]
Transformer 𝑜 𝐴, 𝑜 𝐵,1 … …
𝑜 𝐴,𝑗 … … 𝑜 𝐵,𝑘 𝑜 𝑍, 𝑜 𝑍,𝑙 𝑞 1 𝑞𝑞𝑞𝑞 𝑥 𝑞 𝑤 𝑞 𝑧 𝑞 𝑦 𝑞 𝐴 𝑞 𝐴 𝑞 𝐵 𝑞 𝐶 𝑞 𝐵 𝑞 𝐶 𝑜 𝐴, 𝑜 𝐴,𝑗 … 𝑜 𝐵,1 … … 𝑜 𝐵,𝑘 𝑜 𝑍, 𝑜 𝑍,𝑙 …
𝑒 𝐴, 𝑒 𝐴,𝑗 … 𝑒 𝐵,1 … … 𝑒 𝐵,𝑘 𝑒 𝑍, 𝑒 𝑍,𝑙 𝑒 𝐴,11 𝑒 𝐴,1𝑚 𝑒 𝐴,𝑗1 𝑒 𝐴,𝑗𝑚 … … 𝑒 𝐵,11 𝑒 𝐵,1𝑛 𝑒 𝐵,𝑘𝑒 𝐵,𝑘𝑛
[Figure]
… … … 𝑒 𝑍,11 𝑒 𝑍,1𝑝 𝑒 𝑍,𝑙1 𝑒 𝑍,𝑙𝑝 … … … …
FC
FC FC
Preserve
[Figure]
𝐴,11 𝑞 𝐴,1𝑚 𝑞 𝐴,𝑗1 𝑞 𝐴,𝑗𝑚 … … 𝑞 𝐵, 𝑞 𝐵,𝑘
[Figure]
Aggregation
Abandon
Policy Distribution
[Figure]
where LF represents the linear functions used to compute K , Q , V .
Finally we project the entity features of the last transformer layer R Li to the output space of the value function Q i .
We implement the projection using a linear function P :
[Equation]
3. P OLICY D ECOUPLING
A single transformer-based individual function with self-attention mechanism is still unable to handle various required policy distribution.
A ﬂexible mapping function P in Eq.
6 is needed to deal with the various input and output dimensions and provide strong representation ability.
Using the correlation between input and output, we design a strategy called policy decoupling, which is the key part of UPDeT.
The main idea behind the policy decoupling strategy can be summarized into three points:
• Point 1 : No restriction on policy dimension.
The output dimension of a standard transformer block must be equal to or less than the input dimension.
This is unacceptable in some MARL tasks, as the action number can be larger than the entity number.
• Point 2 : Ability to handle multiple tasks at a time.
This requires a ﬁxed model architecture without new parameters being introduced for new tasks.
Unfortunately, if point 1 is satisﬁed, point 2 becomes very problematic to achieve.
The difﬁculty lies in how to reconcile points 1 and 2 .
• Point 3 : Make the model more explainable.
It would be preferable if we can could replace the conventional RNN-based model with a more explainable policy generation structure.
Following the above three points, we propose three policy decoupling methods, namely Vanilla
Transformer, Aggregation Transformer and Universal Policy Decoupling Transformer (UPDeT).
The pipelines are illustrated in Fig.
The details of the Vanilla Transformer and Aggregation
Transformer are presented in the experiment section and act as our baselines.
In this section, we mainly discuss the mechanism of our proposed UPDeT .
Tasking the entity features of the last transformer layer outlined in Eq. 5, the main challenge is to build a strong mapping between the features and the policy distribution.
UPDeT ﬁrst matches the input entity with the related output policy part.
This correspondence is easy to ﬁnd in the MARL task, as interactive action between two agents is quite common.
Once we match the corresponding entity features and actions, we substantially reduce the burden of model learning representation using the self-attention mechanism.
Moreover, considering that there might be more than one interactive actions of the matched entity feature, we separate the action space into several action groups, each of which consists several actions matched with one entity.
The pipeline of this process is illustrated in the left part of Fig.
In the mapping function, to satisfy point 1 and point 2 , we adopt two strategies.
First, if the action-group of one entity feature contains more than one action, a shared fully connected layer is added to map the output to the action number dimension.
Second, if one entity feature has no corresponding action, we abandon it, there is no danger of losing the information carried by this kind of entity feature, as the transformer has aggregated the information necessary to each output.
The pipeline of UPDeT can be found in the right part of Fig.
With UPDeT, there is no action restriction and no new parameter introduced in new scenarios.
A single model can be trained on multiple tasks and deployed universally.
In addition, matching the corresponding entity feature and action-group satisﬁes point 3 , as the policy is explainable using an attention heatmap, as we will discuss in Section 4.4.
3. T EMPORAL U NIT S TRUCTURE
Notably, however a transformer-based individual value function with policy decoupling strategy cannot handle a partial observation decision process without trajectory or history information.
In
Dec-POMDP (Oliehoek et al.
(2016)), each agent a chooses an action according to π a ( u a | τ a ) , where u and τ represents for action and action-observation history respectively.
In GRU and LSTM, we adopt a hidden state to hold the information of the action-observation history.
However, the combination of a transformer block and a hidden state has not yet been fully studied.
In this section, we provide two approaches to handling the hidden state in UPDeT:
Global temporal unit treats the hidden state as an additional input of the transformer block.
The process is formulated in a similar way to Eq.
5 with the relation: R 1 = { h t − 1 G , e t 1 } and { h tG , e tL } = R L .
Here, we ignore the subscript i and instead use G to represent ’global’.
The global temporal unit is simple but efﬁcient, and provides us with robust performance
in most scenarios.
Individual temporal unit treats the hidden state as the inner part of each entity.
In other words, each input maintains its own hidden state, while each output projects a new hidden state for the next time step.
The individual temporal unit uses a more precise approach to controlling history information as it splits the global hidden state into individual parts.
We use j to represent the number of entities.
The relation of input and output is formulated as R 1 = { h t − 1 1 ...
h t − 1 j , e t 1 } and { h t 1 ...h tj , e tL } = R L .
However, this method introduces the additional burden of learning the hidden state independently for each entity.
In experiment Section 4.1.2, we test both variants and discuss them further.
3. O PTIMIZATION
We use the standard squared TD error in DQNs (Mnih et al.
to optimize our entire framework as follows:
[Equation]
Here, b represents the batch size.
In partially observable settings, agents can beneﬁt from conditioning on action-observation history.
Hausknecht & Stone (2015) propose Deep Recurrent Q-networks (DRQN) for this sequential decision process.
For our part, we replace the widely used GRU (Chung et al.
(2014))/LSTM (Hochreiter & Schmidhuber (1997)) unit in DRQN with a transformer-based temporal unit and then train the whole model.
S TAR C RAFT II E XPERIMENT
In this section, we evaluate UPDeT and its variants with different policy decoupling methods in the context of challenging micromanagement games in StarCraft II.
We compare UPDeT with the
RNN-based model on a single scenario and test the transfer capability on multiple-scenario transfer tasks.
The experimental results show that UPDeT achieves signiﬁcant improvement compared to the RNN-based model.
T e s t W i n R a t e
Step (a) Policy variants
T e s t W i n R a t e
Step (b) Temporal variants
T e s t W i n R a t e
Step (c) MARL methods
T e s t W i n R a t e
Step (d) Easy scenarios
T e s t W i n R a t e
Step (e) Hard scenarios
T e s t W i n R a t e
Step (
[Figure]
Mismatch experiment
4. S INGLE S CENARIO
In the single scenario experiments, we evaluate the model performance on different scenarios from
SMAC (Samvelyan et al.
Speciﬁcally, the scenarios considered are as follows: 3 Marines vs 3 Marines (3m, Easy), 8 Marines vs 8 Marines (8m, Easy), 4 Marines vs 5 Marines (4m vs 5m,
Hard+) and 5 Marines vs 6 Marines (5m vs 6m, Hard).
In all these games, only the units from player’s side are treated as agents.
Dead enemy units will be masked out from the action space to ensure that the executed action is valid.
More detailed settings can be acquired from the SMAC environment (Samvelyan et al.
4.1. M ETHODS AND T RAINING D ETAILS
The MARL methods for evaluation include VDN (Sunehag et al.
, QMIX (Rashid et al.
and QTRAN (Hostallero et al.
All three SOTA methods’ original implementation can be found at https://github.com/oxwhirl/pymarl .
These methods were selected due to their robust performance across different multi-agent tasks.
Other methods, including COMA (Foerster et al.
and IQL (Tan (1993)) do not perform stable across in all tasks, as have been proved in several recent works (Rashid et al.
(2018), Mahajan et al.
Zhou et al.
Therefore, we combined UPDeT with VDN, QMIX and QTRAN to prove that our model can improve the model performance signiﬁcantly compared to the GRU-based model.
4.1. R ESULT
The model performance result with different policy decoupling methods can be found in Fig. 4a.
Vanilla Transformer is our baseline for all transformer-based models.
This transformer only satisﬁes point 2 .
Each output embedding can either be projected to an action or abandoned.
The vanilla transformer fails to beat the enemies in the experiment.
Aggregation Transformer is a variant of vanilla transformer, the embedding of which are aggregated into a global embedding and then projected to a policy distribution.
This transformer only satisﬁes the point 1 .
The performance of the aggregation transformer is worse than that of the GRU-based model.
The result proves that it is only with a policy decoupling strategy that the transformer-based model can outperform the conventional
RNN-based model.
Next, we adopt UPDeT to ﬁnd the best temporal unit architecture in Fig.
4b.
The result shows that without a hidden state, the performance is signiﬁcantly decreased.
The temporal unit with global hidden state is more efﬁcient in terms of convergence speed than the individual hidden state.
However, the ﬁnal performances are almost the same.
To test the generalization of our
[Figure]
→
[Figure]
performance w/o finetune
T e s t
W i n R a t e
Step
[Figure]
Transfer from
[Figure]
to 3 marines 3m → 5m 5m → 7m
T e s t
W i n R a t e
Step
[Figure]
from
[Figure]
marines model, we combine the UPDeT with VDN / QMIX / QTRAN respectively and compare the ﬁnal performance with RNN-based methods in Fig.
4c.
We evaluate the model performance on 5m vs 6m (Hard) scenarios.
Combined with UPDeT, all three MARL methods obtain signiﬁcant improvement by large margins compared to the GRU-based model.
The result proves that our model can be injected into any existing stat- of-the-art MARL method to yield better performance.
Further more, we combine UPDeT with VDN and evaluate the model performance on different scenarios from Easy to Hard+ in Fig.
4d and Fig.
4e.
The results show that the UPDeT performs stably on easy scenarios and signiﬁcantly outperforms the GRU-based model on hard scenarios, in the 4m vs 5m(Hard+) scenario, the performance improvement achieved by UPDeT relative to the GRU-based model is of the magnitude of around 80%.
Finally, we conduct an ablation study on UPDeT with paired and unpaired observation-entity—action-group, the result of which are presented in Fig.
4f.
We disrupt the original correspondence between ’attack’ action and enemy unit.
The ﬁnal performance is heavily decreased compared to the original model, and is even worse than the GRU-based model.
We accordingly conclude that only with policy decoupling and a paired observation-entity—action-group strategy can UPDeT learn a strong policy.
4. M ULTIPLE S CENARIOS
In this section, we discuss the transfer capability of UPDeT compared to the RNN-based model.
We evaluate the model performance in a curriculum style.
First, the model is trained one the 3m (
Marines vs 3 Marines) scenario.
We then used the pretrained 3m model to continually train on the 5m (5 Marines vs 5 Marines) and 7m (7 Marines vs 7 Marines) scenarios.
We also conduct a experiment in reverse from 7m to 3m. During transfer learning, the model architecture of UPDeT remains ﬁxed.
Considering that the RNN-based model cannot handle various input and output dimensions, we modify the architecture of the source RNN model when training on the target scenario.
We preserve the parameters of the GRU cell and initialize the fully connected layer with proper input and output dimensions to ﬁt the new scenario.
The ﬁnal results can be seen in Fig.
5a and Fig.
5b.
Our proposed
UPDeT achieves signiﬁcantly better results than the GRU-based model.
Statistically, UPDeT’s total timestep cost to converge is at least 10 times less than the GRU-based model and 100 times less than training from scratch.
Moreover, the model demonstrates a strong generalization ability without ﬁnetuning, indicating that UPDeT learns a robust policy with meta-level skill.
4. E XTENSIVE EXPERIMENT ON LARGE - SCALE MAS
To evaluate the model performance in large-scale scenarios, we test our proposed UPDeT on the 10m vs 11m and 20m vs 21m scenarios from SMAC and a 64 vs 64 battle game in the MAgent
Environment (Zheng et al.
The ﬁnal results can be found in Appendix E.
[Figure]
4. A TTENTION BASED S TRATEGY : A N ANALYSIS
The signiﬁcant performance improvement achieved by UPDeT on the SMAC multi-agent challenge can be credited to the self-attention mechanism brought by both transformer blocks and the policy decoupling strategy in UPDeT. In this section, we mainly discuss how the attention mechanism assists in learning a much more robust and explainable strategy.
Here, we use the 3 Marines vs
Marines game (therefore, the size of the raw attention matrix is 6x6) as an example to demonstrate how the attention mechanism works.
As mentioned in the caption of Fig.
6, we simplify the raw complete attention matrix to a grouped attention matrix.
Fig.
6b presents the three different stages in one episode including Game Start , Attack and Survive , with their corresponding attention matrix and strategies.
In the Game Start stage, the highest attention is in line 1 col 3 of the matrix, indicating that the agent pays more attention to its allies than its enemies.
This phenomenon can be interpreted as follows: in the startup stage of one game, all the allies are spawned at the left side of the map and are encouraged to ﬁnd and attack the enemies on the right side In the Attack stage, the highest attention is in line 2 col 2 of the matrix, which indicates that the enemy is now in the agent’s attack range; therefore, the agent will attack the enemy to get more rewards.
Surprisingly, the agent chooses to attack the enemy with the lowest health value.
This indicates that a long term plan can be learned based on the attention mechanism, since killing the weakest enemy ﬁrst can decrease the punishment from the future enemy attacks.
In the Survive stage, the agent’s health value is low, meaning that it needs to avoid being attacked.
The highest attention is located in line 1 col 1, which clearly shows that the most important thing under the current circumstances is to stay alive.
For as long as the agent is alive, there is still a chance for it to return to the front line and get more reward while enemies are attacking the allies instead of the agent itself.
In conclusion, the self-attention mechanism and policy decoupling strategy of UPDeT provides a strong and clear relation between attention weights and ﬁnal strategies.
This relation can help us better understand the policy generation based on the distribution of attention among different entities.
An interesting idea presents itself here: namely, if we can ﬁnd a strong mapping between attention matrix and ﬁnal policy, the character of the agent could be modiﬁed in an unsupervised manner.
C ONCLUSION
In this paper, we propose UPDeT, a universal policy decoupling transformer model that extends
MARL to a much broader scenario.
UPDeT is general enough to be plugged into any existing
MARL method.
Moreover, our experimental results show that, when combined with UPDeT, existing state-of-the-art MARL methods can achieve further signiﬁcant improvements with the same training pipeline.
On transfer learning tasks, our model is 100 times faster than training from scratch and 10 times faster than training using the RNN-based model.
In the future, we aim to develop a centralized function based on UPDeT and apply the self-attention mechanism to the entire pipeline of MARL framework to yield further improvement.
A D ETAILS OF SMAC ENVIRONMENT
The action space contains four movement directions, k attack actions (where k is the ﬁxed maximum number of the enemy units in a map), stop and none-operation.
At each time step, the agents receive a joint team reward, which is deﬁned by the total damage incurred by the agents and the total damage from the enemy side.
Each agent is described by several attributes, including health point
HP , weapon cool down (CD), unit type, last action and the relative distance of the observed units.
The enemy units are described in the same way except that CD is excluded.
The partial observation of an agent comprises the attributes of the units, including both the agents and the enemy units, that exist within its view range, which is a circle with a speciﬁc radius.
B D ETAILS OF M ODEL
The transformer block in all different experiments consists of 3 heads and 2 layer transformer blocks.
The other important training hyper parameters are as follows:
List of Hyper Parameters
Name
Value
batch size
test interval
gamma
buffer size
token dimension (UPDeT)
channel dimension (UPDeT)
epsilon start 1.
epsilon end 0.
rnn hidden dimension
target net update interval
mixing embeddding dimension (QMIX)
hypernet layers (QMIX) hypernet embedding (QMIX)
mixing embeddding dimension (QTRAN)
opt loss (QTRAN) nopt min loss (QTRAN) 0.
C SOTA MARL VALUE - BASED F RAMEWORK
The three SOTA method can be brieﬂy summarized as follows:
• VDN (Sunehag et al.
): this method learns an individual Q-value function and represents Q tot as a sum of individual Q-value functions that condition only on individual observations and actions.
• QMIX (Rashid et al.
): this method learns a decentralized Q-function for each agent, with the assumption that the centralized Q-value increases monotonically with the individual Q-values.
• QTRAN (Hostallero et al.
(2019)): this method formulates multi-agent learning as an optimization problem with linear constraints and relaxes it with L2 penalties for tractability.
D UPD E T ON SMAC: A REAL CASE
We take the 3 Marines vs 3 Marines challenge from SMAC with UPDeT as an example; more details can be found in Fig.
The observation are separated into 3 groups: main agent, two other ally agents and three enemies.
The policy output includes basic action corresponding to the main agent’s observation and attack actions, one for each enemy observation.
The hidden state is added after the embedding layer.
The output of other agents is abandoned as there is no corresponding action.
Once an agent or enemy has died, we mask corresponding unavailable action in the action select stage to ensure only the available actions are selected.
[Figure]
E R ESULTS OF E XTENSIVE E XPERIMENT ON L ARGE S CALE
We further test the robustness of UPDeT in a large-scale multi-agent system.
To do so, we enlarge the game size in SMAC (Samvelyan et al.
to incorporate more agents and enemies on the battle ﬁeld.
We use a 10 Marines vs 11 Marines game and a 20 Marines vs 21 Marines game to compare the performance between the UPDeT and GRU-based approaches.
In the 20 Marines vs
Marines game, to accelerate the training and satisfy the hardware limitations, we decrease the batch size of both the GRU baseline and UPDeT from 32 to 24 in the training stage.
The ﬁnal results can be found in Fig.
8a.
The improvement is still signiﬁcant in terms of both sample efﬁciency and ﬁnal performance.
Moreover, it is also worth mentioning that the model size of UPDeT stays ﬁxed, while the GRU-based model becomes larger in large-scale scenarios.
In the 20 Marines vs 21 Marines game, the model size of GRU is almost double that of UPDeT. This indicates that UPDeT is able to ensure the lightness of the model while still maintaining good performance.
We also test the model performance in the MAgent Environment (Zheng et al.
The settings of MAgent are quite different from those of SMAC.
First, the observation size and number of available actions are not related to the number of agents.
Second, the 64 vs 64 battle game we tested is a two-player zero-sum game which is another hot research area that combines both MARL and GT (Game Theory), the most successful attempt in this area involves adopting a mean-ﬁeld approximation of GT in MARL to accelerate the self-play training (Yang et al.
Third, as for the model architecture, there is no need to use a recurrent network like GRU in MAgent and the large observation size requires the use of a CNN from embedding.
However, ny treating UPDeT as a pure encoder without recurrent architecture, we can still conduct experiments on MAgent; the ﬁnal results of these can be found in Fig.
8b.
As the result show, UPDeT performs better than the DQN baseline, although this improvement is not as signiﬁcant as it in SMAC.