arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.01112cs.LGcs.MA

合作多智能体强化学习的高效情景记忆利用

Efficient Episodic Memory Utilization of Cooperative Multi-Agent Reinforcement Learning

Korea Advanced Institute of Science and Technology (KAIST)(韩国科学技术院(KAIST))

Hyungho Na, Yunkyeong Seo, Il-chul Moon

AI总结:

针对合作多智能体强化学习训练慢、易陷入局部最优的问题,提出含可训练编解码结构和情景激励奖励的EMU方法,在两个基准上实现了优于SOTA的性能。

摘要:

在合作多智能体强化学习(MARL)中,智能体旨在实现共同目标,例如击败敌人或进球。现有MARL算法虽然有效,但仍需要大量学习时间,且常因复杂任务陷入局部最优,进而无法发现达成目标的策略。为解决这一问题,我们提出了适用于MARL的高效情景记忆利用(Efficient episodic Memory Utilization,EMU)方法,其有两个核心目标:(a)通过利用情景缓冲区中语义连贯的记忆加速强化学习;(b)选择性地促进理想转换以防止局部收敛。为实现目标(a),EMU在MARL基础上引入了可训练的编码器/解码器结构,生成连贯的记忆嵌入以促进探索性记忆回忆。为实现目标(b),EMU提出了一种基于状态合意性的新型奖励结构,称为情景激励(episodic incentive)。该奖励改进了Q学习中的TD目标,并作为理想转换的额外激励。我们为所提出的激励机制提供了理论支撑,并证明了EMU相较于传统情景控制的有效性。我们在StarCraft II和Google Research Football环境中对该方法进行了评估,实验结果表明,其性能相较于现有最优方法有进一步提升。

英文摘要:

In cooperative multi-agent reinforcement learning (MARL), agents aim to achieve a common goal, such as defeating enemies or scoring a goal. Existing MARL algorithms are effective but still require significant learning time and often get trapped in local optima by complex tasks, subsequently failing to discover a goal-reaching policy. To address this, we introduce Efficient episodic Memory Utilization (EMU) for MARL, with two primary objectives: (a) accelerating reinforcement learning by leveraging semantically coherent memory from an episodic buffer and (b) selectively promoting desirable transitions to prevent local convergence. To achieve (a), EMU incorporates a trainable encoder/decoder structure alongside MARL, creating coherent memory embeddings that facilitate exploratory memory recall. To achieve (b), EMU introduces a novel reward structure called episodic incentive based on the desirability of states. This reward improves the TD target in Q-learning and acts as an additional incentive for desirable transitions. We provide theoretical support for the proposed incentive and demonstrate the effectiveness of EMU compared to conventional episodic control. The proposed method is evaluated in StarCraft II and Google Research Football, and empirical results indicate further performance improvement over state-of-the-art methods.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑