[发明专利]基于图神经网络的时序数据自适应可信采样方法

说明书

本发明公开了一种基于图神经网络的时序数据自适应可信采样方法，包括：基于时序数据构造图神经网络G；提取每一个节点的两层树拓扑连接结构；计算所有同类型节点之间的相似度；计算所有根节点的两层树拓扑连接结构中相连叶子节点的可信度；使所有根节点的可信度高于阈值；从根节点的叶子节点的数据中按照可信度的比例随机采样，将采样结果作为状态信息；将状态信息输入到图神经网络实现状态的更新和聚合，以输出最终的节点表示；将图神经网络G生成的节点表示进行拼接，利用注意力机制实现评分预测，并结合真实评分计算出MSE损失。本发明通过时间序列的归一化时间戳信息分析图神经网络邻居节点的可信度，同时根据可信度利用自适应调节采样节点的拓扑结构，能够实现对时间序列的准确预测。

技术领域

本发明属于深度学习领域，具体涉及一种基于图神经网络的时序数据自适应可信采样方法。

背景技术

图神经网络(GNN)是针对图这一数据结构的深度学习模型，它在多项基于图的机器学习任务当中表现出色，并且具有较高的可解释性，近年来已成为一种广泛应用的图表示学习方法。例如，GNN可以从用户项目交互网络中学习得到有效的用户和项目表示用于推荐系统。GNN在广义上是指以图结构为输入的所有神经网络，其核心思想是学习一个映射，将图结构中的节点及其邻居节点的特征映射为该节点的低维隐特征表示，这就涉及到对节点进行采样。

在现有的很多方法中，采用均匀采样可能会忽略掉重要的邻居节点，这可能使的图的一些重要结构特征在采样中被剔除掉，图神经网络的数据源也就无法得到保障。此外，这些方法大都没有考虑结构本身的时序特性，并且忽略了图神经网络中间层输出的浅层结构特征，这都使得图神经网络的应用效果受到了一定程度的限制。

近年来，针对图神经网络在推荐系统中的应用研究成果颇多，但其中不少方法仍然存在下列问题：1.针对邻居节点的均匀采样方法可能会忽略掉重要邻居提供的信息以及次邻居节点等其他节点的重要信息；2.没有为评分预测提供合适的可学习参数，限制了图神经网络的性能表现；3.没有邻居结构存在的时序特性；4忽略了图神经网络中间层输出的浅层结构特征。

发明内容

鉴于以上所述现有技术的缺点，本发明的目的在于提供一种基于图神经网络的时序数据自适应可信采样方法，以解决现有技术中的至少一个缺陷。

为实现上述目的及其他相关目的，本发明提供一种基于图神经网络的时序数据自适应可信采样方法，包括：

步骤1：基于时序数据构造图神经网络G；其中，在图神经网络G中，相同类型的节点之间没有边，不同类型的节点之间的边存在权重；

步骤2：提取每一个节点的两层树拓扑连接结构；

步骤3：计算所有同类型节点之间的相似度；

步骤4：利用所述时序数据的归一化时间戳信息，计算所有根节点的两层树拓扑连接结构中相连叶子节点的可信度；

步骤5：若某一根节点的所有叶子节点的可信度都小于等于阈值，按照相似度高低，选取与所述某一根节点相似的节点的叶子节点并入到所述某一根节点的叶子节点中，并返回步骤4，直到所有根节点的可信度都高于阈值；

步骤6：从根节点的叶子节点的数据中按照可信度的比例随机采样，将采样结果作为状态信息；其中，所述可信度的比例指叶子节点的可信度与该叶子节点所属的根节点的可信度的比例，根节点的可信度等于该根节点下所有叶子节点的可信度的和；

步骤7：将所述状态信息输入到图神经网络实现状态的更新和聚合，以输出最终的节点表示；

步骤8：将图神经网络G生成的根节点和叶子节点表示进行拼接，利用注意力机制实现评分预测，并结合真实评分计算出MSE损失。

可选的，计算所有同类型节点之间的相似度包括：