基于图神经网络的药物组合预测结题报告

基于图神经网络的药物组合预测结题报告一、研究背景与问题提出在复杂疾病治疗领域，单一药物往往难以应对多靶点、多通路的病理机制，药物联合治疗凭借其协同作用、降低毒副作用、延缓耐药性等优势，逐渐成为癌症、糖尿病、艾滋病等重大疾病的重要治疗策略。然而，传统的药物组合研发模式依赖于大规模湿实验筛选，不仅耗时费力、成本高昂，且受限于实验条件和研究者经验，难以实现对海量潜在药物组合的高效探索。据统计，目前已知的药物分子超过20000种，理论上的药物组合数量可达10^8量级，如此庞大的组合空间使得传统实验方法几乎不可能完成全面筛选。随着生物信息学和人工智能技术的快速发展，基于计算模型的药物组合预测为解决这一难题提供了新的思路。早期的计算方法主要基于机器学习算法，如支持向量机、随机森林等，通过提取药物的化学结构、靶点信息、基因表达谱等特征，构建预测模型。但这些方法大多依赖于人工提取的特征，难以捕捉药物-药物、药物-靶点之间复杂的非线性关系和潜在的生物网络交互信息。图神经网络（GraphNeuralNetworks,GNNs）作为一种专门处理图结构数据的深度学习模型，能够直接以图的形式对生物分子间的相互作用进行建模，自动学习节点和边的隐藏特征，为挖掘药物组合的潜在机制提供了强大的工具。本研究旨在构建基于图神经网络的药物组合预测模型，充分整合多源生物数据，实现对有效药物组合的精准预测，为药物联合治疗的研发提供理论支持和技术指导。二、相关研究现状（一）药物组合预测的传统计算方法传统的药物组合预测方法主要分为基于相似性、基于网络和基于机器学习三类。基于相似性的方法假设具有相似化学结构或作用靶点的药物可能具有相似的组合效果，通过计算药物之间的相似度来预测潜在的组合。例如，基于药物化学结构相似性的方法通过比较药物分子的指纹特征，筛选出结构相似的药物进行组合预测；基于靶点相似性的方法则通过分析药物作用靶点的重叠程度来推断组合效果。这类方法简单直观，但忽略了药物在生物网络中的复杂交互作用，预测精度有限。基于网络的方法将药物、靶点、基因等生物实体抽象为网络节点，实体间的相互作用作为边，通过分析网络的拓扑结构和节点属性来预测药物组合。例如，基于药物-靶点相互作用网络的方法，通过计算药物对在网络中的路径长度、共同邻居数量等指标，评估药物组合的潜在协同作用；基于基因共表达网络的方法则通过分析药物处理后基因表达的变化模式，识别具有协同调节作用的药物组合。这类方法能够在一定程度上反映生物系统的整体性，但对网络结构的依赖较强，且难以处理高维、异质的生物数据。基于机器学习的方法是目前药物组合预测的主流方法之一，通过提取药物的多源特征，构建分类或回归模型来预测药物组合的效果。常用的特征包括药物的化学结构描述符、靶点序列信息、基因表达谱、药物副作用等。例如，有研究人员利用药物的分子指纹和靶点信息作为特征，采用随机森林算法构建了药物组合协同作用预测模型；还有研究通过整合药物的基因表达数据和蛋白质相互作用网络，利用深度学习模型进行药物组合预测。然而，这些方法大多需要人工设计特征，难以充分挖掘生物数据中的复杂模式，且模型的可解释性较差。（二）图神经网络在生物信息学中的应用图神经网络自提出以来，在生物信息学领域得到了广泛的应用，包括蛋白质结构预测、药物靶点预测、基因功能注释、疾病诊断等多个方面。在药物研发领域，图神经网络主要用于药物分子的表示学习、药物-靶点相互作用预测、药物副作用预测等任务。在药物分子表示学习方面，研究人员将药物分子建模为图结构，其中原子作为节点，化学键作为边，利用图神经网络学习药物分子的向量表示。例如，GraphConvolutionalNetworks（GCNs）、GraphAttentionNetworks（GATs）等模型能够自动捕捉药物分子的局部结构特征和全局信息，生成的分子表示可用于后续的药物活性预测、药物筛选等任务。在药物-靶点相互作用预测方面，图神经网络能够整合药物分子图和蛋白质序列信息，构建端到端的预测模型。例如，有研究提出了一种基于图神经网络和卷积神经网络的混合模型，分别对药物分子图和蛋白质序列进行编码，然后通过融合层将两者的特征进行整合，预测药物与靶点的相互作用。这类方法能够有效捕捉药物-靶点之间的潜在结合模式，提高预测精度。（三）基于图神经网络的药物组合预测研究进展近年来，越来越多的研究开始将图神经网络应用于药物组合预测任务。早期的研究主要基于药物-靶点相互作用网络，利用图神经网络学习药物和靶点的嵌入表示，然后通过计算药物对的嵌入表示的相似度或进行分类预测，判断药物组合是否具有协同作用。例如，有研究构建了药物-靶点二部图，采用图卷积网络对药物和靶点节点进行嵌入学习，然后利用多层感知机预测药物组合的协同效果。随着研究的深入，一些研究开始整合更多的生物数据，如基因表达谱、蛋白质相互作用网络、疾病相关基因等，构建更复杂的生物网络，以提高模型的预测性能。例如，有研究提出了一种基于多图融合的图神经网络模型，将药物分子图、药物-靶点相互作用图和基因共表达图进行融合，学习药物和基因的联合表示，从而预测药物组合对疾病的治疗效果。还有研究考虑了药物组合在细胞水平的作用机制，通过整合药物处理后的单细胞RNA测序数据，构建细胞状态转移图，利用图神经网络预测药物组合的协同作用。尽管基于图神经网络的药物组合预测研究取得了一定的进展，但仍存在一些问题和挑战。例如，如何有效整合多源异质的生物数据，如何提高模型的可解释性，如何处理数据不平衡问题等，这些问题都需要进一步的研究和解决。三、研究内容与方法（一）数据收集与预处理本研究收集了多源生物数据，包括药物化学结构数据、药物-靶点相互作用数据、基因表达谱数据、药物组合实验数据等，具体数据来源如下：药物化学结构数据：从DrugBank、PubChem等数据库获取药物的SMILES（SimplifiedMolecular-InputLine-EntrySystem）字符串，用于构建药物分子图。药物-靶点相互作用数据：从DrugBank、ChEMBL等数据库获取药物与靶点的相互作用信息，包括靶点的蛋白质序列和基因名称。基因表达谱数据：从GEO（GeneExpressionOmnibus）数据库获取癌细胞系在药物处理前后的基因表达数据，用于分析药物对细胞转录组的影响。药物组合实验数据：从DrugComb、SynergyDB等数据库获取已报道的药物组合实验数据，包括药物对、细胞系、协同作用评分等信息，作为模型训练和测试的标签。为了保证数据的质量和一致性，对收集到的数据进行了预处理：药物数据预处理：对药物的SMILES字符串进行标准化处理，去除无效字符和重复条目；利用RDKit工具将SMILES字符串转换为分子图结构，提取原子和化学键的特征。靶点数据预处理：对靶点的蛋白质序列进行清洗，去除缺失值和异常序列；利用ProtTrans工具将蛋白质序列转换为向量表示，作为靶点的初始特征。基因表达数据预处理：对基因表达谱数据进行归一化处理，去除批次效应；采用差异表达分析方法，筛选出药物处理后显著差异表达的基因，作为药物作用的基因特征。药物组合数据预处理：对药物组合实验数据进行去重和筛选，保留具有明确协同作用评分的样本；根据协同作用评分将药物组合分为协同、拮抗和无显著作用三类，构建分类任务的标签。（二）图神经网络模型构建本研究构建了一种基于多图融合的图神经网络模型（Multi-GraphFusionGraphNeuralNetwork,MFGNN），用于药物组合预测。模型主要由药物分子图编码模块、药物-靶点相互作用图编码模块、基因共表达图编码模块和多图融合预测模块四部分组成，具体结构如下：1.药物分子图编码模块该模块主要用于对药物分子的化学结构进行编码，学习药物分子的隐藏特征。采用图卷积网络（GCN）作为基础模型，将药物分子表示为图结构，其中原子作为节点，节点特征包括原子类型、电荷、杂化状态等；化学键作为边，边特征包括键类型、键长等。通过多层图卷积操作，将节点的邻居信息聚合到自身，更新节点的特征表示。具体的图卷积操作公式如下：[H^{(l+1)}=\sigma(\tilde{D}^{-1/2}\tilde{A}\tilde{D}^{-1/2}H^{(l)}W^{(l)})]其中，(H^{(l)})表示第(l)层的节点特征矩阵，(\tilde{A}=A+I)为添加自环的邻接矩阵，(\tilde{D})为(\tilde{A})的度矩阵，(W^{(l)})为可学习的权重矩阵，(\sigma)为激活函数。为了捕捉药物分子的全局结构信息，在图卷积操作后，采用全局平均池化（GlobalAveragePooling,GAP）和全局最大池化（GlobalMaxPooling,GMP）相结合的方式，将节点特征聚合为药物分子的向量表示：[h_d=\text{Concat}(\text{GAP}(H^{(L)}),\text{GMP}(H^{(L)}))]其中，(H^{(L)})为图卷积网络最后一层的节点特征矩阵，(h_d)为药物分子的向量表示。2.药物-靶点相互作用图编码模块该模块用于建模药物与靶点之间的相互作用关系，学习药物和靶点的联合特征。构建药物-靶点二部图，其中药物节点和靶点节点分别作为二部图的两类节点，药物与靶点之间的相互作用作为边。采用图注意力网络（GAT）对二部图进行编码，通过注意力机制学习节点之间的依赖关系，自动分配不同邻居节点的权重。具体的图注意力操作公式如下：[e_{ij}=\text{LeakyReLU}(\vec{a}^T[Wh_i\parallelWh_j])][\alpha_{ij}=\text{Softmax}j(e{ij})=\frac{\exp(e_{ij})}{\sum_{k\in\mathcal{N}(i)}\exp(e_{ik})}][h_i'=\sigma(\sum_{j\in\mathcal{N}(i)}\alpha_{ij}Wh_j)]其中，(h_i)和(h_j)分别为节点(i)和节点(j)的特征向量，(W)为可学习的权重矩阵，(\vec{a})为注意力向量，(\mathcal{N}(i))为节点(i)的邻居节点集合，(\alpha_{ij})为节点(i)对节点(j)的注意力权重，(h_i')为更新后的节点特征向量。通过多层图注意力网络的操作，分别得到药物节点和靶点节点的嵌入表示，然后将药物节点的嵌入表示与药物分子图编码模块得到的药物分子向量表示进行融合，得到药物的综合特征表示。3.基因共表达图编码模块该模块用于分析药物处理后基因表达的变化模式，学习药物作用的基因特征。构建基因共表达图，其中基因作为节点，基因之间的共表达关系作为边，边的权重由基因表达的皮尔逊相关系数计算得到。采用图SAGE（GraphSampleandAggregate）模型对基因共表达图进行编码，通过采样邻居节点并聚合其特征，学习基因的嵌入表示。图SAGE的聚合操作支持多种聚合函数，如均值聚合、最大聚合、LSTM聚合等，本研究采用均值聚合函数，具体公式如下：[h_v^{(l+1)}=\sigma(W^{(l)}\cdot\text{Mean}({h_v^{(l)}}\cup{h_u^{(l)},u\in\mathcal{N}(v)}))]其中，(h_v^{(l)})表示第(l)层节点(v)的特征向量，(\mathcal{N}(v))为节点(v)的邻居节点集合，(\text{Mean})为均值聚合函数，(W^{(l)})为可学习的权重矩阵，(\sigma)为激活函数。通过图SAGE模型学习得到基因的嵌入表示后，根据药物处理后差异表达基因的权重，将基因特征聚合为药物的基因层面特征表示。4.多图融合预测模块该模块将药物分子图编码模块、药物-靶点相互作用图编码模块和基因共表达图编码模块得到的药物特征进行融合，构建药物组合的特征表示，然后通过分类器预测药物组合的协同作用。具体的融合方式采用拼接（Concatenation）和注意力机制相结合的方法，首先将三个模块得到的药物特征进行拼接，得到初始的药物组合特征向量：[h_{d1,d2}=\text{Concat}(h_{d1}^1,h_{d1}^2,h_{d1}^3,h_{d2}^1,h_{d2}^2,h_{d2}^3)]其中，(h_{d1}^1)、(h_{d1}^2)、(h_{d1}^3)分别为药物(d1)在三个模块中的特征表示，(h_{d2}^1)、(h_{d2}^2)、(h_{d2}^3)分别为药物(d2)在三个模块中的特征表示。然后，引入注意力机制，自动学习不同特征维度的重要性权重，对初始的药物组合特征向量进行加权融合：[\beta=\text{Softmax}(W_ah_{d1,d2}+b_a)][h_{d1,d2}'=\beta\odoth_{d1,d2}]其中，(W_a)和(b_a)为可学习的权重矩阵和偏置向量，(\beta)为注意力权重向量，(\odot)为元素-wise乘法操作，(h_{d1,d2}')为加权融合后的药物组合特征向量。最后，将加权融合后的特征向量输入到多层感知机（Multi-LayerPerceptron,MLP）分类器中，输出药物组合的协同作用概率：[y_{d1,d2}=\text{Softmax}(W_mh_{d1,d2}'+b_m)]其中，(W_m)和(b_m)为多层感知机的权重矩阵和偏置向量，(y_{d1,d2})为药物组合的协同作用概率分布。（三）模型训练与评估1.训练设置本研究采用交叉验证的方法对模型进行训练和评估，将数据集按照8:1:1的比例划分为训练集、验证集和测试集。模型的训练采用随机梯度下降（StochasticGradientDescent,SGD）优化器，损失函数采用交叉熵损失函数：[\text{Loss}=-\frac{1}{N}\sum_{i=1}^N\sum_{c=1}^Cy_i^c\log\hat{y}_i^c]其中，(N)为样本数量，(C)为类别数量，(y_i^c)为第(i)个样本的真实标签，(\hat{y}_i^c)为模型预测的第(i)个样本属于类别(c)的概率。训练过程中，采用早停（EarlyStopping）策略防止模型过拟合，当验证集上的损失连续5个epoch没有下降时，停止训练。模型的超参数通过网格搜索的方法在验证集上进行优化，包括图卷积网络的层数、图注意力网络的头数、图SAGE的聚合函数、多层感知机的隐藏层维度等。2.评估指标为了全面评估模型的性能，采用以下评估指标：准确率（Accuracy）：正确预测的样本数占总样本数的比例，反映模型的整体预测能力。精确率（Precision）：预测为正类的样本中真正为正类的比例，反映模型预测正类样本的准确性。召回率（Recall）：真正为正类的样本中被预测为正类的比例，反映模型对正类样本的识别能力。F1值（F1-Score）：精确率和召回率的调和平均数，综合反映模型的性能。ROC曲线与AUC值：ROC曲线以假阳性率为横坐标，真阳性率为纵坐标，AUC值为ROC曲线下的面积，反映模型对正负样本的区分能力。同时，将本研究提出的MFGNN模型与目前主流的药物组合预测模型进行对比，包括基于传统机器学习的模型（如支持向量机、随机森林）和基于图神经网络的模型（如GCN、GAT），验证MFGNN模型的优越性。四、实验结果与分析（一）数据集统计本研究共收集了12000组药物组合实验数据，其中协同作用的药物组合有3500组，拮抗作用的药物组合有2500组，无显著作用的药物组合有6000组。涉及的药物分子有1200种，靶点基因有800个，癌细胞系有50种。数据集的具体统计信息如下表所示：类别数量药物组合总数12000协同作用药物组合数3500拮抗作用药物组合数2500无显著作用药物组合数6000药物分子数1200靶点基因数800癌细胞系数50（二）模型性能评估1.交叉验证结果采用5折交叉验证的方法对MFGNN模型进行训练和评估，在验证集上的平均评估指标如下表所示：评估指标平均值标准差准确率0.8720.021精确率0.8560.032召回率0.8430.028F1值0.8490.025AUC值0.9350.018从结果可以看出，MFGNN模型在验证集上表现出了较好的性能，准确率达到了0.872，AUC值达到了0.935，说明模型对药物组合的协同作用具有较高的预测能力。2.对比实验结果将MFGNN模型与其他主流的药物组合预测模型进行对比，在测试集上的评估指标如下表所示：模型名称准确率精确率召回率F1值AUC值支持向量机（SVM）0.7650.7320.7180.7250.821随机森林（RF）0.8010.7760.7630.7690.865GCN0.8230.8010.7890.7950.892GAT0.8450.8260.8120.8190.910MFGNN（本研究）0.8680.8520.8390.8450.931从对比结果可以看出，本研究提出的MFGNN模型在各项评估指标上均优于其他对比模型。与传统的机器学习模型SVM和RF相比，MFGNN模型的准确率分别提高了10.3%和6.7%，AUC值分别提高了11.0%和6.6%，说明基于图神经网络的模型能够更好地捕捉生物数据中的复杂模式，提高预测精度。与单一的图神经网络模型GCN和GAT相比，MFGNN模型的准确率分别提高了4.5%和2.3%，AUC值分别提高了3.9%和2.1%，这得益于MFGNN模型整合了多源生物数据，通过多图融合的方式充分挖掘了药物-药物、药物-靶点、药物-基因之间的交互信息，从而提升了模型的性能。（三）模型可解释性分析为了提高模型的可解释性，本研究采用注意力可视化和特征重要性分析的方法，对MFGNN模型的预测结果进行解释。1.注意力可视化通过可视化药物组合特征向量的注意力权重，分析不同特征维度对模型预测结果的贡献。以一组具有协同作用的药物组合为例，可视化结果显示，药物分子的化学结构特征和药物-靶点相互作用特征的注意力权重较高，说明这两类特征对药物组合的协同作用预测具有重要影响。进一步分析发现，药物分子中与靶点结合的关键官能团的特征权重明显高于其他部分，表明模型能够自动识别药物分子中的关键结构信息，为理解药物组合的作用机制提供了线索。2.特征重要性分析采用置换重要性（PermutationImportance）的方法，计算不同特征对模型性能的影响程度。具体步骤为：随机打乱某一特征的取值，重新计算模型在测试集上的性能指标，性能指标的下降幅度即为该特征的重要性。实验结果表明，药物分子图编码模块得到的药物结构特征、药物-靶点相互作用图编码模块得到的靶点结合特征和基因共表达图编码模块得到的差异表达基因特征的重要性较高，分别占总重要性的35%、30%和25%，其余特征的重要性相对较低。这一结果与注意力可视化的结果一致，进一步验证了多源特征融合对模型性能的提升作用。（四）案例分析为了验证模型在实际应用中的有效性，本研究选取了一组已在临床中应用的药物组合进行案例分析。以治疗非小细胞肺癌的药物组合“厄洛替尼+贝伐珠单抗”为例，MFGNN模型预测该药物组合具有较高的协同作用概率，与临床实验结果一致。通过分析模型的注意力权重和特征重要性，发现厄洛替尼的EGFR靶点结合特征和贝伐珠单抗的VEGF靶点结合特征的权重较高，且这两个靶点在肺癌细胞的信号通路中具有重要的调控作用，说明模型能够准确捕捉药物组合的作用靶点和潜在机制。同时，本研究利用MFGNN模型对潜在的药物组合进行预测，筛选出了100组具有较高协同作用概率的药物组合，并通过文献调研和初步实验验证，发现其中20组药物组合已在相关研究中被报道具有协同作用，其余80组为新发现的潜在有效药物组合，为后续的实验研究提供了参考。五、研究创新点与不足（一）研究创新点多源生物数据的整合与多图融合建模：本研究整合了药物化学结构、药物-靶点相互作用、基因表达谱等多源生物数据，构建了药物分子图、药物-靶点相互作用图和基因共表达图，通过多图融合的图神经网络模型，充分挖掘了药物-药物、药物-靶点、药物-基因之间的复杂交互信息，提高了模型的预测精度。注意力机制的引入：在多图融合预测模块中引入了注意力机制，自动学习不同特征维度的重要性权重，实现了对药物组合特征的自适应融合，增强了模型的可解释性。模型性能的提升：通过与传统机器学习模型和单一图神经网络模型的对比实验，验证了本研究提出的MFGNN模型在药物组合预测任务中的优越性，