基于图编辑网络的分子图相似度学习结题报告

基于图编辑网络的分子图相似度学习结题报告一、研究背景与问题提出在cheminformatics（cheminformatics，即化学信息学）领域，分子结构的相似性评估是药物发现、材料设计和生物活性预测等任务的核心基础。传统的分子相似度计算方法，如基于指纹的Tanimoto系数、基于药效团的匹配算法等，虽然在工业界得到广泛应用，但存在明显的局限性。这些方法大多依赖于人工设计的特征表示，难以捕捉分子结构中复杂的空间拓扑关系和原子间的相互作用。例如，两个具有相似药理活性的分子可能具有完全不同的指纹特征，导致传统方法无法准确识别其相似性。随着深度学习技术的兴起，图神经网络（GraphNeuralNetworks,GNNs）为分子结构的建模提供了新的思路。分子可以自然地表示为图结构，其中原子是节点，化学键是边。GNNs能够通过消息传递机制学习分子的隐式表示，从而更全面地捕捉分子的结构信息。然而，现有的GNNs在处理分子相似度学习任务时，通常是将分子图映射到低维向量空间，然后通过计算向量间的距离来评估相似度。这种“嵌入后比较”的范式存在两个关键问题：一是向量空间中的距离度量并不总能准确反映分子图之间的结构相似性；二是模型缺乏对分子编辑操作（如原子替换、键的添加/删除）的显式建模，而这些操作正是药物化学家在分子优化过程中常用的手段。图编辑网络（GraphEditNetworks,GENs）是一类专门用于处理图结构数据的深度学习模型，其核心思想是通过学习图之间的编辑操作序列来度量图的相似性。与传统的GNNs不同，GENs直接建模分子图之间的转换过程，能够更精准地捕捉分子结构的细微变化。因此，本研究提出基于图编辑网络的分子图相似度学习方法，旨在解决传统方法和现有GNNs在分子相似度评估中的不足，为药物发现和材料设计提供更准确的工具。二、相关工作综述2.1传统分子相似度计算方法传统的分子相似度计算方法主要分为基于指纹的方法和基于药效团的方法。基于指纹的方法将分子编码为二进制向量（指纹），其中每个位表示分子中是否存在特定的子结构。常用的指纹包括MACCSkeys、Morgan指纹和RDKit指纹等。Tanimoto系数是衡量两个指纹相似度的常用指标，其计算方式为两个指纹交集的大小与并集大小的比值。然而，指纹的设计依赖于人工经验，无法捕捉分子的全局结构信息，且不同的指纹可能导致不同的相似度结果。基于药效团的方法则关注分子中与生物活性相关的特征，如氢键供体、氢键受体、疏水基团和带电中心等。药效团模型通过匹配分子中的药效团特征来评估相似度，适用于药物活性预测等任务。但药效团的定义和提取同样依赖于人工标注，且无法考虑分子的三维空间结构。2.2基于GNNs的分子相似度学习近年来，GNNs在分子表示学习中取得了显著进展。例如，GraphConvolutionalNetworks（GCNs）通过对邻居节点的特征进行聚合来更新节点表示；GraphAttentionNetworks（GATs）引入注意力机制，能够自适应地学习邻居节点的权重；MessagePassingNeuralNetworks（MPNNs）则统一了多种GNNs的消息传递框架。这些模型将分子图映射到低维向量空间，然后通过余弦相似度、欧氏距离等度量来计算分子间的相似度。然而，这些方法存在一定的局限性。首先，向量空间中的距离度量与分子图的结构相似性之间缺乏明确的对应关系。例如，两个分子图的向量表示可能在欧氏距离上很近，但在结构上却差异很大。其次，模型无法解释相似度的来源，即无法指出哪些原子或键的差异导致了相似度的变化。最后，现有的GNNs大多是为分子属性预测任务设计的，在相似度学习任务上的性能并未得到充分优化。2.3图编辑网络的研究进展图编辑网络的概念最早由Bosc等人于2019年提出，其核心是通过学习图之间的编辑操作序列来度量图的相似性。GENs通常由两个部分组成：编辑成本预测网络和编辑序列生成网络。编辑成本预测网络用于预测将一个图转换为另一个图所需的编辑操作（如节点插入、节点删除、节点替换、边插入、边删除、边替换）的成本；编辑序列生成网络则根据编辑成本生成最优的编辑序列。在分子图处理领域，已有一些研究尝试将GENs应用于分子优化和生成任务。例如，GraphEditDistanceNetworks（GEDNs）将图编辑距离作为分子相似度的度量，并通过GNNs来预测编辑成本。然而，这些方法大多关注于分子生成，而在分子相似度学习任务上的应用还比较有限。此外，现有的GENs在处理大规模分子图数据时，存在计算效率低下的问题，限制了其在实际工业场景中的应用。三、基于图编辑网络的分子图相似度学习方法3.1分子图的表示与编辑操作定义在本研究中，分子图被表示为$G=(V,E,X,A)$，其中$V$是节点（原子）的集合，$E$是边（化学键）的集合，$X\in\mathbb{R}^{|V|\timesd_v}$是节点特征矩阵，$A\in\mathbb{R}^{|V|\times|V|}$是邻接矩阵。节点特征$X$包含原子的类型、电荷、杂化状态等信息；邻接矩阵$A$中的元素$A_{ij}$表示节点$i$和节点$j$之间的化学键类型（如单键、双键、三键）。分子图之间的编辑操作定义为以下六种类型：节点替换：将分子图$G_1$中的节点$u$替换为分子图$G_2$中的节点$v$，记为$\text{Sub}(u,v)$；节点插入：在分子图$G_1$中插入一个新的节点$v$，记为$\text{Ins}(v)$；节点删除：从分子图$G_1$中删除节点$u$，记为$\text{Del}(u)$；边替换：将分子图$G_1$中节点$u$和$v$之间的边替换为分子图$G_2$中的边，记为$\text{Sub}(u,v,e)$；边插入：在分子图$G_1$中的节点$u$和$v$之间插入一条边$e$，记为$\text{Ins}(u,v,e)$；边删除：从分子图$G_1$中删除节点$u$和$v$之间的边，记为$\text{Del}(u,v)$。每种编辑操作都对应一个编辑成本，编辑成本的大小反映了该操作的难度或代价。例如，将一个碳原子替换为氧原子的成本可能低于将碳原子替换为金属原子的成本；插入一个双键的成本可能高于插入一个单键的成本。3.2图编辑网络模型架构本研究提出的图编辑网络模型主要由三个部分组成：节点嵌入模块、编辑成本预测模块和相似度计算模块。3.2.1节点嵌入模块节点嵌入模块的作用是将分子图中的节点特征转换为低维的隐式表示，以便后续的编辑成本预测。我们采用图注意力网络（GATs）作为节点嵌入模块的基础模型，因为GATs能够通过注意力机制自适应地学习邻居节点的权重，从而更好地捕捉分子图的局部结构信息。具体来说，对于每个节点$u$，其隐式表示$h_u$通过以下方式计算：$$h_u=\sigma\left(\sum_{v\in\mathcal{N}(u)}\alpha_{uv}Wh_v\right)$$其中，$\mathcal{N}(u)$是节点$u$的邻居节点集合，$\alpha_{uv}$是节点$u$对邻居节点$v$的注意力权重，$W$是可学习的权重矩阵，$\sigma$是非线性激活函数（如ReLU）。注意力权重$\alpha_{uv}$通过以下方式计算：$$\alpha_{uv}=\frac{\exp\left(\text{LeakyReLU}\left(\mathbf{a}^T[Wh_u\parallelWh_v]\right)\right)}{\sum_{k\in\mathcal{N}(u)}\exp\left(\text{LeakyReLU}\left(\mathbf{a}^T[Wh_u\parallelWh_k]\right)\right)}$$其中，$\mathbf{a}$是注意力向量，$\parallel$表示向量拼接操作。3.2.2编辑成本预测模块编辑成本预测模块的目标是学习每种编辑操作的成本函数。对于节点替换操作$\text{Sub}(u,v)$，其编辑成本$c_{\text{Sub}}(u,v)$定义为节点$u$和节点$v$的隐式表示之间的距离：$$c_{\text{Sub}}(u,v)=|h_u-h_v|_2$$对于节点插入操作$\text{Ins}(v)$，其编辑成本$c_{\text{Ins}}(v)$定义为节点$v$的隐式表示的范数：$$c_{\text{Ins}}(v)=|h_v|_2$$对于节点删除操作$\text{Del}(u)$，其编辑成本$c_{\text{Del}}(u)$与节点插入操作对称，定义为节点$u$的隐式表示的范数：$$c_{\text{Del}}(u)=|h_u|_2$$对于边编辑操作，我们首先通过节点对的隐式表示计算边的隐式表示。对于节点对$(u,v)$，其边的隐式表示$e_{uv}$定义为：$$e_{uv}=\tanh\left(W_e[h_u\parallelh_v]\right)$$其中，$W_e$是可学习的权重矩阵。然后，边替换操作$\text{Sub}(u,v,e)$的编辑成本$c_{\text{Sub}}(u,v,e)$定义为原边的隐式表示$e_{uv}$与目标边的隐式表示$e'_{uv}$之间的距离：$$c_{\text{Sub}}(u,v,e)=|e_{uv}-e'_{uv}|_2$$边插入操作$\text{Ins}(u,v,e)$的编辑成本$c_{\text{Ins}}(u,v,e)$定义为目标边的隐式表示的范数：$$c_{\text{Ins}}(u,v,e)=|e'_{uv}|_2$$边删除操作$\text{Del}(u,v)$的编辑成本$c_{\text{Del}}(u,v)$定义为原边的隐式表示的范数：$$c_{\text{Del}}(u,v)=|e_{uv}|_2$$为了提高模型的泛化能力，我们在编辑成本函数中引入了可学习的参数，使得模型能够根据训练数据自动调整不同编辑操作的成本权重。例如，对于节点替换操作，我们将编辑成本函数修改为：$$c_{\text{Sub}}(u,v)=\theta_{\text{Sub}}\cdot|h_u-h_v|_2$$其中，$\theta_{\text{Sub}}$是可学习的参数。类似地，我们为其他编辑操作也引入了相应的可学习参数。3.2.3相似度计算模块相似度计算模块的目标是根据编辑成本预测模块输出的编辑成本，计算两个分子图之间的相似度。图编辑距离（GraphEditDistance,GED）是衡量两个图之间相似性的经典指标，其定义为将一个图转换为另一个图所需的最小编辑成本总和。然而，精确计算GNP是一个NP难问题，对于大规模分子图来说计算效率极低。因此，我们采用近似算法来计算GED的下界，作为分子图相似度的度量。具体来说，我们采用匈牙利算法来求解节点之间的最优匹配，然后基于节点匹配结果计算边的编辑成本。对于两个分子图$G_1$和$G_2$，设其节点数分别为$n_1$和$n_2$，我们构造一个$n_1\timesn_2$的成本矩阵$C$，其中$C_{ij}$表示将$G_1$中的节点$i$替换为$G_2$中的节点$j$的编辑成本。通过匈牙利算法找到节点之间的最优匹配$\pi$，使得匹配的总成本最小：$$C_{\text{node}}=\sum_{i=1}^{n_1}C_{i,\pi(i)}$$对于未匹配的节点，我们分别计算节点插入和删除的成本：$$C_{\text{insert}}=\sum_{j\notin\pi(G_1)}c_{\text{Ins}}(j)$$$$C_{\text{delete}}=\sum_{i\notin\pi^{-1}(G_2)}c_{\text{Del}}(i)$$然后，我们基于节点匹配结果计算边的编辑成本。对于$G_1$中的每条边$(i,k)$，如果$\pi(i)$和$\pi(k)$在$G_2$中存在边，则计算边替换成本；否则，计算边删除成本。对于$G_2$中的每条边$(j,l)$，如果$\pi^{-1}(j)$和$\pi^{-1}(l)$在$G_1$中不存在边，则计算边插入成本。边的总编辑成本为：$$C_{\text{edge}}=C_{\text{edge-sub}}+C_{\text{edge-del}}+C_{\text{edge-ins}}$$最后，两个分子图之间的近似GED为：$$\text{GED}(G_1,G_2)=C_{\text{node}}+C_{\text{insert}}+C_{\text{delete}}+C_{\text{edge}}$$分子图之间的相似度则定义为GED的倒数：$$\text{Sim}(G_1,G_2)=\frac{1}{1+\text{GED}(G_1,G_2)}$$3.3模型训练与优化本研究采用监督学习的方式训练图编辑网络模型。训练数据由成对的分子图及其对应的相似度标签组成。相似度标签可以来自于实验测定的生物活性数据，也可以通过传统方法（如Tanimoto系数）计算得到。在本研究中，我们使用ChEMBL数据库中的生物活性数据来构建训练数据集，其中两个分子的相似度标签定义为它们对同一靶点的生物活性的相关性。模型的损失函数定义为预测相似度与真实相似度之间的均方误差（MSE）：$$\mathcal{L}=\frac{1}{N}\sum_{i=1}^{N}\left(\text{Sim}(G_{1i},G_{2i})-\text{Sim}^*(G_{1i},G_{2i})\right)^2$$其中，$N$是训练样本的数量，$\text{Sim}^*(G_{1i},G_{2i})$是真实的相似度标签。为了提高模型的训练效率和泛化能力，我们采用以下优化策略：批量训练：将训练数据划分为小批量，每次训练一个批量的数据，以减少内存消耗并加速训练过程；学习率衰减：在训练过程中逐渐降低学习率，以避免模型陷入局部最优解；正则化：在损失函数中加入L2正则化项，以防止模型过拟合；早停：当验证集上的损失不再下降时，提前停止训练，以避免过拟合。三、实验设计与结果分析3.1数据集与评价指标本研究使用三个公开的分子数据集进行实验，分别是：ChEMBL数据集：包含约100万个分子的生物活性数据，我们从中筛选出针对同一靶点的分子对，构建了一个包含50万个分子对的训练数据集和10万个分子对的测试数据集；ZINC数据集：包含约1000万个类药物分子，我们从中随机选择10万个分子，生成了包含20万个分子对的数据集，用于评估模型在无监督学习场景下的性能；QM9数据集：包含约13万个小分子的量子化学性质数据，我们使用分子的能量差作为相似度标签，构建了一个包含5万个分子对的数据集，用于评估模型在分子性质预测任务中的性能。为了全面评估模型的性能，我们采用以下评价指标：均方误差（MSE）：衡量预测相似度与真实相似度之间的平均平方误差，MSE越小表示模型的预测精度越高；皮尔逊相关系数（Pearson'sr）：衡量预测相似度与真实相似度之间的线性相关性，r越接近1表示模型的预测结果与真实结果越一致；排序准确率（RankAccuracy）：衡量模型对分子对的相似度排序能力，具体来说，对于每个查询分子，我们计算其与数据集中其他分子的相似度，并将相似度从高到低排序，然后计算前k个分子中与查询分子具有相似生物活性的比例；计算效率：衡量模型处理分子对的速度，以每秒处理的分子对数量为指标。3.2对比实验设置为了验证本研究提出的基于图编辑网络的分子图相似度学习方法的有效性，我们与以下几种主流的分子相似度计算方法进行对比：Tanimoto系数（Morgan指纹）：基于Morgan指纹计算分子对的Tanimoto系数；GCN+余弦相似度：使用图卷积网络（GCNs）将分子图映射到低维向量空间，然后计算向量间的余弦相似度；GAT+余弦相似度：使用图注意力网络（GATs）将分子图映射到低维向量空间，然后计算向量间的余弦相似度；GraphSAGE+余弦相似度：使用GraphSAGE将分子图映射到低维向量空间，然后计算向量间的余弦相似度；DeepGraphMatch：一种基于GNNs的图匹配模型，通过学习节点之间的匹配概率来计算分子图的相似度。所有对比模型均使用相同的训练数据集进行训练，并在相同的测试数据集上进行评估。模型的超参数通过网格搜索进行优化，以确保所有模型都处于最佳性能状态。3.3实验结果与分析3.3.1预测精度对比表1展示了不同模型在ChEMBL数据集上的MSE和Pearson'sr指标。从表中可以看出，本研究提出的基于图编辑网络的方法在MSE和Pearson'sr指标上均显著优于其他对比模型。具体来说，本方法的MSE为0.023，比次优的DeepGraphMatch模型低约15%；Pearson'sr为0.921，比DeepGraphMatch模型高约5%。这表明本方法能够更准确地预测分子对的相似度，尤其是在捕捉分子结构的细微变化方面具有明显优势。表1不同模型在ChEMBL数据集上的预测精度对比模型MSEPearson'srTanimoto系数0.0870.723GCN+余弦相似度0.0450.856GAT+余弦相似度0.0390.878GraphSAGE+余弦相似度0.0420.867DeepGraphMatch0.0270.876本研究方法0.0230.921在ZINC数据集上，由于没有真实的相似度标签，我们采用无监督学习的方式训练模型，并通过评估模型对分子聚类的性能来间接衡量其相似度学习能力。图1展示了不同模型在ZINC数据集上的聚类结果的轮廓系数（SilhouetteCoefficient）。轮廓系数是衡量聚类质量的常用指标，其取值范围为[-1,1]，值越接近1表示聚类效果越好。从图中可以看出，本研究方法的轮廓系数为0.78，显著高于其他对比模型，表明本方法能够更准确地将相似的分子聚类在一起。3.3.2排序能力对比表2展示了不同模型在ChEMBL数据集上的排序准确率指标。我们分别计算了前10、前50和前100个分子的排序准确率。从表中可以看出，本研究方法在所有排序准确率指标上均显著优于其他对比模型。例如，前10个分子的排序准确率为0.89，比次优的DeepGraphMatch模型高约12%；前100个分子的排序准确率为0.76，比DeepGraphMatch模型高约8%。这表明本方法能够更准确地将与查询分子相似的分子排在前面，对于药物发现中的虚拟筛选任务具有重要意义。表2不同模型在ChEMBL数据集上的排序准确率对比模型前10个分子前50个分子前100个分子Tanimoto系数0.620.510.45GCN+余弦相似度0.750.630.57GAT+余弦相似度0.780.660.60GraphSAGE+余弦相似度0.760.640.58DeepGraphMatch0.790.700.70本研究方法0.890.780.763.3.3计算效率对比表3展示了不同模型在处理分子对时的计算效率。从表中可以看出，本研究方法的计算效率略低于基于GNNs的“嵌入后比较”方法，但显著高于DeepGraphMatch模型。具体来说，本方法每秒能够处理约120个分子对，而DeepGraphMatch模型每秒只能处理约30个分子对。这是因为本方法采用了近似算法来计算GED，避免了精确计算GED的高复杂度。虽然本方法的计算效率不如基于GNNs的方法，但在实际应用中，每秒120个分子对的处理速度已经能够满足大多数药物发现任务的需求。表3不同模型的计算效率对比模型每秒处理分子对数量Tanimoto系数1200GC