FlickerFusion: Intra-trajectory Domain Generalizing Multi-Agent RL

Woosung Koh, Wonbeen Oh, Siyeol Kim, Suhin Shin, Hyeongjin Kim, Jaein Jang, Junghyun Lee, Se-Young Yun

Keywords: Domain Generalization, Multi-agent Reinforcement Learning

Abstract: Multi-agent reinforcement learning has demonstrated significant potential in addressing complex cooperative tasks across various real-world applications. However, existing MARL approaches often rely on the restrictive assumption that the number of entities (e.g., agents, obstacles) remains constant between training and inference. This overlooks scenarios where entities are dynamically removed or \textit{added} \textit{during} the inference trajectory---a common occurrence in real-world environments like search and rescue missions and dynamic combat situations. In this paper, we tackle the challenge of \textbf{intra-trajectory dynamic entity composition} under zero-shot \textbf{out-of-domain (OOD) generalization}, where such dynamic changes cannot be anticipated beforehand. Our empirical studies reveal that existing MARL methods suffer \textit{significant} performance degradation and increased uncertainty in these scenarios. In response, we propose \textsc{FlickerFusion}, a novel OOD generalization method that acts as a \textit{universally} applicable augmentation technique for MARL backbone methods. Our results show that \textsc{FlickerFusion} not only achieves superior inference rewards but also \textit{uniquely} reduces uncertainty vis-à-vis the backbone, compared to existing methods. For standardized evaluation, we introduce \textsc{MPEv2}, an enhanced version of Multi Particle Environments (MPE), consisting of 12 benchmarks. Benchmarks, implementations, and trained models are organized and open-sourced at \texttt{\href{flickerfusion305.github.io}{flickerfusion305.github.io}}, accompanied by ample demo video renderings.

Submission Number: 12