arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.02683cs.LGcs.AIq-bio.BM
收藏

用于3D分子生成的几何促进去噪扩散模型

Geometric-Facilitated Denoising Diffusion Model for 3D Molecule Generation

Zhejiang Lab(浙江实验室)

Can Xu, Haosen Wang, Weigang Wang, Pengfei Zheng, Hongyang Chen

AI总结:

针对3D分子生成中多体关系建模和键预测的挑战,提出GFMDiff方法,通过双轨Transformer网络和几何促进损失提升生成质量。

摘要:

去噪扩散模型在多个研究领域展现出巨大潜力。现有的基于扩散的从头3D分子生成方法面临两大挑战。由于分子中大多数重原子允许通过单键连接多个原子,仅使用成对距离来建模分子几何结构是不充分的。因此,第一个挑战是提出一种有效的神经网络作为去噪核,能够捕获复杂的多体原子间关系并学习高质量特征。由于图的离散性质,主流的基于扩散的分子方法严重依赖预定义规则并以间接方式生成边。第二个挑战是使分子生成适应扩散过程并准确预测键的存在。在我们的研究中,我们认为扩散过程中更新分子构象的迭代方式与分子动力学一致,并引入了一种名为Geometric-Facilitated Molecular Diffusion (GFMDiff)的新型分子生成方法。针对第一个挑战,我们引入了Dual-Track Transformer Network (DTN)以充分挖掘全局空间关系并学习高质量表示,从而有助于准确预测特征和几何结构。对于第二个挑战,我们设计了Geometric-Facilitated Loss (GFLoss),它在训练期间干预键的形成,而不是直接将边嵌入到潜在空间中。在当前基准上的综合实验证明了GFMDiff的优越性。

英文摘要:

Denoising diffusion models have shown great potential in multiple research areas. Existing diffusion-based generative methods on de novo 3D molecule generation face two major challenges. Since majority heavy atoms in molecules allow connections to multiple atoms through single bonds, solely using pair-wise distance to model molecule geometries is insufficient. Therefore, the first one involves proposing an effective neural network as the denoising kernel that is capable to capture complex multi-body interatomic relationships and learn high-quality features. Due to the discrete nature of graphs, mainstream diffusion-based methods for molecules heavily rely on predefined rules and generate edges in an indirect manner. The second challenge involves accommodating molecule generation to diffusion and accurately predicting the existence of bonds. In our research, we view the iterative way of updating molecule conformations in diffusion process is consistent with molecular dynamics and introduce a novel molecule generation method named Geometric-Facilitated Molecular Diffusion (GFMDiff). For the first challenge, we introduce a Dual-Track Transformer Network (DTN) to fully excevate global spatial relationships and learn high quality representations which contribute to accurate predictions of features and geometries. As for the second challenge, we design Geometric-Facilitated Loss (GFLoss) which intervenes the formation of bonds during the training period, instead of directly embedding edges into the latent space. Comprehensive experiments on current benchmarks demonstrate the superiority of GFMDiff.

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

↑