Abstract:Effective exploration is crucial to discovering optimal strategies for multi-agent reinforcement learning (MARL) in complex coordination tasks. Existing methods mainly utilize intrinsic rewards to enable committed exploration or use role-based learning for decomposing joint action spaces instead of directly conducting a collective search in the entire action-observation space. However, they often face challenges obtaining specific joint action sequences to reach successful states in long-horizon tasks. To address this limitation, we propose Imagine, Initialize, and Explore (IIE), a novel method that offers a promising solution for efficient multi-agent exploration in complex scenarios. IIE employs a transformer model to imagine how the agents reach a critical state that can influence each other's transition functions. Then, we initialize the environment at this state using a simulator before the exploration phase. We formulate the imagination as a sequence modeling problem, where the states, observations, prompts, actions, and rewards are predicted autoregressively. The prompt consists of timestep-to-go, return-to-go, influence value, and one-shot demonstration, specifying the desired state and trajectory as well as guiding the action generation. By initializing agents at the critical states, IIE significantly increases the likelihood of discovering potentially important under-explored regions. Despite its simplicity, empirical results demonstrate that our method outperforms multi-agent exploration baselines on the StarCraft Multi-Agent Challenge (SMAC) and SMACv2 environments. Particularly, IIE shows improved performance in the sparse-reward SMAC tasks and produces more effective curricula over the initialized states than other generative methods, such as CVAE-GAN and diffusion models.

Subspace-Aware Exploration for Sparse-Reward Multi-Agent Tasks.

Optimal Exploration Algorithm of Multi-Agent Reinforcement Learning Methods (Student Abstract)

Underexplored Subspace Mining for Sparse-Reward Cooperative Multi-Agent Reinforcement Learning

Expert demonstrations guide reward decomposition for multi-agent cooperation

CMBE: Curiosity-driven Model-Based Exploration for Multi-Agent Reinforcement Learning in Sparse Reward Settings

Self-Motivated Multi-Agent Exploration

Coordinated Exploration via Intrinsic Rewards for Multi-Agent Reinforcement Learning

Multi-agent Exploration with Sub-state Entropy Estimation

MESA: Cooperative Meta-Exploration in Multi-Agent Learning through Exploiting State-Action Space Structure

Two Heads Are Better Than One: A Simple Exploration Framework for Efficient Multi-Agent Reinforcement Learning.

Keeping Your Distance: Solving Sparse Reward Tasks Using Self-Balancing Shaped Rewards

Multi-Agent Exploration of an Unknown Sparse Landmark Complex via Deep Reinforcement Learning

Dynamic Subgoal-based Exploration via Bayesian Optimization

Multi-Agent Exploration Via Self-Learning and Social Learning

Adaptive trajectory-constrained exploration strategy for deep reinforcement learning

Cooperative multi-agent target searching: a deep reinforcement learning approach based on parallel hindsight experience replay

Imagine, Initialize, and Explore: An Effective Exploration Method in Multi-Agent Reinforcement Learning

Discovering and Exploiting Sparse Rewards in a Learned Behavior Space

Exploration in Deep Reinforcement Learning: From Single-Agent to Multiagent Domain

Influence-Based Multi-Agent Exploration

MuDE: Multi-agent decomposed reward-based exploration