一种图神经网络的分布式全图训练计算方法、系统、设备及存储介质

AITNT
正文
推荐专利
一种图神经网络的分布式全图训练计算方法、系统、设备及存储介质
申请号:CN202410856927
申请日期:2024-06-28
公开号:CN118504609A
公开日期:2024-08-16
类型:发明专利
摘要
本申请公开了一种图神经网络的分布式全图训练计算方法、系统、设备及存储介质,该方法包括如下步骤:获取图数据集中的节点属性;根据测试集节点及其邻居节点,构建计算依赖图和通信依赖图;根据Metis算法将计算依赖图和通信依赖图划分到各个GPU设备上,在各个GPU设备上启动工作进程初始化模型参数并获取图分区数据;根据图分区数据与计算依赖图构建本地图和跨分区图;通过本地图、跨分区图和通信依赖图对图神经网络进行分布式全图训练;本方法实现了一个高效的基于部分聚合的计算与通信重叠的异步流水线,以及训练过程中零冗余的计算与通信,有效提高图神经网络的分布式全图训练计算效率,缩短了大规模图数据集上图神经网络模型训练的计算时间。
技术关键词
分区 计算方法 邻居 地图 神经网络模型训练 数据 模块 存储计算机程序 进程 训练集 节点特征 算法 处理器通信 超参数 可读存储介质 存储器
系统为您推荐了相关专利信息
1
基于深度相机的机器人可交互式局部地图构建方法及系统
局部地图构建方法 深度相机 地形特征 彩色图像 栅格地图
2
一种基于可见光的停车场智能寻车系统及使用方法
停车场寻车系统 人机交互界面 可见光通信模块 中央处理器 接收端
3
一种基于时间窗口冲突度的天基空间目标轮巡观测在轨任务规划方法
规划 工作量 星载遥感器 切片 参数计算方法
4
一种景区人流密度检测与预警方法及系统
预警方法 耦合特征 指数 壮锦制作 深度学习模型
5
一种基于解纠缠表示学习的能源科技新闻推荐方法
新闻推荐方法 科技 兴趣 能源 语义
添加客服微信openai178,进AITNT官方交流群
驱动智慧未来:提供一站式AI转型解决方案
沪ICP备2023015588号