本发明所提供的靶点蛋白分子与药物分子之间相互作用强度的预测方法，包括：获取待预测的目标靶点蛋白分子与药物分子的复合物；对所述复合物进行图构建处理，得到目标分子图；将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值；其中，所述图神经网络中包括边的自注意力机制图网络。本发明通过使用边的自注意力机制图网络对复合物的目标分子图上边的信息进行更新，进而提高了图神经网络预测两个分子之间相互作用强度的准确率。
靶点蛋白分子与药物分子之间相互作用强度的预测方法
技术领域
[0001] 本发明涉及人工智能技术领域，尤其涉及的是一种靶点蛋白分子与药物分子之间相互作用强度的预测方法。
背景技术
[0002] 随着AI（Artificial Intelligence）技术和理论的日益成熟以及数据量的不断增长，AI算法被越来越多地运用到药物发现中。在AI算法中，图神经网络能够能很好地建模药物分子，在药物发现的任务中得到了广泛的应用。在药物开发中，需要找出与靶点蛋白分子结合强度高的药物分子，进入下一步开发，为了更有效地对候选药物分子进行筛选，需要通过计算对药物分子和靶点蛋白分子之间的结合强度做出预测。特别的，在知道候选药物分子和靶点蛋白分子结合构象的情况下，利用AI来预测药物分子和靶点蛋白分子之间的相互作用强度可以大大提高药物筛选的效率。
[0003] 在预测药物分子和靶点蛋白分子之间的相互作用强度时，现有的图神经网络都是针对两个分子构建的图的节点进行运算，边的信息作为辅助，难以对边的信息进行有效更新，导致对药物分子和靶点蛋白分子之间的相互作用强度的预测并不准确。
[0004] 因此，现有技术存在缺陷，有待改进与发展。
发明内容
[0005] 本发明要解决的技术问题在于，针对现有技术的上述缺陷，提供一种靶点蛋白分子与药物分子之间相互作用强度的预测方法，旨在解决现有技术中现有的图神经网络难以对边的信息进行有效更新，导致对药物分子和靶点蛋白分子之间的相互作用强度的预测并不准确的问题。
[0006] 本发明解决技术问题所采用的技术方案如下：
[0007] 一种靶点蛋白分子与药物分子之间相互作用强度的预测方法，所述方法包括：
[0008] 获取待预测的目标靶点蛋白分子与药物分子的复合物；
[0009] 对所述复合物进行图构建处理，得到目标分子图；
[0010] 将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值；
[0011] 其中，所述图神经网络中包括边的自注意力机制图网络。
[0012] 在一种实现方式中，所述获取待预测的目标靶点蛋白分子与药物分子的复合物，包括：
[0013] 获取待预测的目标靶点蛋白分子和药物分子候选集合；
[0014] 利用计算化学方法或者机器学习方法计算得到所述目标靶点蛋白分子与所述药物分子候选集合中各个药物分子的复合物。
[0015] 在一种实现方式中，对所述复合物进行图构建处理，得到目标分子图，包括：
[0016] 将所述复合物中的原子构建为节点；
[0017] 若两个原子之间存在化学键，则在两个原子之间构建一条边；
[0018] 若两个原子之间的距离小于预设距离阈值，则在两个原子之间构建一条边；
[0019] 根据构建的节点和边得到所述复合物对应的目标分子图。
[0020] 在一种实现方式中，根据构建的节点和边得到所述复合物对应的目标分子图，包括：
[0021] 根据构建的节点和边得到所述复合物的初始分子图；
[0022] 截取所述初始分子图中的药物分子结合部位作为目标分子图；
[0023] 其中，所述药物分子结合部位上的目标靶点蛋白分子与药物分子之间的距离处于预设范围内。
[0024] 在一种实现方式中，将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值，包括：
[0025] 将所述目标分子图上的节点特征和边特征共同输入预先训练的图神经网络，所述图神经网络中还包括：节点自注意力机制图网络以及池化和全连接神经网络；
[0026] 所述目标分子图上的节点特征经所述节点自注意力机制图网络处理后得到新的节点特征，所述目标分子图上的边特征经所述节点自注意力机制图网络及所述边的自注意力机制图网络处理后得到新的边特征；
[0027] 从新的边特征中提取相互作用的边，将提取的相互作用的边和新的节点特征共同输入到所述池化和全连接神经网络，得到所述复合物中目标靶点蛋白分子与药物分子之间的相互作用强度预测值。
[0028] 在一种实现方式中，将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值之后，还包括：
[0029] 获取预测得到的各个所述复合物对应的相互作用强度预测值；
[0030] 按照相互作用强度预测值由大到小的顺序对所述药物分子候选集合中的各个药物分子进行排序。
[0031] 在一种实现方式中，所述图神经网络的训练步骤包括：
[0032] 获取训练数据集，所述训练数据集中包括训练复合物，以及所述训练复合物中目标靶点蛋白分子与训练药物分子的相互作用强度训练值；
[0033] 对所述训练复合物进行图构建处理，得到目标训练分子图；
[0034] 将所述目标训练分子图输入初始图神经网络，以所述相互作用强度训练值为优化目标对所述初始图神经网络进行训练；
[0035] 训练完成后，得到已训练的图神经网络；
[0036] 其中，所述初始图神经网络中包括边的自注意力机制图网络。
[0037] 在一种实现方式中，所述训练复合物的获取步骤包括：
[0038] 获取目标靶点蛋白分子和训练药物分子候选集合；
[0039] 利用分子对接方法或自由能微扰方法得到所述目标靶点蛋白分子和所述训练药物分子候选集合中各个训练药物分子的训练复合物。
[0040] 在一种实现方式中，所述训练药物分子候选集合为随机选取预设候选分子库中的药物分子得来。
[0041] 在一种实现方式中，所述相互作用强度训练值是利用分子对接方法或者自由能微扰方法计算得到。
[0042] 在一种实现方式中，对所述训练复合物进行图构建处理，得到目标训练分子图，包括：
[0043] 将所述训练复合物中的原子构建为节点；
[0044] 若两个原子之间存在化学键，则在两个原子之间构建一条边；
[0045] 若两个原子之间的距离小于预设距离阈值，则在两个原子之间构建一条边；
[0046] 根据构建的节点和边得到所述训练复合物对应的目标训练分子图。
[0047] 在一种实现方式中，根据构建的节点和边得到所述训练复合物对应的目标训练分子图，包括：
[0048] 根据构建的节点和边得到所述训练复合物的初始训练分子图；
[0049] 截取所述初始训练分子图中的药物分子结合部位作为目标训练分子图；
[0050] 其中，所述药物分子结合部位上的目标靶点蛋白分子与训练药物分子之间的距离处于预设范围内。
[0051] 在一种实现方式中，将所述目标训练分子图输入初始图神经网络，以所述相互作用强度训练值为优化目标对所述初始图神经网络进行训练，包括：
[0052] 将所述目标训练分子图上的节点特征和边特征共同输入初始图神经网络，所述初始图神经网络中还包括：节点自注意力机制图网络以及池化和全连接神经网络；
[0053] 所述目标训练分子图上的节点特征经所述节点自注意力机制图网络处理后得到新的节点特征，所述目标训练分子图上的边特征经所述节点自注意力机制图网络及所述边的自注意力机制图网络处理后得到新的边特征；
[0054] 从新的边特征中提取相互作用的边，将提取的相互作用的边和新的节点特征共同输入到所述池化和全连接神经网络，得到所述训练复合物中目标靶点蛋白分子与训练药物分子之间的相互作用强度预测值；
[0055] 利用计算损失函数或者梯度下降方法对所述初始图神经网络进行训练，以使所述相互作用强度预测值接近所述相互作用强度训练值。
[0056] 本发明还提供一种靶点蛋白分子与药物分子之间相互作用强度的预测装置，所述装置包括：
[0057] 获取模块，用于获取待预测的目标靶点蛋白分子与药物分子的复合物；
[0058] 构建模块，用于对所述复合物进行图构建处理，得到目标分子图；
[0059] 预测模块，用于将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值；
[0060] 其中，所述图神经网络中包括边的自注意力机制图网络。
[0061] 本发明还提供一种终端，包括：存储器、处理器及存储在所述存储器上并可在所述处理器上运行的靶点蛋白分子与药物分子之间相互作用强度的预测程序，所述靶点蛋白分子与药物分子之间相互作用强度的预测程序被所述处理器执行时实现如上所述的靶点蛋白分子与药物分子之间相互作用强度的预测方法的步骤。
[0062] 本发明还提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序能够被执行以用于实现如上所述的靶点蛋白分子与药物分子之间相互作用强度的预测方法的步骤。
[0063] 本发明所提供的靶点蛋白分子与药物分子之间相互作用强度的预测方法，包括：获取待预测的目标靶点蛋白分子与药物分子的复合物；对所述复合物进行图构建处理，得到目标分子图；将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值；其中，所述图神经网络中包括边的自注意力机制图网络。本发明通过使用边的自注意力机制图网络对复合物的目标分子图上边的信息进行更新，进而提高了图神经网络预测两个分子之间相互作用强度的准确率。
附图说明
[0064] 图1是本发明中靶点蛋白分子与药物分子之间相互作用强度的预测方法较佳实施例的流程图。
[0065] 图2是本发明中图神经网络的工作原理示意图。
[0066] 图3是本发明中靶点蛋白分子与药物分子之间相互作用强度的预测装置的较佳实施例的功能原理框图。
[0067] 图4是本发明中终端的较佳实施例的功能原理框图。
具体实施方式
[0068] 为使本发明的目的、技术方案及优点更加清楚、明确，以下参照附图并举实施例对本发明进一步详细说明。应当理解，此处所描述的具体实施例仅仅用以解释本发明，并不用于限定本发明。
[0069] 现有图神经网络用于预测两个分子之间的相互作用强度时，都是针对两个分子构建的图的节点进行运算，边的信息作为辅助，难以对边的信息进行有效更新。这是由于现有的图神经网络中信息传递机制对边的信息操作相对简单，从而不利于对边的信息进行有效更新。
[0070] 并且，现有图神经网络一般是通过两个分子对应的分子图上的节点之间的注意力进行两个分子相互作用强度的计算，难以直接对两个分子之间的相互作用进行建模。这是由于现有的图神经网络考虑的是两个分子对应的分子图的节点之间的相互作用，间接建模相互结合强度，对相互作用本身没有直接建模。
[0071] 因此，导致了现有的图神经网络对药物分子和目标靶点蛋白分子之间的相互作用强度的预测并不准确。
[0072] 针对现有技术的上述缺陷，本实施例通过获取待预测的目标靶点蛋白分子与药物分子的复合物；对所述复合物进行图构建处理，得到目标分子图；将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值；所述图神经网络中包括边的自注意力机制图网络。本发明使用边的自注意力机制图网络对复合物的目标分子图上边的信息进行更新，进而能够提高图神经网络预测两个分子之间相互作用强度的准确率。
[0073] 请参见图1，图1是本发明中靶点蛋白分子与药物分子之间相互作用强度的预测方法的流程图。如图1所示，本发明实施例所述的靶点蛋白分子与药物分子之间相互作用强度的预测方法包括以下步骤：
[0074] 步骤S100、获取待预测的目标靶点蛋白分子与药物分子的复合物。
[0075] 具体地，首先获取待预测的复合物，所述复合物是指目标靶点蛋白分子与药物分子结合后的复合物。
[0076] 在一种实现方式中，所述步骤S100具体包括：
[0077] 步骤S110、获取待预测的目标靶点蛋白分子和药物分子候选集合；
[0078] 步骤S120、利用计算化学方法或者机器学习方法计算得到所述目标靶点蛋白分子与所述药物分子候选集合中各个药物分子的复合物。
[0079] 具体地，在药物开发中，需要找出与目标靶点蛋白分子结合强度高的药物分子，进入下一步开发。本实施例提供目标靶点蛋白分子A和包含n个不同药物分子的药物分子候选集合D，可以使用分子对接方法得到所述目标靶点蛋白分子A与所述药物分子候选集合D={D_1,...,D_n}中各个药物分子结合后的复合物(E_1,...,E_n)。本实施例利用计算化学方法或者机器学习方法计算出各个药物分子的复合物，以便于预测各个复合物的相互作用强度值，进而能够实现根据各个复合物的相互作用强度值筛选符合要求的药物分子，为药物开发提供条件。
[0080] 如图1所示，本发明实施例所述的方法还包括：
[0081] 步骤S200、对所述复合物进行图构建处理，得到目标分子图。
[0082] 具体地，本实施例将结合在一起的复合物E_i(i=1,...,n)抽象为一个图来进行表示，以便于图神经网络对复合物进行处理。
[0083] 在一种实现方式中，所述步骤S200具体包括：
[0084] 步骤S210、将所述复合物中的原子构建为节点；
[0085] 步骤S220、若两个原子之间存在化学键，则在两个原子之间构建一条边；
[0086] 步骤S230、若两个原子之间的距离小于预设距离阈值，则在两个原子之间构建一条边；
[0087] 步骤S240、根据构建的节点和边得到所述复合物对应的目标分子图。
[0088] 具体地，本实施例在进行图表示时，按照步骤S210、步骤S220和步骤S230的规则来进行处理。其中，所述预设距离阈值可以设置为5埃到8埃之间的数值。即，分子图中的节点是复合物中的原子，当一对原子之间存在化学键，或者当两个原子之间的距离小于预设距离阈值时，均在两个原子之间加入一条边。本实施例按照以上规则进行具体的图构建，对两个原子之间存在化学键以及两个原子之间的距离小于预设距离阈值这两种情况，均构建了边，以便于图神经网络对目标分子图中边的信息进行有效更新。
[0089] 在一种实施例中，所述步骤S240具体包括：
[0090] 步骤S241、根据构建的节点和边得到所述复合物的初始分子图；
[0091] 步骤S242、截取所述初始分子图中的药物分子结合部位作为目标分子图。
[0092] 其中，所述药物分子结合部位上的目标靶点蛋白分子与药物分子之间的距离处于预设范围内。
[0093] 具体地，由于包含整个目标靶点蛋白分子A的分子图的节点数量可能很大，因此，本实施例为了加快计算速度，并没有将构建好的初始分子图全部输入图神经网络中去，而是在复合物的初始分子图中截取距离药物分子一定范围内的子图（如15埃到20埃），截取的子图作为目标分子图。本实施例通过截取初始分子图中的药物分子结合部位作为目标分子图，以便于输入图神经网络，进而降低了图神经网络的计算量，提高了计算效率。
[0094] 如图1所示，本发明实施例所述的方法还包括：
[0095] 步骤S300、将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值。
[0096] 其中，所述图神经网络中包括边的自注意力机制图网络。本实施例的图神经网络结构在连接两个分子之间的边上使用了注意力机制，能够对边的信息进行有效更新，进而提高了预测两个分子之间相互作用强度的准确率，即，得到的相互作用强度预测值(yd_1,...,yd_n)更加准确。
[0097] 在一种实现方式中，所述步骤S300具体包括：
[0098] 步骤S310、将所述目标分子图上的节点特征和边特征共同输入预先训练的图神经网络，所述图神经网络中还包括：节点自注意力机制图网络以及池化和全连接神经网络；
[0099] 步骤S320、所述目标分子图上的节点特征经所述节点自注意力机制图网络处理后得到新的节点特征，所述目标分子图上的边特征经所述节点自注意力机制图网络及所述边的自注意力机制图网络处理后得到新的边特征；
[0100] 步骤S330、从新的边特征中提取相互作用的边，将提取的相互作用的边和新的节点特征共同输入到所述池化和全连接神经网络，得到所述复合物中目标靶点蛋白分子与药物分子之间的相互作用强度预测值。
[0101] 具体地，所述目标分子图在输入图神经网络时，被识别为节点特征和边特征。目标分子图的每个节点和边可以用化学和几何方法得到相应的特征。本发明的图神经网络包括：节点自注意力机制图网络、边的自注意力机制图网络，以及池化和全连接神经网络。节点自注意力机制图网络能够对节点特征进行处理，使得目标分子图中的节点使用注意力机制，边的自注意力机制图网络能够对边特征进行处理，使得两个分子连接的边上使用注意力机制。
[0102] 也就是说，本实施例在对图中节点进行运算的同时，能够直接对两个分子之间的相互作用进行建模，也能够对边的信息进行有效更新，进而提高了预测两个分子之间相互作用强度的准确率。
[0103] 在一种实施例中，所述步骤S300之后还包括：
[0104] 步骤S410、获取预测得到的各个所述复合物对应的相互作用强度预测值；
[0105] 步骤S420、按照相互作用强度预测值由大到小的顺序对所述药物分子候选集合中的各个药物分子进行排序。
[0106] 具体地，对每个复合物进行预测后，得到每个复合物对应的相互作用强度预测值，从而对药物分子候选集合D中的药物分子进行排序，排序规则可以是按照相互作用强度预测值由大到小的顺序，这样，可以筛选出相互作用强度预测值最高的药物分子作为候选药物分子进入药物研发的下一个环节。本实施例通过使用准确度高的图神经网络得到相互作用强度，提高了计算效率，进而提高了药物研发的效率。
[0107] 在一种实现方式中，所述图神经网络的训练步骤包括：
[0108] 步骤A100、获取训练数据集，所述训练数据集中包括训练复合物，以及所述训练复合物中目标靶点蛋白分子与训练药物分子的相互作用强度训练值；
[0109] 步骤A200、对所述训练复合物进行图构建处理，得到目标训练分子图；
[0110] 步骤A300、将所述目标训练分子图输入初始图神经网络，以所述相互作用强度训练值为优化目标对所述初始图神经网络进行训练；
[0111] 步骤A400、训练完成后，得到已训练的图神经网络。
[0112] 其中，所述初始图神经网络中包括边的自注意力机制图网络。
[0113] 具体地，本实施例使用边的自注意力机制图网络对边的信息进行更新，能够更好地建模分子之间的相互作用，使得模型能更精确地对两个分子之间的相互作用强度进行建模，提升训练之后模型的预测准确度。
[0114] 在一种实施例中，在所述步骤A100中，所述训练复合物的获取步骤包括：获取目标靶点蛋白分子和训练药物分子候选集合；利用计算化学方法，例如，分子对接方法或自由能微扰方法得到所述目标靶点蛋白分子和所述训练药物分子候选集合中各个训练药物分子的训练复合物。
[0115] 具体地，本发明利用计算化学方法，比如但不限于如分子对接（Molecular docking），自由能微扰（Free Energy Perturbation, FEP）等方法，来计算目标靶点蛋白分子和训练药物分子候选集合中各个训练药物分子的训练复合物。本发明提供目标靶点蛋白分子A和包含m个训练药物分子的训练药物分子候选集合B，利用分子对接方法（Molecular docking），得到目标靶点蛋白分子A和集合B={B_1,...,B_m}中每个分子B_i的复合物集合C=(C_1,...,C_m) (i=1,...,m)。本实施例通过选取训练药物分子候选集合，以获得较多数量的复合物，进一步提升训练之后模型的预测准确度。
[0116] 在一种实现方式中，所述训练药物分子候选集合为随机选取预设候选分子库中的药物分子得来。也就是说，本实施例可以提供预设候选分子库，在所述预设候选分子库中随机选取药物分子，形成训练药物分子候选集合。这样，本实施例的训练药物分子候选集合中的药物分子是随机选取，使得训练数据具有全面性，进一步提升了训练之后模型的预测准确度。
[0117] 在一种实现方式中，所述相互作用强度训练值是利用分子对接方法或者自由能微扰方法计算得到。本实施例是以相互作用强度训练值(y_1,...,y_m)对初始图神经网络进行训练，相互作用强度训练值可以是分子对接方法或者自由能微扰方法计算得到，将相互作用强度训练值作为复合物的标签来进行训练。分子对接方法可以得到复合物结构，但是计算相互作用强度的准确性不够高，可以用来做初步筛选。自由能微扰方法的计算速度很慢，例如，计算一对相互作用强度需要10小时。因此，本发明利用分子对接方法或者自由能微扰方法计算得到相互作用强度训练值作为训练标签，以对初始图神经网络进行训练，进一步通过已训练的模型来预测各个复合物对应的相互作用强度，从而提高计算相互作用强度的效率。
[0118] 在本实施例中，在预设候选分子库中随机选取一些药物分子形成训练药物分子候选集合，在预测时，就可以将所述预设候选分子库中除所述训练药物分子候选集合以外的药物分子利用已训练的图神经网络进行预测，以提高预测效率。
[0119] 在一种实施例中，所述步骤A200具体包括：
[0120] 步骤A210、将所述训练复合物中的原子构建为节点；
[0121] 步骤A220、若两个原子之间存在化学键，则在两个原子之间构建一条边；
[0122] 步骤A230、若两个原子之间的距离小于预设距离阈值，则在两个原子之间构建一条边；
[0123] 步骤A240、根据构建的节点和边得到所述训练复合物对应的目标训练分子图。
[0124] 具体地，本实施例训练过程中，在进行图表示时，按照步骤A210、步骤A220和步骤A230的规则来进行处理。其中，所述预设距离阈值可以设置为5埃到8埃之间的数值。即，分子图中的节点是复合物中的原子，当一对原子之间存在化学键，或者当两个原子之间的距离小于预设距离阈值时，均在两个原子之间加入一条边。本实施例按照以上规则进行具体的图构建，对两个原子之间存在化学键以及两个原子之间的距离小于预设距离阈值这两种情况，均构建了边，以便于训练图神经网络对目标分子图中边的信息进行有效更新。
[0125] 在一种实现方式中，所述步骤A240具体包括：
[0126] 步骤A241、根据构建的节点和边得到所述训练复合物的初始训练分子图；
[0127] 步骤A242、截取所述初始训练分子图中的药物分子结合部位作为目标训练分子图。
[0128] 其中，所述药物分子结合部位上的目标靶点蛋白分子与训练药物分子之间的距离处于预设范围内。
[0129] 具体地，由于包含整个目标靶点蛋白分子A的分子图的节点数量可能很大，因此，本实施例为了加快图神经网络的计算速度，在训练复合物的初始训练分子图中截取距离训练药物分子一定范围内的子图（如15埃到20埃），截取的子图作为目标训练分子图。本实施例通过截取初始训练分子图中的药物分子结合部位作为目标训练分子图，以便于输入图神经网络，进而降低了图神经网络的计算量，提高了计算效率。
[0130] 在一种实施例中，所述步骤A300具体包括：
[0131] 步骤A310、将所述目标训练分子图上的节点特征和边特征共同输入初始图神经网络，所述初始图神经网络中还包括：节点自注意力机制图网络以及池化和全连接神经网络；
[0132] 步骤A320、所述目标训练分子图上的节点特征经所述节点自注意力机制图网络处理后得到新的节点特征，所述目标训练分子图上的边特征经所述节点自注意力机制图网络及所述边的自注意力机制图网络处理后得到新的边特征；
[0133] 步骤A330、从新的边特征中提取相互作用的边，将提取的相互作用的边和新的节点特征共同输入到所述池化和全连接神经网络，得到所述训练复合物中目标靶点蛋白分子与训练药物分子之间的相互作用强度预测值；
[0134] 步骤A340、利用计算损失函数或者梯度下降方法对所述初始图神经网络进行训练，以使所述相互作用强度预测值接近所述相互作用强度训练值。
[0135] 具体地，依次将目标训练分子图输入初始图神经网络，所述目标训练分子图在输入图神经网络时，被识别为节点特征和边特征。目标训练分子图的每个节点和边可以用化学和几何方法得到相应的特征。本发明的初始图神经网络包括：节点自注意力机制图网络、边的自注意力机制图网络，以及池化和全连接神经网络。节点自注意力机制图网络能够对节点特征进行处理，使得目标分子图中的节点使用注意力机制，边的自注意力机制图网络能够对边特征进行处理，使得两个分子连接的边上使用注意力机制。本实施例以相互作用强度训练值作为标签，可以使用例如均方误差等损失函数和梯度下降的方法，训练初始图神经网络，使得相互作用强度训练值和相互作用强度预测值尽可能接近。
[0136] 也就是说，本实施例在对图中节点进行运算的同时，能够直接对两个分子之间的相互作用进行建模，也能够对边的信息进行有效更新，进而提高了预测两个分子之间相互作用强度的准确率。
[0137] 如图2所示，更新节点特征和边的特征这个过程重复N次。相互作用的边指的是，目标靶点蛋白分子和药物分子之间的边，不包含蛋白分子或者药物分子内部的边。池化(pooling)是指把边特征和节点特征组合得到一个特征。全连接神经网络(FC)是一个多层感知机(MLP)，由线性层组成。
[0138] 对于图2中节点自注意力机制图网络，用h和E分别表示图上节点特征和边的特征。
[0139] headk = Attention(QhWQk, KhWKk, VhWVk)=Attention(Qk, Kk, Vk)；
[0140] ；
[0141] ；
[0142] h’= concat(head1, …,headk)；
[0143] E’= concat(E1, …,Ek)；
[0144] 其中，headk表示第k个自注意力的输出，Attention()表示自注意力机制，(Qk, Kk, Vk)是自注意力机制的参数矩阵，Qh, Kh, Vh分别表示节点特征h的查询矩阵、键矩阵和值矩阵，d表示自注意力机制中的向量维度，T表示转置，h’是更新后节点特征，E’是更新后的边的特征，concat()表示拼接操作。
[0145] 对于图2中边的自注意力机制图网络，用E分别表示图上边的特征。
[0146] headk = Attention(QEWQk, KEWKk, VEWVk)；
[0147] ；
[0148] E’=concat(head1, …,headk)；
[0149] 其中headk表示第k个自注意力的输出，Attention()表示自注意力机制，WQk,WKk,VWk是自注意力机制的参数矩阵，QE,KE,VE分别表示边的特征E的查询矩阵、键矩阵和值矩阵，d表示自注意力机制中的向量维度，T表示转置，E’是更新后的边的特征，concat()表示拼接操作。
[0150] 进一步地，如图3所示，基于上述靶点蛋白分子与药物分子之间相互作用强度的预测方法，本发明还相应提供了一种靶点蛋白分子与药物分子之间相互作用强度的预测装置，包括：
[0151] 获取模块100，用于获取待预测的目标靶点蛋白分子与药物分子的复合物；
[0152] 构建模块200，用于对所述复合物进行图构建处理，得到目标分子图；
[0153] 预测模块300，用于将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值；
[0154] 其中，所述图神经网络中包括边的自注意力机制图网络。
[0155] 进一步地，如图4所示，基于上述靶点蛋白分子与药物分子之间相互作用强度的预测方法，本发明还相应提供了一种终端，包括：存储器20、处理器10及存储在所述存储器20上并可在所述处理器10上运行的靶点蛋白分子与药物分子之间相互作用强度的预测程序30，所述靶点蛋白分子与药物分子之间相互作用强度的预测程序30被所述处理器执行时实现如上所述的靶点蛋白分子与药物分子之间相互作用强度的预测方法的步骤。
[0156] 本发明还提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序能够被执行以用于实现如上所述的靶点蛋白分子与药物分子之间相互作用强度的预测方法的步骤。
[0157] 综上所述，本发明公开的靶点蛋白分子与药物分子之间相互作用强度的预测方法，包括：获取待预测的目标靶点蛋白分子与药物分子的复合物；对所述复合物进行图构建处理，得到目标分子图；将所述目标分子图输入预先训练的图神经网络，得到所述目标靶点蛋白分子与所述药物分子之间的相互作用强度预测值；其中，所述图神经网络中包括边的自注意力机制图网络。本发明通过使用边的自注意力机制图网络对复合物的目标分子图上边的信息进行更新，进而提高了图神经网络预测两个分子之间相互作用强度的准确率。
[0158] 应当理解的是，本发明的应用不限于上述的举例，对本领域普通技术人员来说，可以根据上述说明加以改进或变换，所有这些改进和变换都应属于本发明所附权利要求的保护范围。
