Accelerating 3D Molecule Generative Models with Trajectory Diagnosis

0citations
0
Citations
#1943
in NeurIPS 2025
of 5858 papers
8
Authors
3
Data Points

Abstract

Geometric molecule generative models have found expanding applications across various scientific domains, but their generation inefficiency has become a critical bottleneck. Through a systematic investigation of the generative trajectory, we discover a unique challenge for molecule geometric graph generation: generative models require determining the permutation order of atoms in the molecule before refining its atomic feature values. Based on this insight, we decompose the generation process into permutation phase and adjustment phase, and propose a geometric-informed prior and consistency parameter objective to accelerate each phase. Extensive experiments demonstrate that our approach achieves competitive performance with approximately 10 sampling steps, 7.5 × faster than previous state-of-the-art models and approximately 100 × faster than diffusion-based models, offering a significant step towards scalable molecular generation.

Citation History

Jan 26, 2026
0
Jan 27, 2026
0
Jan 27, 2026
0