Timezone: »

 
DiffMol: 3D Structured Molecule Generation with Discrete Denoising Diffusion Probabilistic Models
Weitong Zhang · Xiaoyun Wang · Justin Smith · Joe Eaton · Brad Rees · Quanquan Gu

@
in Structured Probabilistic Inference and Generative Modeling »
Event URL: https://openreview.net/forum?id=x43ZyXJC9q »

3D structures of molecules are often required to investigate atomistic phenomena accurately in industries such as drug design. We propose DiffMol, a novel method that utilizes diffusion models to generate the 3D position of atoms and utilizes the discrete denoising diffusion process to generate the atom type. Compared to existing methods, our algorithm offers greater flexibility for post-processing and refining the generated molecules and demonstrates faster performance. We provide theoretical proof of the equivariance of the diffusion process for molecule position generation. Our model achieved better than state-of-the-art performance in molecule/atom stability and molecule validity on benchmarks generating 3D molecules.

Keywords: Graph Neural Networks · Diffusion Model · Generative AI · 3D Molecule Generations

Author Information

Weitong Zhang (University of California, Los Angeles)
Xiaoyun Wang (NVIDIA)
Justin Smith
Joe Eaton
Brad Rees
Quanquan Gu (University of California, Los Angeles)

More from the Same Authors