用可平移旋转的模糊扩散模型分步生成分子三维构象。
Equivariant Blurring Diffusion for Hierarchical Molecular Conformer Generation
- 分两阶段生成:先建片段粗略结构,再细化原子细节。
- 在药物分子数据集上优于现有去噪扩散模型,构象精度更高。
- 适合分子生成、药物设计研究者使用,支持多尺度建模。
如何让扩散模型以从粗到细的方式处理3D几何结构,类似人类对世界的多尺度认知?本文聚焦于在分子图条件下生成3D分子构象这一基础生化问题,提出一种分层方法:第一阶段从分子图生成片段级粗略3D结构;第二阶段基于粗略结构生成原子级精细细节,同时允许粗略结构动态调整。针对第二阶段中需保持粗粒度信息且保证SE(3)等变性的挑战,提出新型生成模型Equivariant Blurring Diffusion (EBD),其前向过程通过模糊原子细节向片段级粗略结构演进,反向过程则利用等变网络实现逆向恢复。在药物样分子基准测试中,通过几何与化学对比验证了EBD的有效性。消融实验深入分析了损失函数设计与数据扰动策略对性能的影响。代码已开源:https://github.com/Shen-Lab/EBD。
原文摘要 · Abstract (English)
How can diffusion models process 3D geometries in a coarse-to-fine manner, akin to our multiscale view of the world? In this paper, we address the question by focusing on a fundamental biochemical problem of generating 3D molecular conformers conditioned on molecular graphs in a multiscale manner. Our approach consists of two hierarchical stages: i) generation of coarse-grained fragment-level 3D structure from the molecular graph, and ii) generation of fine atomic details from the coarse-grained approximated structure while allowing the latter to be adjusted simultaneously. For the challenging second stage, which demands preserving coarse-grained information while ensuring SE(3) equivariance, we introduce a novel generative model termed Equivariant Blurring Diffusion (EBD), which defines a forward process that moves towards the fragment-level coarse-grained structure by blurring the fine atomic details of conformers, and a reverse process that performs the opposite operation using equivariant networks. We demonstrate the effectiveness of EBD by geometric and chemical comparison to state-of-the-art denoising diffusion models on a benchmark of drug-like molecules. Ablation studies draw insights on the design of EBD by thoroughly analyzing its architecture, which includes the design of the loss function and the data corruption process. Codes are released at https://github.com/Shen-Lab/EBD .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。