Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Energy-Guided Flow Matching Enables Few-Step Conformer Generation and Ground-State Identification

About

Generating low-energy conformer ensembles and identifying ground-state conformations from molecular graphs remain computationally demanding with physics-based pipelines. Current learning-based approaches often suffer from a fragmented paradigm: generative models capture diversity but lack reliable energy calibration, whereas deterministic predictors target a single structure and fail to represent ensemble variability. Here we present EnFlow, a unified framework that couples flow matching (FM) with an explicitly learned energy model through an energy-guided sampling scheme defined along a non-Gaussian FM path. By incorporating energy-gradient guidance during sampling, our method steers trajectories toward lower-energy regions, substantially improving conformational fidelity, particularly in the few-step regime. The learned energy function further enables efficient energy-based ranking of generated ensembles for accurate ground-state identification. Extensive experiments on GEOM-QM9 and GEOM-Drugs demonstrate that EnFlow simultaneously improves generation metrics with 1--2 ODE-steps and reduces ground-state prediction errors compared with state-of-the-art methods.

Guikun Xu, Xiaohan Yi, Peilin Zhao, Yatao Bian• 2025

Related benchmarks

TaskDatasetResultRank
Conformer GenerationGEOM-QM9 δ = 0.5Å (test)
Recall COV Mean96.7
30
Molecule Conformer GenerationGEOM-Drugs δ = 0.75Å (test)
COV-R (mean)78.8
30
Ground-State Conformation PredictionGEOM-DRUGS (test)
MAE (Distance)0.644
13
Showing 3 of 3 rows

Other info

Follow for update