arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.04313cs.LGcs.AIq-bio.BMq-bio.QMstat.ML
收藏

几何完备扩散用于3D分子生成与优化

Geometry-Complete Diffusion for 3D Molecule Generation and Optimization

University of Missouri(密苏里大学)

Alex Morehead, Jianlin Cheng

AI总结:

提出几何完备扩散模型(GCDM),通过几何完备去噪网络生成3D分子,在QM9和GEOM-Drugs数据集上显著优于现有方法,并能生成有效稳定的大型分子及优化分子性质。

摘要:

去噪扩散概率模型(DDPMs)在计算机视觉和计算生物学等多个领域开创了新的最先进成果,应用范围从文本引导的图像生成到结构引导的蛋白质设计。在蛋白质设计这一研究方向中,近期已有方法在DDPM框架内使用等变图神经网络(GNNs)生成3D分子。然而,这些方法采用与分子无关且非几何的GNN作为其3D图去噪网络,无法学习3D分子的重要几何性质,这显著阻碍了它们生成有效大型3D分子的能力。在本工作中,我们通过引入用于3D分子生成的几何完备扩散模型(GCDM)来解决这些不足,该模型在QM9数据集的条件生成和无条件生成设置以及更大的GEOM-Drugs数据集上,均显著优于现有的3D分子扩散模型,并且在QM9数据集上生成更多新颖且独特的无条件3D分子。重要的是,我们证明了GCDM为3D分子生成而学习的几何完备去噪过程,使模型能够在GEOM-Drugs规模下生成相当比例的有效且能量稳定的大型分子,而先前的方法无法利用其学习的特征做到这一点。此外,我们展示了GCDM的扩展不仅能有效设计针对特定蛋白质口袋的3D分子,而且GCDM的几何特征可被重新利用,以持续优化现有3D分子的几何结构和化学组成,以实现分子稳定性和性质特异性,展示了分子扩散模型的新多功能性。我们的源代码和数据可在https://github.com/BioinfoMachineLearning/Bio-Diffusion免费获取。

英文摘要:

Denoising diffusion probabilistic models (DDPMs) have pioneered new state-of-the-art results in disciplines such as computer vision and computational biology for diverse tasks ranging from text-guided image generation to structure-guided protein design. Along this latter line of research, methods have recently been proposed for generating 3D molecules using equivariant graph neural networks (GNNs) within a DDPM framework. However, such methods are unable to learn important geometric properties of 3D molecules, as they adopt molecule-agnostic and non-geometric GNNs as their 3D graph denoising networks, which notably hinders their ability to generate valid large 3D molecules. In this work, we address these gaps by introducing the Geometry-Complete Diffusion Model (GCDM) for 3D molecule generation, which outperforms existing 3D molecular diffusion models by significant margins across conditional and unconditional settings for the QM9 dataset and the larger GEOM-Drugs dataset, respectively, and generates more novel and unique unconditional 3D molecules for the QM9 dataset compared to previous methods. Importantly, we demonstrate that the geometry-complete denoising process of GCDM learned for 3D molecule generation enables the model to generate a significant proportion of valid and energetically-stable large molecules at the scale of GEOM-Drugs, whereas previous methods fail to do so with the features they learn. Additionally, we show that extensions of GCDM can not only effectively design 3D molecules for specific protein pockets but also that GCDM's geometric features can be repurposed to consistently optimize the geometry and chemical composition of existing 3D molecules for molecular stability and property specificity, demonstrating new versatility of molecular diffusion models. Our source code and data are freely available at https://github.com/BioinfoMachineLearning/Bio-Diffusion.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑