Skip to yearly menu bar Skip to main content


Poster
in
Workshop: The Synergy of Scientific and Machine Learning Modelling (SynS & ML) Workshop

Hybrid Diffusions for Stable Molecular Structure Generation via Explicit Energy-based Model

Youngwoo Cho · Seunghoon Yi · Sookyung Kim · Hongkee Yoon · Joonseok Lee

Keywords: [ Molecule Generation ] [ Diffusion ] [ energy-based model ]


Abstract:

Generation of 3D molecules utilizing diffusion models often encounters difficulties in producing stable structures, primarily due to the emergence of unstable intermediate structures during diffusion steps. To account for this issue, we introduce a diffusion-based molecule generation model that incorporates an energy-based model (EBM), pretrained on density functional theory (DFT) data. Specifically, we propose three strategic use of EBM: 1) guided exploration using the EBM, 2) stability evaluation to accept the structure or to reject and restart the generation at the end of diffusion steps, and 3) performing post-relaxation refinement. With these three strategies, we demonstrate that the energy estimator significantly enhances the generated molecule’s stability.

Chat is not available.