中国科学院机构知识库网格
Chinese Academy of Sciences Institutional Repositories Grid
M3:Modularization for Multi-task and Multi-agent Offline Pre-training

文献类型:会议论文

作者Linghui Meng; Jingqing Ruan; Xuantang Xiong; Xiyun Li; Xi Zhang; Dengpeng Xing; Bo Xu
出版日期2023-05
会议日期2023-5
会议地点London, United Kingdom
英文摘要

Learning a multi-task policy is crucial in multi-agent reinforcement learning (MARL). Recent work has focused on learning in the context of online multi-task reinforcement learning, where a policy is jointly trained from scratch, aiming to generalize well to few-shot or even zero-shot tasks. However, existing online methods require tremendous interactions and are therefore unsuitable for environments where interactions are expensive. In this work, we novelly introduce the modularization for multi-task and multi-agent offline pre-training (M3) to learn high-level transferable policy representations. We claim that the discrete policy representation is critical for multi-task offline learning and accordingly leverage contexts as a task prompt to enhance the adaptability of pre-trained models to various tasks. To disentangle multiple agents of variation under heterogeneous and non-stationary properties even though they receive the same task, we employ an agent-invariant VQ-VAE to identify each of the multiple agents. We encapsulate the pre-trained model as part of an online MARL algorithm and fine-tune it to improve generalization. We also theoretically analyze the generalization error of our method. We test the proposed method on the challenging StarCraft Multi-Agent Challenge (SMAC) tasks, and empirical results show that it can achieve supreme performance in few-shot or even zero-shot settings across multiple tasks over state-of-the-art MARL methods.

源URL[http://ir.ia.ac.cn/handle/173211/59416]  
专题数字内容技术与服务研究中心_听觉模型与认知计算
作者单位中科院自动化所
推荐引用方式
GB/T 7714
Linghui Meng,Jingqing Ruan,Xuantang Xiong,et al. M3:Modularization for Multi-task and Multi-agent Offline Pre-training[C]. 见:. London, United Kingdom. 2023-5.

入库方式: OAI收割

来源:自动化研究所

浏览0
下载0
收藏0
其他版本

除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。