DMol: A Highly Efficient and Chemical Motif-Preserving Molecule Generation Platform

Peizhi Niu (University of Illinois at Urbana-Champaign) · Yu-Hsiang Wang (University of Illinois at Urbana-Champaign) · Vishal Rana (University of Illinois, Urbana-Champaign) · Chetan Rupakheti (AbbVie Inc.) · Abhishek Pandey (Abbvie) · Olgica Milenkovic (University of Illinois at Urbana-Champaign)
benchmarking datasetschembl-likenessdefog methoddiffusion stepsdigress modeldrug molecule generationgraph diffusion modelloss function modificationmolecular scaffold integrationmotif compressionmotif-conservationnode and edge noise schedulesqedshingles distance scoressmiles validitysubgraph perturbations

We introduce a new graph diffusion model for small drug molecule generation which simultaneously offers a 10-fold reduction in the number of diffusion steps when compared to existing methods, preservation of small molecule graph motifs via motif compression, and an average 3\% improvement in SMILES validity over the DiGress model across all real-world molecule benchmarking datasets. Furthermore, our approach outperforms the state-of-the-art DeFoG method with respect to motif-conservation by roughly 4\%, as evidenced by high ChEMBL-likeness, QED and newly introduced shingles distance scores. The key ideas behind the approach are to use a combination of deterministic and random subgraph perturbations, so that the node and edge noise schedules are codependent; to modify the loss function of the training process in order to exploit the deterministic component of the schedule; and, to ''compress'' a collection of highly relevant carbon ring and other motif structures into supernodes in a way that allows for simple subsequent integration into the molecular scaffold.