Search papers, labs, and topics across Lattice.
This paper introduces the Planner-Conditioned Diffusion Policy (PCDP), a novel approach for coordinated multi-agent exploration that leverages multimodal trajectory generation conditioned on various planner styles. By training a single model on diverse planner demonstrations, PCDP enables agents to generate non-redundant trajectories while maintaining efficient coverage over extended planning horizons. The results show that PCDP achieves a perfect success rate comparable to diffusion-based baselines while enhancing travel efficiency and agent balance, highlighting the significance of multimodal planning in improving coordination among agents.
Planner-Conditioned Diffusion Policy enables agents to generate diverse, non-redundant trajectories, achieving perfect success rates while enhancing travel efficiency and coordination.
Coordinated multi-agent exploration requires not only efficient individual coverage but also non-redundant coverage across agents over extended planning horizons. Conventional approaches rely on hand-crafted coordination rules, while end-to-end multi-agent learning methods are difficult to scale and train. Diffusion-based planners such as DARE offer a promising alternative by generating long-horizon trajectories instead of single-step actions, but existing methods are trained on a narrow planner distribution, limiting behavioral diversity and inference-time controllability. We propose a Planner-Conditioned Diffusion Policy (PCDP) for graph-based multi-agent exploration. PCDP is trained on demonstrations from multiple planner styles with planner identity as an explicit conditioning input, enabling a single shared model to learn a multimodal trajectory distribution and generate diverse, controllable trajectory candidates from the same observation. Rather than learning coordination end-to-end, we reuse this multimodal single-agent policy across all agents and introduce coordination through local reranking, in which nearby agents jointly select the trajectory combination with minimal predicted overlap. We evaluate PCDP against classical and diffusion-based baselines on 100 held-out maps in a four-agent simulation setting. PCDP matches the perfect success rate of the diffusion-based baselines while improving mean max-agent travel, total team travel, and agent imbalance. Crucially, reranking alone over a single-planner baseline yields only marginal gains, indicating that planner-conditioned multimodality is the main contributor to improved coordination. Qualitative simulation results and real-robot experiments with two agents further validate that diverse long-horizon trajectory generation produces emergent spatial separation between agents without any explicit repulsion mechanism.
