Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Equivariant Efficient Joint Discrete and Continuous MeanFlow for Molecular Graph Generation

About

Graph-structured data jointly contain discrete topology and continuous geometry, which poses fundamental challenges for generative modeling due to heterogeneous distributions, incompatible noise dynamics, and the need for equivariant inductive biases. Existing flow-matching approaches for graph generation typically decouple structure from geometry, lack synchronized cross-domain dynamics, and rely on iterative sampling, often resulting in physically inconsistent molecular conformations and slow sampling. To address these limitations, we propose Equivariant MeanFlow (EQUIMF), a unified SE(3)-equivariant generative framework that jointly models discrete and continuous components through synchronized MeanFlow dynamics. EQUIMF introduces a unified time bridge and average-velocity updates with mutual conditioning between structure and geometry, enabling efficient few-step generation while preserving physical consistency. Moreover, we develop a novel discrete MeanFlow formulation with a simple yet effective parameterization to support efficient generation over discrete graph structures. Extensive experiments demonstrate that EQUIMF consistently outperforms prior diffusion and flow-matching methods in generation quality, physical validity, and sampling efficiency.

Rongjian Xu, Teng Pang, Zhiqiang Dong, Guoqiang Wu• 2026

Related benchmarks

TaskDatasetResultRank
Molecular property predictionQM9
Cv1.011
80
Graph generationSBM
VUN0.912
51
Graph generationPlanar
V.U.N.99.6
48
Molecular Graph GenerationQM9
Validity95.8
37
Graph generationTree
A.Ratio1.6
36
Molecular Graph Generationdrug
Atom Stability84.5
6
Showing 6 of 6 rows

Other info

Follow for update