GGFlow: A Graph Flow Matching Method with Efficient Optimal Transport

TMLR Paper4673 Authors

Abstract: Generating graph-structured data is crucial in various domains but remains challenging due to the complex interdependencies between nodes and edges. While diffusion models have demonstrated their superior generative capabilities, they often suffer from unstable training and inefficient sampling. To enhance generation performance and training stability, we propose GGFlow, a discrete flow matching generative model incorporating an efficient optimal transport for graph structures and it incorporates an edge-augmented graph transformer to enable direct communications among edges. Additionally, GGFlow introduces a novel goal-guided generation framework to control the generative trajectory of our model towards desired properties. GGFlow demonstrates superior performance on both unconditional and conditional generation tasks, outperforming existing baselines and underscoring its effectiveness and potential for wider application.

Submission Length: Regular submission (no more than 12 pages of main content)

Assigned Action Editor: ~Jan-Willem_van_de_Meent1

Submission Number: 4673