高级检索

    基于注意力等变几何扩散模型的完整分子生成

    Complete Molecular Generation Based on Attention-Equivariant Geometric Diffusion Models

    • 摘要: 针对分子生成任务,本文提出了一种注意力等变几何扩散模型(AEGDM),该模型应用离散扩散与连续扩散混合生成扩散框架,实现原子坐标、原子种类及化学键的完整生成。在AEGDM模型中,注意力等变图神经网络(Attention Equivariant Graph Neural Network,AEGNN)与残差增强神经网络(Residual Boosting Neural Network,RBNN)分别用以提高模型的表征能力及加快模型训练速度。在QM9数据集上的实验结果表明,该模型在生成分子的唯一性与新颖性方面均取得显著提升,唯一性达到98.2%,新颖性达到70.9%。这一性能提升表明AEGDM在化学空间探索方面具有较强的表征能力,能够有效推动创新性候选分子的发展。

       

      Abstract: This paper proposes an attention-equivariant diffusion model (comprising AEGNN and RBNN) for molecular generation tasks. The AEGNN leverages multi-head self-attention to jointly update edge features, atomic coordinates, and node features within a molecular graph. Subject to strict rotational and translational equivariance, it progressively reconstructs atom types, three-dimensional structures, and chemical bonds via a reverse diffusion process. The RBNN further improves generative performance through co-training two submodels, both built by stacking identical basic modules (denoted as AEM in this paper). The Knowledge Generator (KG) adopts a deeper layer stack to capture complex nonlinear relationships, producing high-precision initial predictions that align with the ground truth. By contrast, the Residual Refiner (RR) adopts a shallower architecture, focusing on fitting the residual between the KG’s output and the true target values. This design cuts computational overhead while boosting the model’s error correction capacity. The two modules are connected in a cascade structure: the output of the KG acts as the input to the RR, and the final prediction is obtained by adding the residual correction from the RR to the initial output of the KG. Experiments conducted on the QM9 dataset show substantial improvements in the uniqueness and novelty of generated molecules, demonstrating that the AEGDM can efficiently explore chemical space and aid the discovery of structurally innovative candidate molecules. Moreover, the RBNN mechanism speeds up experimental iteration and elevates overall model performance, offering critical technical support for the iterative optimization of molecular generation frameworks.

       

    /

    返回文章
    返回