摘要
本申请公开了一种图神经网络的分布式全图训练计算方法、系统、设备及存储介质,该方法包括如下步骤:获取图数据集中的节点属性;根据测试集节点及其邻居节点,构建计算依赖图和通信依赖图;根据Metis算法将计算依赖图和通信依赖图划分到各个GPU设备上,在各个GPU设备上启动工作进程初始化模型参数并获取图分区数据;根据图分区数据与计算依赖图构建本地图和跨分区图;通过本地图、跨分区图和通信依赖图对图神经网络进行分布式全图训练;本方法实现了一个高效的基于部分聚合的计算与通信重叠的异步流水线,以及训练过程中零冗余的计算与通信,有效提高图神经网络的分布式全图训练计算效率,缩短了大规模图数据集上图神经网络模型训练的计算时间。
技术关键词
分区
计算方法
邻居
地图
神经网络模型训练
数据
模块
存储计算机程序
进程
训练集
节点特征
算法
处理器通信
超参数
可读存储介质
存储器
系统为您推荐了相关专利信息
局部地图构建方法
深度相机
地形特征
彩色图像
栅格地图
停车场寻车系统
人机交互界面
可见光通信模块
中央处理器
接收端