Poster
in
Workshop: Structured Probabilistic Inference and Generative Modeling
DiffMol: 3D Structured Molecule Generation with Discrete Denoising Diffusion Probabilistic Models
Weitong Zhang · Xiaoyun Wang · Justin Smith · Joe Eaton · Brad Rees · Quanquan Gu
Keywords: [ Generative AI ] [ Diffusion Model ] [ Graph Neural Networks ] [ 3D Molecule Generations ]
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.