A Theory of Multi-Agent Generative Flow Networks
This provides a foundational theory for multi-agent generative modeling, addressing a gap in the field, though it is incremental as it extends existing single-agent GFlowNets to multi-agent settings.
The paper tackles the lack of a theoretical framework for multi-agent generative flow networks (MA-GFlowNets) by proposing one that enables collaborative object generation through joint actions, with experimental results showing superiority over reinforcement learning and MCMC-based methods.
Generative flow networks utilize a flow-matching loss to learn a stochastic policy for generating objects from a sequence of actions, such that the probability of generating a pattern can be proportional to the corresponding given reward. However, a theoretical framework for multi-agent generative flow networks (MA-GFlowNets) has not yet been proposed. In this paper, we propose the theory framework of MA-GFlowNets, which can be applied to multiple agents to generate objects collaboratively through a series of joint actions. We further propose four algorithms: a centralized flow network for centralized training of MA-GFlowNets, an independent flow network for decentralized execution, a joint flow network for achieving centralized training with decentralized execution, and its updated conditional version. Joint Flow training is based on a local-global principle allowing to train a collection of (local) GFN as a unique (global) GFN. This principle provides a loss of reasonable complexity and allows to leverage usual results on GFN to provide theoretical guarantees that the independent policies generate samples with probability proportional to the reward function. Experimental results demonstrate the superiority of the proposed framework compared to reinforcement learning and MCMC-based methods.