CoFlow: Coordinated Few-Step Flow for Offline Multi-Agent Decision Making
How can generative offline multi-agent reinforcement learning achieve both fast joint trajectory generation and effective cooperation? Multi-agent diffusion models generate joint trajectories through iterative sampling but incur high inference latency. Distillation reduces sampling steps but may lose the teacher's cross-agent dependencies, reducing the success rate of cooperative tasks. A joint few-step generator can be trained directly to avoid losing these dependencies during distillation. However, insufficient modeling of cross-agent dependencies may still impair cooperation. In addition, its consistency training process introduces heavy computational and memory costs. Therefore, we propose Coordinated Few-Step Flow (CoFlow), which directly learns a joint averaged-velocity field for distillation-free one-step and few-step multi-agent trajectory generation. To support cooperation during generation, Coordinated Velocity Attention (CVA) incorporates teammate trajectory information into each agent's prediction. To reduce the consistency-training cost of this joint model, we approximate the correction term using finite differences. Across 48 configurations on MPE and SMAC, CoFlow supports centralized and decentralized execution. Under centralized execution, CoFlow outperforms our reproduced baseline by 11.2%, averaging relative gains equally across both benchmark suites, with a 12.93-fold model-sampling speedup. CVA improves normalized scores by 165.1% on average over the same model with cross-agent attention disabled. Compared with the exact-derivative implementation, finite-difference training achieves a 1.78-fold speedup and reduces peak GPU memory by 41.1%. Project page: https://github.com/Guowei-Zou/coflow.