arXiv · 2406.05426
Baking Symmetry into GFlowNets
Abstract
GFlowNets have exhibited promising performance in generating diverse candidates with high rewards. These networks generate objects incrementally and aim to learn a policy that assigns probability of sampling objects in proportion to rewards. However, the current training pipelines of GFlowNets do not consider the presence of isomorphic actions, which are actions resulting in symmetric or isomorphic states. This lack of symmetry increases the amount of samples required for training GFlowNets and can result in inefficient and potentially incorrect flow functions. As a consequence, the reward and diversity of the generated objects decrease. In this study, our objective is to integrate symmetries into GFlowNets by identifying equivalent actions during the generation process. Experimental results using synthetic data demonstrate the promising performance of our proposed approaches.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
George Ma, Emmanuel Bengio, Yoshua Bengio, Dinghuai Zhang. 2024-06-08. Baking Symmetry into GFlowNets. https://arxiv.org/abs/2406.05426
Cite the original work for its findings. Save a collection to share your selection of sources.