Skip to yearly menu bar Skip to main content


Poster
in
Workshop: Geometry-grounded Representation Learning and Generative Modeling

Lift Your Molecules: Molecular Graph Generation in Latent Euclidean Space

Mohamed Amine Ketata · Nicholas Gao · Johanna Sommer · Tom Wollschläger · Stephan Günnemann

Keywords: [ Molecule Generation ] [ Computational Biology ] [ Graph generation ] [ Latent Diffusion Models ] [ Drug Design ]


Abstract: We introduce a new framework for molecular graph generation using 3D molecular generative models. Our Synthetic Coordinate Embedding (SyCo) framework maps molecular graphs to Euclidean point clouds via synthetic conformer coordinates and learns the inverse map using an E($n$)-Equivariant Graph Neural Network (EGNN). The induced point cloud-structured latent space is well-suited to apply existing 3D molecular generative models. This approach simplifies the graph generation problem - without relying on molecular fragments nor autoregressive decoding - into a point cloud generation problem followed by node and edge classification tasks. As a concrete implementation of our framework, we develop EDM-SyCo based on the E(3) Equivariant Diffusion Model (EDM). It achieves state-of-the-art performance in distribution learning of molecular graphs, outperforming the best non-autoregressive methods by more than 30\% on ZINC250K and 16\% on GuacaMol while improving conditional generation by up to 3.9 times.

Chat is not available.