Abstract
This work has been submitted to the IEEE for possible publication. Copyrightmay be transferred without notice, after which this version may no longer beaccessible. Multi-agent task allocation (MATA) plays a vital role in cooperativemulti-agent systems, with significant implications for applications such aslogistics, search and rescue, and robotic coordination. Although traditionaldeep reinforcement learning (DRL) methods have been shown to be promising,their effectiveness is hindered by a reliance on manually designed rewardfunctions and inefficiencies in dynamic environments. In this paper, an inversereinforcement learning (IRL)-based framework is proposed, in which multi-headself-attention (MHSA) and graph attention mechanisms are incorporated toenhance reward function learning and task execution efficiency. Expertdemonstrations are utilized to infer optimal reward densities, allowingdependence on handcrafted designs to be reduced and adaptability to beimproved. Extensive experiments validate the superiority of the proposed methodover widely used multi-agent reinforcement learning (MARL) algorithms in termsof both cumulative rewards and task execution efficiency.