[发明专利]模型训练方法及计算系统

权利要求书

1.一种计算系统，其特征在于，包括：处理单元和多个并行处理单元；所述处理单元与所述多个并行处理单元之间通信连接；

所述处理单元，用于将获取的图数据集存储至所述处理单元的内存中，并控制所述多个并行处理单元中的第一并行处理单元从所述处理单元的内存中读取所述图数据集中的图拓扑结构数据；

所述第一并行处理单元，用于对所述图拓扑结构数据进行采样，得到子图的拓扑结构数据；

所述处理单元，还用于控制所述多个并行处理单元中的第二并行处理单元获取所述子图的拓扑结构数据；以及，基于所述子图的拓扑结构数据中的顶点标识，从所述图数据集中的顶点特征数据中获取所述子图的顶点特征数据；并控制所述第二并行处理单元从所述处理单元的内存中读取所述子图的顶点特征数据；

所述第二并行处理单元，用于以所述子图的拓扑结构数据和顶点特征数据为样本，进行图神经网络训练。

2.根据权利要求1所述的系统，其特征在于，所述处理单元具体用于：

从所述第一并行处理单元中读取所述子图的拓扑结构数据，并存储至所述处理单元的内存中；

控制所述第二并行处理单元从所述处理单元的内存中读取所述子图的拓扑结构数据。

3.根据权利要求1所述的系统，其特征在于，所述多个并行处理单元之间通过通信连接；

所述处理单元，具体用于：控制所述第二并行处理单元从所述第一并行处理单元中读取所述子图的拓扑结构数据。

4.根据权利要求1-3任一项所述的系统，其特征在于，所述处理单元还用于：

在所述第一并行处理单元从所述处理单元的内存中读取所述图拓扑结构数据之后，释放与所述图拓扑结构数据对应的处理单元的内存区域。

5.根据权利要求4所述的系统，其特征在于，所述处理单元为至少两个，并且其中：

所述至少两个处理单元中的第一处理单元执行以下数据控制操作：控制第一并行处理单元从处理单元的内存中读取图数据集中的图拓扑结构数据；控制所述第一并行处理单元对所述图拓扑结构数据采样，得到子图的拓扑结构数据；以及，控制第二并行处理单元获取所述子图的拓扑结构数据和顶点特征数据；控制所述第二并行处理单元以所述子图的拓扑结构数据和顶点特征数据为样本进行图神经网络训练；

所述至少两个处理单元中的第二处理单元执行基于所述子图的拓扑结构数据中的顶点标识从所述图数据集中的顶点特征数据中获取所述子图的顶点特征数据的计算操作。

6.一种模型训练方法，其特征在于，包括：

将获取的图数据集存储至处理单元的内存中；

控制第一并行处理单元从处理单元的内存中读取图数据集中的图拓扑结构数据；

控制所述第一并行处理单元对所述图拓扑结构数据采样，得到子图的拓扑结构数据；

基于所述子图的拓扑结构数据中的顶点标识，从所述处理单元的内存存储的图数据集中的顶点特征数据中获取所述子图的顶点特征数据；

控制第二并行处理单元获取所述子图的拓扑结构数据和顶点特征数据；

控制所述第二并行处理单元以所述子图的拓扑结构数据和顶点特征数据为样本进行图神经网络训练。

7.根据权利要求6所述的方法，其特征在于，包括：

启动多个线程；利用所述多个线程循环执行以下步骤，直至所述图神经网络满足设定的条件：

利用第一线程控制所述第一并行处理单元对所述图拓扑结构数据采样，得到子图的拓扑结构数据；

利用所述第一线程和第二线程基于所述子图的拓扑结构数据中的顶点标识，从所述处理单元的内存存储的顶点特征数据中获取所述子图的顶点特征数据；

利用所述第二线程控制所述第二并行处理单元获取所述子图的拓扑结构数据和顶点特征数据；

利用第三线程控制所述第二并行处理以所述子图的拓扑结构数据和顶点特征数据为样本进行图神经网络训练。