Hybrid Training for Enhanced Multi-task Generalization in Multi-agent Reinforcement Learning

التفاصيل البيبلوغرافية
العنوان: Hybrid Training for Enhanced Multi-task Generalization in Multi-agent Reinforcement Learning
المؤلفون: Zhang, Mingliang, Su, Sichang, He, Chengyang, Sartoretti, Guillaume
سنة النشر: 2024
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Machine Learning, Computer Science - Multiagent Systems
الوصف: In multi-agent reinforcement learning (MARL), achieving multi-task generalization to diverse agents and objectives presents significant challenges. Existing online MARL algorithms primarily focus on single-task performance, but their lack of multi-task generalization capabilities typically results in substantial computational waste and limited real-life applicability. Meanwhile, existing offline multi-task MARL approaches are heavily dependent on data quality, often resulting in poor performance on unseen tasks. In this paper, we introduce HyGen, a novel hybrid MARL framework, Hybrid Training for Enhanced Multi-Task Generalization, which integrates online and offline learning to ensure both multi-task generalization and training efficiency. Specifically, our framework extracts potential general skills from offline multi-task datasets. We then train policies to select the optimal skills under the centralized training and decentralized execution paradigm (CTDE). During this stage, we utilize a replay buffer that integrates both offline data and online interactions. We empirically demonstrate that our framework effectively extracts and refines general skills, yielding impressive generalization to unseen tasks. Comparative analyses on the StarCraft multi-agent challenge show that HyGen outperforms a wide range of existing solely online and offline methods.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2408.13567
رقم الأكسشن: edsarx.2408.13567
قاعدة البيانات: arXiv